結果
問題 |
No.2428 Returning Shuffle
|
ユーザー |
|
提出日時 | 2023-08-19 09:46:37 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
TLE
|
実行時間 | - |
コード長 | 1,369 bytes |
コンパイル時間 | 83 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 224,696 KB |
最終ジャッジ日時 | 2024-11-29 01:09:10 |
合計ジャッジ時間 | 19,292 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 19 TLE * 4 |
ソースコード
def Multiple_LCM(A: list) -> int: ''' 複数の整数に対する最小公倍数を返す Parameters ---------- A:list ''' from collections import defaultdict MOD = 998244353 Exponent = defaultdict(int) for a in A: primes = [] for i in range(2, a+1): if i*i > a: break if a % i != 0: continue cnt = 0 while a % i == 0: cnt += 1 a //= i primes.append((i, cnt)) if a > 1: primes.append((a, 1)) for p, v in primes: Exponent[p] = max(Exponent[p], v) res = 1 for p, exp in Exponent.items(): res *= pow(p, exp, MOD) res %= MOD return res N, M = map(int, input().split()) P = [i for i in range(N)] for _ in range(M): L, *T = list(map(int, input().split())) sigma = [] for t in T: sigma.append(P[t-1]) for i in range(L): if i < L-1: j = T[i+1]-1 else: j = T[0]-1 P[j] = sigma[i] loop = [] seen = [False]*N for i in range(N): if seen[i]: continue tmp = 1 seen[i] = True next = P[i] while not seen[next]: seen[next] = True tmp += 1 next = P[next] loop.append(tmp) ans = Multiple_LCM(loop) print(ans)