글 읽기 - 무엇이 문제일까요(파이썬)

wbjkwjj 3년 전 0

틀렸습니다가 나온 코드입니다

문제를 풀 때 이용한 로직을 정리해서 주석으로 달았습니다.

어떤 부분에서 문제가 생기는 것일까요 지적 부탁드려요ㅠㅠ

from sys import stdin
import copy
def is_operator(x):
    return x=='+' or x=='*' or x=='-'
def operate(x,gwal):#list x, boolean gwal
    #(selecting index of terms to cover with bracket)
    #and evaluate results based on location of brackets
    lis=[]
    i=0
    for i in x:
        lis.append(i)
    for i in range(1,len(x),2):
        if gwal[i]:
            a=int(x[i-1])
            b=int(x[i+1])
            c=0
            if x[i]=='+':
                c=a+b
                lis[i-1]=c
            elif x[i]=='-':
                c=a-b
                lis[i-1]=c
            else:
                c=a*b
                lis[i-1]=c
            lis[i]='a'
            lis[i+1]='a'
    while 'a' in lis:
        lis.remove('a')
    ress=int(lis[0])
    for i in range(len(lis)):
        if is_operator(lis[i]):
            if lis[i]=='+':
                ress+=int(lis[i+1])
            elif lis[i]=='-':
                ress-=int(lis[i+1])
            else:
                ress*=int(lis[i+1])
    print(lis)
    return ress
             
if __name__=='__main__':
    def dfs(x,result,i,gwal):
        #calculate all values that can be generated and store them into list.
        if(i==len(x)):
            return
        n=len(x)
        a=operate(x,gwal)
        if a not in result:
            result.append(a)
        if i+1<n and not gwal[i-1] and not gwal[i] and not gwal[i+1]:
            #if i-1,i,i+1th terms are not covered in brackets
            #do dfs making them covered in brackets
            gwa=copy.deepcopy(gwal)
            gwa[i]=True
            gwa[i-1]=True
            gwa[i+1]=True#cover them
            dfs(x,result,i+2,gwa)
        dfs(x,result,i+2,gwal)# dfs without covering i-1,i,i+1th terms into bracket
    n=int(stdin.readline())
    exp=list(stdin.readline().rstrip())
    result=[]
    gwalho=[False]*n
    i=1
    maxx=float("-inf")
    if n==1:
        print(exp[0])
    else:
        dfs(exp,result,i,gwalho)
        for i in result:
            if i>maxx:
                maxx=i
        #result.sort()
        #print(result)
    print(maxx)

           

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

문제

문제

출처

ICPC

무엇이 문제일까요(파이썬)

wbjkwjj 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말