백준 5624 : 좋은 수

https://www.acmicpc.net/problem/5624

5624번: 좋은 수

[ 문제풀이 ]

"A + B + C = D"를 "A + B = D - C"로 생각하는 것이 포인트인 문제이다.

또한, A + B로 만들 수 있는 수는 -200000~200000이기 때문에 배열을 하나 선언하여 해당 숫자 존재 유무를 빠르게 알 수 있도록 해준다.

 

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.StringTokenizer;

public class Main {

	public static void main(String[] args) {
		try (BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
			BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out))) {
			int N = Integer.parseInt(br.readLine());
			StringTokenizer st = new StringTokenizer(br.readLine());
			int[] data = new int[N];
			boolean[] visit = new boolean[400002];
			int answer = 0;
			for (int i = 0; i < N; ++i) {
				data[i] = Integer.parseInt(st.nextToken());
				for (int j = 0; j < i; ++j) {
					if (visit[data[i] - data[j] + 200000]) {
						answer++;
						break;
					}
				}
				for (int j = 0; j <= i; ++j) {
					visit[data[i] + data[j] + 200000] = true;
				}
			}
			bw.write(answer + "\n");
		} catch (Exception e) {
			e.printStackTrace();
		}
	}
}