[백준] #1252 좋다 (python)

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

1253번: 좋다

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

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

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

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

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

예제 입력 1

10
1 2 3 4 5 6 7 8 9 10

예제 출력 1

8

✔️ Code

N = int(input())

num = list(map(int,input().split()))

count = 0

num.sort()

for i in range (0, N):
    a=0
    b=N-1
    while(a < b):
        if num[i] == num[a] + num[b]:
            if a != i and b != i:
                count+=1
                break

            elif a == i:
                a+=1

            elif b == i:
                b-=1

        if num[i] > num[a] + num[b]:
            a+=1

        if num[i] < num[a] + num[b]:
            b-=1 

print(count)