結果
問題 |
No.1189 Sum is XOR
|
ユーザー |
![]() |
提出日時 | 2020-08-22 14:26:37 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 185 ms / 2,000 ms |
コード長 | 1,561 bytes |
コンパイル時間 | 401 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 104,184 KB |
最終ジャッジ日時 | 2024-10-15 08:42:16 |
合計ジャッジ時間 | 3,601 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 21 |
ソースコード
""" https://yukicoder.me/problems/no/1189 各桁1回づつしか使えない 各桁が重ならないような取り方の数を数える Aも実は1000通りぐらいしかない あとはdp 1000*1000ぐらい """ from sys import stdin import sys def modfac(n, MOD): f = 1 factorials = [1] for m in range(1, n + 1): f *= m f %= MOD factorials.append(f) inv = pow(f, MOD - 2, MOD) invs = [1] * (n + 1) invs[n] = inv for m in range(n, 1, -1): inv *= m inv %= MOD invs[m - 1] = inv return factorials, invs def modnCr(n,r,mod,fac,inv): #上で求めたfacとinvsを引数に入れるべし(上の関数で与えたnが計算できる最大のnになる) return fac[n] * inv[n-r] * inv[r] % mod def cdig(n): ret = 0 for i in range(10): if (2**i) & n > 0: ret += 1 return ret mod = 998244353 N,K = map(int,stdin.readline().split()) A = list(map(int,stdin.readline().split())) nums = [0] * 1024 if K > 10: print (0) sys.exit() facs,invs = modfac(K+10,mod) for i in A: nums[i] += 1 dp = [ [0]*(K+1) for i in range(1024) ] dp[0][0] = 1 tmps = [ (cdig(i),i) for i in range(1024) ] tmps.sort() print (tmps[:10],file=sys.stderr) for dnum,x in tmps: for i in range(1,1024): for nk in range(K): if x & i == 0: dp[x | i][nk+1] += dp[x][nk] * nums[i] dp[x | i][nk+1] %= mod ans = 0 for i in range(1024): ans += dp[i][K] ans %= mod print ((ans * invs[K]) % mod)