알고리즘
[파이썬🐍] 프로그래머스 : 양과 늑대
코딩개미뚠뚠
2023. 2. 6. 00:58
반응형
def solution(info, edges):
visited = [0] * len(info)
answer = []
#양이 늑대보다 많을 때마다 answer리스트에 저장함
def dfs(sheep, wolf):
if sheep > wolf:
answer.append(sheep)
else:
return
for parent, child in edges:
#부모노드는 방문했지만 자식노드는 방문전일때
if visited[parent] and not visited[child]:
visited[child] = 1
if info[child] == 0:
dfs(sheep+1, wolf)
else:
dfs(sheep, wolf+1)
visited[child] = 0
visited[0] = 1
dfs(1, 0)
return max(answer)
반응형