結果

問題 No.1189 Sum is XOR
ユーザー SPD_9X2
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff # 

"""

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)
0