Atcoderの ナップザック問題に関する質問です。
動的計画法で解いたのですが、一部のケースでTLEになってしまい、理由を教えていただきたいです。
python
1import sys 2import numpy as np 3sys.setrecursionlimit(200000) 4 5n,w = (int(x) for x in input().split()) 6a = [tuple(map(int, input().split())) for _ in range(n)] 7dp = np.zeros((n+1, w+1)) 8 9def rec(i, j): 10 if dp[i][j] != 0: 11 return dp[i][j] 12 if i == n: 13 res = 0 14 elif j < a[i][0]: 15 res = rec(i+1, j) 16 else: 17 res = max(rec(i+1, j), rec(i+1, j - a[i][0]) + a[i][1]) 18 dp[i][j] = res 19 return dp[i][j] 20 21print(int(rec(0, w))) 22
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2019/07/04 11:50