BOJ 백준 1940: 주몽

Table of contents

들어가며

이 문제는 대놓고 투 포인터 문제입니다.

AC 받은 파이썬 코드

import sys

input = sys.stdin.readline


def OOB(idx):
    if idx < 0 or idx >= N:
        return True
    return False


if __name__ == "__main__":
    N = int(input().rstrip())
    M = int(input().rstrip())
    parts = [*sorted(map(int, input().rstrip().split()))]
    ans = 0
    st = 0
    en = N - 1
    while not OOB(st) and not OOB(en) and st < en:
        tmp_sum = parts[st] + parts[en]
        if tmp_sum == M:
            ans += 1
            st += 1
            en -= 1
            continue
        if tmp_sum < M:
            st += 1
            continue
        if tmp_sum > M:
            en -= 1
            continue
    print(ans)