글 읽기 - 파이썬 고수님들 도와주세요ㅜㅜ

fghkk12d 1년 전 0

가지치기를 하고 돌아오는 과정에서 무언가 문제가 있는거 같은데

어떤 부분을 추가하거나 수정시켜야 할지 모르겠습니다.

아직 백트래킹을 쓰는것이 미숙하여 너무 어렵네요...

도와주시면 정말 감사하겠습니다. ㅜㅜ

import sys
input = sys.stdin.readline
# 숫자의 개수
N = int(input())
# 숫자 차례대로 들어가 있는 list
number_list = list(map(int, input().split()))
# 연산자 list(차례대로 덧셈(+)의 개수, 뺄셈(-)의 개수, 곱셈(×)의 개수, 나눗셈(÷)의 개수)
operator_list = list(map(int, input().split()))
# 결과값을 담을 list
result_list = []
# 이전값 되찾을 list
before = []

cnt = 0

sum_ = operator_list[0]
sub_ = operator_list[1]
mul_ = operator_list[2]
div_ = operator_list[3]
rs = number_list[0]

def function_(c):
    if c == N-1:
        result_list.append(rs)
        return

    for i in range(cnt,N-1):
        if (sum_!=0):
            rs = rs + number_list[i+1]  
            before.append(rs)
            cnt += 1
            sum_ -= 1
            function_(c+1)
            before.pop()
            rs = before[len(before)]
            sum_ +=1

        elif(sub_!=0):
            rs = rs - number_list[i]
            cnt += 1
            sub_ -= 1
            function_(c+1)
            before.pop()
            rs = before[len(before)]
            sub_ += 1

        elif(mul_!=0):
            rs = rs * number_list[i]
            cnt +=1
            mul_-=1
            function_(c+1)
            before.pop()
            rs = before[len(before)]
            mul_ += 1

        elif(div_!=0):
            rs = rs / number_list[i]
            cnt += 1
            div -=1
            function_(c+1)
            before.pop()
            rs = before[len(before)]
            div_ += 1

function_(0)

print(result_list)

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

파이썬 고수님들 도와주세요ㅜㅜ

fghkk12d 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말