백준 3649 로봇 프로젝트 Python3

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

3649번: 로봇 프로젝트

 

최종 코드

• GitHub -> https://github.com/hwayeon351/BEAKJOON-Algorithms/blob/main/BJ3649.py

GitHub - hwayeon351/BEAKJOON-Algorithms: 백준 알고리즘 소스 코드 모음

import sys
input = sys.stdin.readline
while True:
    try:
        xx = int(input())*(10**7)
        n = int(input())
        lego = []
        for _ in range(n):
            lego.append(int(input()))
        lego.sort()
        check = False
        left = 0
        right = n-1
        while left < right:
            sum = lego[left] + lego[right]
            if sum == xx:
                check = True
                print('yes', lego[left], lego[right])
                break
            elif sum < xx:
                left += 1
            elif sum > xx:
                right -= 1
        if not check: print('danger')
    except: break

---

풀이 과정

풀이 시간 40분

import sys
input = sys.stdin.readline
while True:
    try:
    	#xcm을 나노로 변환
        xx = int(input())*(10**7)
        n = int(input())
        lego = []
        for _ in range(n):
            lego.append(int(input()))
            
        #레고를 길이 순으로 정렬
        lego.sort()
        
        #두 레고 조각의 길이 합 = xx가 되는 경우가 있으면 True, 없으면 False
        check = False
        
        left = 0
        right = n-1
        while left < right:
            sum = lego[left] + lego[right]
            
            #두 레고 조각의 길이 합 = xx인 경우, 해당 경우가 가장 큰 차를 가지므로 답 출력
            if sum == xx:
                check = True
                print('yes', lego[left], lego[right])
                break
            #두 레고 조각의 길이 합이 xx보다 작은 경우, left 포인터를 증가시킨다
            elif sum < xx:
                left += 1
            #두 레고 조각의 길이 합이 xx보다 큰 경우, right 포인터를 감소시킨다
            elif sum > xx:
                right -= 1
                
        #구멍을 막을 레고 조각이 없는 경우, danger 출력
        if not check: print('danger')
    except: break
#시간복잡도 = O(T*N), 공간복잡도 = O(T*N)