글 읽기 - 파이썬 permutations 함수사용 시간초과

sososunny0317 4년 전 0

permutations 함수를 써서

N개 연산자중에 N-1개를 뽑아서 경우의 수를 구하는 방법을 사용했습니다.

ㅠㅠ 시간초과가 나네요 ㅠㅠ

permutations 함수를 사용하면 안될까요?

아님 입력또는출력에서 시간 개선을 할 수 있는 부분이 있을까요?

from itertools import permutations

N = int(input())
a = list(map(int,input().split()))
cnt = list(map(int,input().split()))
cal = [0]*cnt[0] + [1]*cnt[1] + [2]*cnt[2] + [3]*cnt[3]
result = []

for i in set(permutations(cal, r=N-1)):
    ans = a[0]
    idx = 0
    for j in i:
        idx += 1
        if j == 0:
            ans += a[idx]
        if j == 1:
            ans -= a[idx]
        if j == 2 :
            ans *= a[idx]
        if j == 3:
            if ans < 0 :
                ans = abs(ans)//a[idx]
                ans = -ans
            else:
                ans = ans//a[idx]        
    result.append(ans)
        
print(max(result))
print(min(result))

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

문제

문제

출처

ICPC

파이썬 permutations 함수사용 시간초과

sososunny0317 4년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말