結果
問題 | No.2082 AND OR XOR |
ユーザー | taiga0629kyopro |
提出日時 | 2022-09-12 17:53:30 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 200 ms / 2,000 ms |
コード長 | 3,148 bytes |
コンパイル時間 | 216 ms |
コンパイル使用メモリ | 82,388 KB |
実行使用メモリ | 82,320 KB |
最終ジャッジ日時 | 2024-05-07 05:56:59 |
合計ジャッジ時間 | 4,894 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 39 ms
53,284 KB |
testcase_01 | AC | 38 ms
54,892 KB |
testcase_02 | AC | 44 ms
60,984 KB |
testcase_03 | AC | 107 ms
78,476 KB |
testcase_04 | AC | 84 ms
76,964 KB |
testcase_05 | AC | 125 ms
79,072 KB |
testcase_06 | AC | 125 ms
78,484 KB |
testcase_07 | AC | 96 ms
77,812 KB |
testcase_08 | AC | 90 ms
77,104 KB |
testcase_09 | AC | 83 ms
76,736 KB |
testcase_10 | AC | 94 ms
77,804 KB |
testcase_11 | AC | 125 ms
78,956 KB |
testcase_12 | AC | 140 ms
80,124 KB |
testcase_13 | AC | 147 ms
80,132 KB |
testcase_14 | AC | 92 ms
77,248 KB |
testcase_15 | AC | 101 ms
78,064 KB |
testcase_16 | AC | 106 ms
78,408 KB |
testcase_17 | AC | 76 ms
77,212 KB |
testcase_18 | AC | 98 ms
77,380 KB |
testcase_19 | AC | 106 ms
78,784 KB |
testcase_20 | AC | 142 ms
79,520 KB |
testcase_21 | AC | 75 ms
77,496 KB |
testcase_22 | AC | 98 ms
77,924 KB |
testcase_23 | AC | 110 ms
78,340 KB |
testcase_24 | AC | 140 ms
79,436 KB |
testcase_25 | AC | 107 ms
78,576 KB |
testcase_26 | AC | 87 ms
77,072 KB |
testcase_27 | AC | 108 ms
77,732 KB |
testcase_28 | AC | 200 ms
82,320 KB |
testcase_29 | AC | 175 ms
81,936 KB |
testcase_30 | AC | 198 ms
82,292 KB |
testcase_31 | AC | 173 ms
81,944 KB |
ソースコード
from _collections import defaultdict mod=998244353 def sol1(n,a,b,c): def op(x,y): return (a*(x&y)+b*(x|y)+c*(x^y))%4 ans=0 for bit in range(4**n): A=[] for i in range(n): A.append(bit%4) bit//=4 res1=A[0] res2=A[n-1] for i in range(n-1): res1=op(res1,A[i+1]) res2=op(res2,A[n-2-i]) if res1==res2: ans+=1 else: print(A,res1,res2) return ans%mod def sol2(n,a,b,c): def op(x,y): return (a*(x&y)+b*(x|y)+c*(x^y))%4 def mp(f,g): ff=[] gg=[] for i in range(4): ff.append(f%4) gg.append(g%4) f//=4 g//=4 res=[0]*4 for i in range(4): res[i]=ff[gg[i]] return res[0]+4*res[1]+16*res[2]+64*res[3] def tof(f): ff=[] for i in range(4): ff.append(f%4) f//=4 return ff func=[0]*4 for i in range(4): fs=[0]*4 for x in range(4):fs[x]=op(x,i) func[i]=fs[0]+4*fs[1]+16*fs[2]+64*fs[3] dp=[defaultdict(int) for i in range(n+1)] for i in range(4): dp[1][func[i],func[i]]+=1 for i in range(1,n): for f,g in dp[i]: dp[i][f,g]%=mod for j in range(4): dp[i+1][mp(func[j],f),mp(g,func[j])]+=dp[i][f,g] ans=0 for i in range(4): for j in range(4): for f,g in dp[n-2]: res1=tof(f)[i] res1=op(res1,j) res2=tof(g)[j] res2=op(res2,i) if res1==res2: ans+=dp[n-2][f,g] return ans%mod def sol3(n,a,b,c): def op(x,y): return (a*(x&y)+b*(x|y)+c*(x^y))%4 def mp(f,g): ff=[] gg=[] for i in range(4): ff.append(f%4) gg.append(g%4) f//=4 g//=4 res=[0]*4 for i in range(4): res[i]=ff[gg[i]] return res[0]+4*res[1]+16*res[2]+64*res[3] def tof(f): ff=[] for i in range(4): ff.append(f%4) f//=4 return ff func=[0]*4 for i in range(4): fs=[0]*4 for x in range(4):fs[x]=op(x,i) func[i]=fs[0]+4*fs[1]+16*fs[2]+64*fs[3] dp=[defaultdict(int) for i in range(n+1)] for i in range(4): dp[1][i,func[i]]+=1 for i in range(1,n): for x,F in dp[i]: dp[i][x,F]%=mod for j in range(4): dp[i+1][op(x,j),mp(F,func[j])]+=dp[i][x,F] ans=0 for an in range(4): for x,F in dp[n-1]: res1=op(x,an) res2=tof(F)[an] if res1==res2:ans+=dp[n-1][x,F] return ans%mod n,a,b,c=map(int,input().split()) print(sol3(n,a,b,c)) exit() cnt=0 for n in range(N,N+1): for a in range(4): for b in range(4): for c in range(4): ans=sol2(n,a,b,c) if ans!= 4**n and ans!=(4**n)//2: cnt+=1 print(a,b,c,cnt) print(ans,4**n)