본문 바로가기
프로그래밍/알고리즘 & 자료구조

백준 1940 : 주몽 with 파이썬

by Play_With 2024. 2. 1.
반응형

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

 

1940번: 주몽

첫째 줄에는 재료의 개수 N(1 ≤ N ≤ 15,000)이 주어진다. 그리고 두 번째 줄에는 갑옷을 만드는데 필요한 수 M(1 ≤ M ≤ 10,000,000) 주어진다. 그리고 마지막으로 셋째 줄에는 N개의 재료들이 가진 고

www.acmicpc.net

# 포인트

1. result를 while문 안에 넣어 반복동안에 갱신되도록

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

cnt=0
st, end=0, n-1

while st<end:
    result=nums[st]+nums[end] #result를 while문 안에 넣어 반복동안에 갱신되도록
    if result==m:
        cnt+=1
        result-=nums[end]
        end-=1
    elif result>m:
        result-=nums[end]
        end-=1
    else :
        result-=nums[st]
        st+=1

print(cnt)
반응형

댓글