인프런 커뮤니티 질문&답변
작성한 질문수
4번 성능 차이
작성
·
159
0
안녕하세요 유사하게 코드가 구현되고 4번 케이스에서 time limit에 걸리는데 혹시 다른 점이 있을까요?
def DFS(tmp):
global small
if sum(res) > small:
return
if tmp == M:
if sum(res) < small:
small = sum(res)
return
elif tmp > M:
return
else:
for i in range(N):
res[i]+=1
DFS(tmp+coin[i])
res[i]-=1
N = int(input())
coin = list(map(int, input().split()))
coin.sort(reverse=True)
M = int(input())
res = [0]*N
small = 217400000
DFS(0)
print(small)
답변 1
1
김태원
지식공유자
sum(res)는 res리스트를 전체 탐색하면서 모든 원소의 합을 구하는 작업을 하는 코드입니다.
이 함수가 DFS가 작동되는 순간 몇 번 호출될까요? 거의 DFS가 호출될 때마다 sum(res)도 호출된다고 봐도 과장은 아니라 생각됩니다. 이것때문에 time limit 이 생기는 거라 생각됩니다.
영상에서 설명한 것처럼 DFS(L+1, tmp+coin[i]) 호출구조로 바꾸면 sum함수를 호출할 필요가 없이 L자체가 사용한 동전의 개수입니다. DFS(L+1, tmp+coin[i]) 호출구조로 바꾸면 해결될거라 생각합니다.
def DFS(L, tmp):
global small
if tmp>M:
return
if L>=small: #영상설명에서는 L>small 와 같이 설명했는데, L>=small과 같이 "같다"를 추가하면 훨씬 빨라집니다.
return
if tmp == M:
if L< small:
small = L
return
else:
for i in range(N):
DFS(L+1, tmp+coin[i])