結果
問題 | No.2733 Just K-times TSP |
ユーザー |
|
提出日時 | 2024-04-20 08:53:07 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 638 ms / 2,000 ms |
コード長 | 1,212 bytes |
コンパイル時間 | 191 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 103,552 KB |
最終ジャッジ日時 | 2024-10-12 04:16:13 |
合計ジャッジ時間 | 5,838 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 32 |
ソースコード
import typingimport sysfrom collections import defaultdict, dequeimport heapqinput = lambda: sys.stdin.readline().strip()inf = 10**18mod = 998244353# import pypyjit# pypyjit.set_param('max_unroll_recursion=-1')# sys.setrecursionlimit(10**6)def solve():N, M, K = map(int, input().split())# [v, k0, k1,...kN-1]# 頂点数はN*(K+1)^N <= 6*9^6 < 6*10^9# sum(k)について単調性あり# つまりbfsでOKadj = [[] for _ in range(N)]for _ in range(M):a, b = map(lambda x: int(x)-1, input().split())adj[a].append(b)adj[b].append(a)powk = [1]for i in range(N):powk.append(powk[-1]*(K+1)%mod)def dec(k):l = [0 for _ in range(N)]i = 0while k:l[i] = k%(K+1)k//=(K+1)i+=1return ldp = [[0 for _ in range((K+1)**N)] for _ in range(N)]for i in range(N):dp[i][powk[i]] = 1for k in range(powk[N]):ks = dec(k)for v in range(N):for v_to in adj[v]:if ks[v_to] < K:dp[v_to][k + powk[v_to]] += dp[v][k]dp[v_to][k + powk[v_to]]%=modans = 0for i in range(N):ans += dp[i][-1]print(ans%mod)def main():t = 1for _ in range(t):solve()main()