結果
問題 | No.1710 Minimum OR is X |
ユーザー |
|
提出日時 | 2021-10-15 22:50:48 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 416 ms / 2,000 ms |
コード長 | 1,561 bytes |
コンパイル時間 | 171 ms |
コンパイル使用メモリ | 82,604 KB |
実行使用メモリ | 77,148 KB |
最終ジャッジ日時 | 2024-09-17 17:56:19 |
合計ジャッジ時間 | 4,806 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 |
ソースコード
import sysinput = lambda :sys.stdin.readline()[:-1]ni = lambda :int(input())na = lambda :list(map(int,input().split()))sys.setrecursionlimit(10**7)yes = lambda :print("yes");Yes = lambda :print("Yes");YES = lambda : print("YES")no = lambda :print("no");No = lambda :print("No");NO = lambda : print("NO")#######################################################################p = 998244353N = 10 ** 3 # N は必要分だけ用意するfact = [1, 1] # fact[n] = (n! mod p)factinv = [1, 1] # factinv[n] = ((n!)^(-1) mod p)inv = [0, 1] # factinv 計算用for i in range(2, N + 1):fact.append((fact[-1] * i) % p)inv.append((-inv[p % i] * (p // i)) % p)factinv.append((factinv[-1] * inv[-1]) % p)def cmb(n, r):if (r < 0) or (n < r):return 0r = min(r, n - r)return ((fact[n] * factinv[r])%p) * factinv[n-r] % pdef pem(n,r):if (r < 0) or (n < r):return 0return (fact[n] * factinv[r])%pn,m,k = na()x = [int(i) for i in list(input())]c = []for i in range(n+1):c.append(0)for j in range(k):c[i] += cmb(i,j)c[i] %= p#print(c)dp = [0]*(n+1)dp[-1] = 1for i in range(m):op = [0]*(n+1)dp,op = op,dpif x[i]:for j in range(n+1):dp[j] += op[j]*c[j]*pow(2,n-j,p)dp[j] %= pelse:for j in range(n+1):for nj in range(j+1):if nj<k:continuedp[nj] += op[j] * pow(2,n-j,p)*cmb(j,nj)dp[nj] %= pprint(sum(dp)%p)