結果
| 問題 |
No.2733 Just K-times TSP
|
| コンテスト | |
| ユーザー |
|
| 提出日時 | 2024-04-20 13:05:14 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 1,231 ms / 2,000 ms |
| コード長 | 1,022 bytes |
| コンパイル時間 | 234 ms |
| コンパイル使用メモリ | 82,072 KB |
| 実行使用メモリ | 116,568 KB |
| 最終ジャッジ日時 | 2024-10-12 08:29:57 |
| 合計ジャッジ時間 | 8,358 ms |
|
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 2 |
| other | AC * 32 |
ソースコード
import sys
sys.setrecursionlimit(5*10**5)
input = sys.stdin.readline
from collections import defaultdict, deque, Counter
from heapq import heappop, heappush
from bisect import bisect_left, bisect_right
from math import gcd
n,m,k = map(int,input().split())
graph = [[] for i in range(n+1)]
for i in range(m):
u,v = map(int,input().split())
graph[u-1].append(v-1)
graph[v-1].append(u-1)
mod = 998244353
mem = [-1]*(9**7)
wa = 0
for i in range(n):
wa += k*pow(9, i)
pows = [pow(9, i) for i in range(9)]
def f(cnt):
if cnt//9 == wa:
mem[cnt] = 1
return 1
if mem[cnt] != -1:
return mem[cnt]
res = 0
last = cnt % 9
for nxt in graph[last]:
now = (cnt//pows[nxt+1]) % 9
if now == k: continue
new = cnt//9
new += pows[nxt]
new = new*9 + nxt
res += f(new)
res %= mod
mem[cnt] = res
return res
ans = 0
for i in range(n):
cnt = 0
cnt += pow(9, i+1)
ans += f(cnt+i)
ans %= mod
print(ans)