[BOJ] 19774 ABCD-код

문제 링크

www.acmicpc.net/problem/19774

풀이

입력받은 4자리의 수를 2자리씩 끊어서 제곱하고 더해야 한다.

이는 각각 100으로 나눈 몫과 나머지를 통해 구할 수 있다.

Java 코드

import java.io.*;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
        int T = Integer.parseInt(in.readLine());

        StringBuilder out = new StringBuilder();
        while (T-- > 0) {
            int n = Integer.parseInt(in.readLine());
            int a = n / 100, b = n % 100;
            if ((a * a + b * b) % 7 == 1) out.append("YES\n");
            else out.append("NO\n");
        }

        System.out.println(out);
    }
}