-
카테고리
-
세부 분야
알고리즘 · 자료구조
-
해결 여부
미해결
sys.exit() 사용하지 않기
20.05.26 01:14 작성 조회수 574
1
코딩테스트 환경에서는 sys 라이브러리 사용이 안되는경우도 있다고 합니다. 메인 코드에 switch라는 글로벌 변수를 두고(1로 초기화), DFS 조건식이 만족되었을 때 switch를 0로 바꾸어 코딩해도 실행시간이나 이 외에 다른 이슈가 없는지 궁금해 질문 올립니다.
이 외에 sys.exit(0) 말고 다른 방법으로 종료시킬 방법이 있는지 궁금합니다.
def DFS(L, subTotal):
global switch
if L == n:
if subTotal == (total_sum - subTotal):
print("Yes")
switch = 0
sys.exit()
else:
if switch :
DFS(L+1, subTotal + num_lst[L])
if switch :
DFS(L+1, subTotal)
if __name__ == '__main__':
sys.stdin = open('TestData/4/in5.txt')
switch = 1
n = int(input()) # 원소 개수
num_lst = list(map(int, input().split()))
total_sum = sum(num_lst)
DFS(0, 0)
답변을 작성해보세요.
8
김태원
지식공유자2020.05.26
exit() 함수를 사용못 할 경우 저도 손수현님 처럼 신호변수를 두어 답을 찾으면 신호을 넣어 그 이후 스택에 남아 있는 함수들이 호출되면 바로 return 문으로 종료해버리는 방법을 씁니다. 아래는 손수현님의 코드를 제가 하듯이 고친것입니다. 둘의 차이는 없습니다.
def DFS(L, subTotal):
global switch
if switch: return
if L == n:
if subTotal == (total_sum - subTotal):
print("Yes")
switch = 1
else:
DFS(L+1, subTotal + num_lst[L])
DFS(L+1, subTotal)
if __name__ == '__main__':
switch = 0
n = int(input()) # 원소 개수
num_lst = list(map(int, input().split()))
total_sum = sum(num_lst)
DFS(0, 0)
if switch==0: print("NO")
답변 1