[백준] 1253번: 좋다

문제

N개의 수 중에서 어떤 수가 다른 수 두 개의 합으로 나타낼 수 있다면 그 수를 “좋다(GOOD)”고 한다.

N개의 수가 주어지면 그 중에서 좋은 수의 개수는 몇 개인지 출력하라.

수의 위치가 다르면 값이 같아도 다른 수이다.

입력

첫째 줄에는 수의 개수 N(1 ≤ N ≤ 2,000), 두 번째 줄에는 i번째 수를 나타내는 Ai가 N개 주어진다. (|Ai| ≤ 1,000,000,000, Ai는 정수)

출력

좋은 수의 개수를 첫 번째 줄에 출력한다.

풀이

이분 탐색을 활용한 풀이

이분 탐색을 활용하기 위해서 주어진 수를 정렬한다.
이후 주어진 수들의 각 개수를 카운팅 해서 저장한다.

다른 두 수를 골라서 합을 구한다.
그 합의 값이 배열에 있는 지 bisect_left으로 인덱스를 구한다.
그 인덱스의 값이 두 수의 합이고 두 수의 인덱스 값과 다르면
set에 추가한다.

두 수의 인덱스 값과 달라야 하는 이유는
어떤 수가 다른 두 수의 합으로 나타낼 수 있다면 이기 때문이다.

즉, c = a + b 가 성립되어야하기 때문에
a != c, b != c 여야한다.

이 과정이 모두 끝나면 set안의 요소의 개수를 출력한다.

import sys
from bisect import bisect_left

input = sys.stdin.readline

N = int(input())
numbers = list(map(int, input().split(" ")))


def solution(N, numbers):
	answer = 0
	like_numbers = set()
	numbers.sort()

	numbers_count = {}
	for number in numbers:
		if number in numbers_count:
			numbers_count[number] += 1
		else:
			numbers_count[number] = 1

	for i in range(N - 1):
		for j in range(i + 1, N):
			sum_number = numbers[i] + numbers[j]
			index = bisect_left(numbers, sum_number)
			if index < len(numbers) and numbers[index] == sum_number and index != i and index != j:
				like_numbers.add(numbers[index])

	for number in like_numbers:
		answer += numbers_count[number]
	print(answer)
	return

solution(N, numbers)