inflearn logo
강의

강의

N
챌린지

챌린지

멘토링

멘토링

N
클립

클립

로드맵

로드맵

지식공유

파이썬 알고리즘 문제풀이 입문(코딩테스트 대비)

4. 합이 같은 부분집합(DFS)

sys.exit() 사용하지 않기

876

손수현

작성한 질문수 1

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)

python 코테 준비 같이 해요!

답변 1

8

김태원

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")

기존에 윈도우 10으로 잘 써왔는데 윈도우 11로 바꾸고 나서 채점이 안됩니다.

0

76

2

스택에서 ')'을 만나는 경우

0

78

3

문제가 어디있나요?

0

65

2

변수 or 함수명

0

61

1

침몰하는 타이타닉 문제 질문입니다

0

56

1

AA.py 책점 에러

0

57

1

오늘 구매했는데 파이썬 자료구조 궁금한거 있으면 답변이 잘 될까요.

0

111

2

5.동전분배하기 문제 밑에코드도 정답이될까요?

0

110

1

아나그램 비교 코드

0

116

2

AA.PY파일 복사 후 채점 진행할때 오류 발생합니다.

0

160

2

문제 링크가있나여?

0

147

2

채점기 Time Limit Exceeded 오류 문의

1

163

2

동적계획법은 사용하는 문제

0

126

2

제 코드 좀 봐주세요

0

148

1

예외가 존재할 가능성?

0

97

1

3번이 안풀립니다

0

93

0

5번 틀림

0

114

0

오류원인?

0

98

0

리스트 선언

0

106

1

침몰하는 타이타닉(그리디) 문제 질문

0

109

1

알고리즘

0

69

1

코딩테스트

0

92

1

DFS 순서 질문드립니다.

0

126

2

left, right를 사용한 풀이법에 대한 질문입니다

0

91

1