[백준 1253] -Python

취알스 9주차 기타 알고리즘 - 4/5

1253 좋다

prefix sum 이용

import sys
input = sys.stdin.readline
n = int(input())
array = list(map(int, input().split()))
array.sort()
ans = 0

for i in range(n):
    arr = array[:i]+array[i+1:]
    start = 0
    end = n-2
    while start < end:
        check = arr[start] + arr[end]
        if check == array[i]:
            ans += 1
            break
        if check < array[i]:
            start += 1
        else:
            end -= 1

print(ans)

주의

처음에는 딕셔너리를 만들어서 풀었는데

11
0 1 2 3 4 5 6 7 8 9 10

이 예제가 통과하지 못함. 0과 자기 자신을 더한 경우 세면 안됨. 그래서 0을 없애봤는데

3
-1 0 1

이런 예제가 통과하지 못함.