結果

問題 No.1189 Sum is XOR
ユーザー 👑 SPD_9X2SPD_9X2
提出日時 2020-08-22 14:26:37
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 187 ms / 2,000 ms
コード長 1,561 bytes
コンパイル時間 328 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 104,452 KB
最終ジャッジ日時 2024-04-23 08:47:22
合計ジャッジ時間 4,162 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 142 ms
104,452 KB
testcase_01 AC 136 ms
104,244 KB
testcase_02 AC 147 ms
104,316 KB
testcase_03 AC 62 ms
77,440 KB
testcase_04 AC 59 ms
74,752 KB
testcase_05 AC 68 ms
86,016 KB
testcase_06 AC 79 ms
97,792 KB
testcase_07 AC 63 ms
79,744 KB
testcase_08 AC 49 ms
67,072 KB
testcase_09 AC 68 ms
67,328 KB
testcase_10 AC 49 ms
64,640 KB
testcase_11 AC 108 ms
80,000 KB
testcase_12 AC 143 ms
104,172 KB
testcase_13 AC 135 ms
101,244 KB
testcase_14 AC 124 ms
90,952 KB
testcase_15 AC 114 ms
80,164 KB
testcase_16 AC 117 ms
77,548 KB
testcase_17 AC 143 ms
79,436 KB
testcase_18 AC 109 ms
84,572 KB
testcase_19 AC 128 ms
95,904 KB
testcase_20 AC 187 ms
95,692 KB
testcase_21 AC 90 ms
76,480 KB
testcase_22 AC 90 ms
76,488 KB
権限があれば一括ダウンロードができます

ソースコード

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