結果

問題 No.1958 Bit Game
ユーザー pitP
提出日時 2022-05-28 12:56:27
言語 PyPy3
(7.3.15)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 1,089 bytes
コンパイル時間 286 ms
コンパイル使用メモリ 81,984 KB
実行使用メモリ 144,420 KB
最終ジャッジ日時 2024-09-20 21:44:19
合計ジャッジ時間 46,228 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 24 TLE * 6
権限があれば一括ダウンロードができます

ソースコード

diff # 

def dot(A1,A2):
    # n:行列のサイズ
    l = len(A1)
    m = len(A1[0])
    m1 = len(A2)
    n = len(A2[0])
    if m != m1:
        print("CANNOT CALUCULATE!!")
        return
    A = [[0]*n for _ in range(l)]
    for i in range(l):
        for j in range(n):
            for k in range(m):
                A[i][j] += A1[i][k]*A2[k][j]
                A[i][j] %= mod
    return A

mod = 998244353
n,x,y = map(int,input().split())
a = list(map(int,input().split()))
b = list(map(int,input().split()))

cnta = [0] * 19
cntb = [0] * 19

for num in a:
    digit = 0
    while num:
        res = num%2
        cnta[digit] += res
        num //= 2
        digit += 1
    
for num in b:
    digit = 0
    while num:
        res = num%2
        cntb[digit] += res
        num //= 2
        digit += 1

ans = 0
for i in range(19):
    a1 = [[x-cnta[i],0],[cnta[i],x]] 
    b1 = [[y,y-cntb[i]],[0,cntb[i]]]
    A = [[1,0],[0,1]]
    A = dot(a1,A) 
    A = dot(b1,A) 
    B = [[1,0],[0,1]]
    for j in range(n):
        B = dot(A,B)
    ans += pow(2,i,mod) * B[1][0]
    ans %= mod

print(ans)
0