백준 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

import sys
input = sys.stdin.readline

n = int(input())
arr = list(map(int, input().split()))
arr.sort()

num = 0
for i in range(n):
    goal = arr[i]
    S = 0
    E = len(arr)-1
    while S < E:
        if arr[S] + arr[E] == goal:
            if S == i:
                S += 1
            elif E == i:
                E -= 1
            else:
                num += 1
                break
        elif arr[S] + arr[E] > goal:
            E -= 1
        elif arr[S] + arr[E] < goal:
            S += 1
print(num)

입력은 readline으로 받아서 시간을 단축하고 for문을 n번 돌린다. while문을 열러서 s가 e보다 작을때 if문을 사용해서 조건을 만족했을때 break가 걸리게 한다