[BaekJoon] 3273번: 두 수의 합 (Python)

• Reference

Solution

import sys
from typing import List


def solution(nums: List[int], result: int) -> int:
    answer = 0

    nums.sort()

    lt, rt = 0, len(nums) - 1
    while lt < rt:
        tmp_sum = nums[lt] + nums[rt]

        if tmp_sum == result:
            answer += 1

        # 이제 현재의 lt는 더 이상 볼 필요 없다는 뜻
        if tmp_sum < result:
            lt += 1
            continue

        rt -= 1

    return answer


n = int(sys.stdin.readline().rstrip())
array = list(map(int, sys.stdin.readline().split()))
x = int(sys.stdin.readline().rstrip())

print(solution(array, x))