[python] 백준 3649 로봇 프로젝트

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

3649번: 로봇 프로젝트

 

문제 해결

• 전형적인 투포인터 문제
• n==0일 때와 n==1일 때 불가능하므로 danger 출력하고 다음 테스트케이스 출력
• n>=2일 때부터 올림차순으로 정렬된 리스트에서 start =0 end = n-1을 잡고 두 원소의 합이 x가 될 때까지 숫자를 조절하여 찾는다.
• start>=end가 되면 찾을 수 없는 것이므로 danger 출력하고 다음 테스트 케이스 출력

 

CODE

while True:
    try:
        L = []
        x = int(input())*10000000
        n = int(input())
        if n==0:
            print("danger")
            continue
        elif n==1:
            input()
            print('danger')
            continue
        for _ in range(n):
            L.append(int(input()))
        L.sort()
        start = 0
        end = n-1
        while True:
            if L[start] + L[end] == x:
                print('yes', L[start], L[end])
                break
            elif L[start] + L[end] > x:
                end -= 1
            elif L[start] + L[end]<x:
                start += 1
            if start >= end:
                print('danger')
                break
    except EOFError:
        break