結果

問題 No.1189 Sum is XOR
ユーザー 👑 SPD_9X2SPD_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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 136 ms
104,184 KB
testcase_01 AC 128 ms
103,936 KB
testcase_02 AC 142 ms
104,112 KB
testcase_03 AC 57 ms
77,312 KB
testcase_04 AC 56 ms
74,496 KB
testcase_05 AC 65 ms
84,992 KB
testcase_06 AC 78 ms
97,024 KB
testcase_07 AC 60 ms
80,000 KB
testcase_08 AC 49 ms
66,560 KB
testcase_09 AC 49 ms
66,944 KB
testcase_10 AC 48 ms
64,896 KB
testcase_11 AC 105 ms
80,128 KB
testcase_12 AC 143 ms
103,808 KB
testcase_13 AC 132 ms
101,120 KB
testcase_14 AC 122 ms
90,624 KB
testcase_15 AC 113 ms
79,956 KB
testcase_16 AC 115 ms
77,356 KB
testcase_17 AC 142 ms
79,360 KB
testcase_18 AC 105 ms
84,096 KB
testcase_19 AC 128 ms
95,232 KB
testcase_20 AC 185 ms
95,360 KB
testcase_21 AC 89 ms
76,384 KB
testcase_22 AC 89 ms
75,904 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