結果
問題 | No.1969 XOR Equation |
ユーザー |
|
提出日時 | 2022-02-12 00:35:59 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 560 ms / 2,000 ms |
コード長 | 1,831 bytes |
コンパイル時間 | 210 ms |
コンパイル使用メモリ | 82,292 KB |
実行使用メモリ | 92,084 KB |
最終ジャッジ日時 | 2024-09-21 03:29:01 |
合計ジャッジ時間 | 9,448 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 36 |
ソースコード
def solve_mini(N,A,Y,M=60):dp = [[1<<(M+1) for j in range(N+1)] for i in range(M+1)]for j in range(Y>>M,N+1,2):dp[M][j] = 0low_sort = [[i for i in range(N)] for _ in range(M)]"""for t in range(M):low_sort[t].sort(key=lambda i:A[i]%(1<<(t+1)),reverse=True)"""pre = [i for i in range(N)]for t in range(M):zero,one = [],[]for i in pre:if A[i]>>t & 1:one.append(i)else:zero.append(i)pre = one + zerolow_sort[t] = [i for i in pre]for i in range(M)[::-1]:zero,one,two = 0,0,0for j in range(N):if A[j]>>i & 1:one += 1else:zero += 1if one&1==(Y>>i & 1):dp[i][0] = min(dp[i][0], 2*dp[i+1][two])if (zero+two)&1==(Y>>i & 1):dp[i][0] = min(dp[i][0], 2*dp[i+1][one+two]+1)if i==0:breakfor j in range(1,N+1):next_up = low_sort[i-1][j-1]if A[next_up]>>i & 1:one -= 1two += 1else:zero -= 1one += 1if j!=N:next_next_up = low_sort[i-1][j]if A[next_up]%(1<<(i)) == A[next_next_up]%(1<<(i)):continueif one&1==(Y>>i & 1):dp[i][j] = min(dp[i][j], 2*dp[i+1][two])if (zero+two)&1==(Y>>i & 1):dp[i][j] = min(dp[i][j], 2*dp[i+1][one+two]+1)if dp[0][0]==1<<(M+1):return -1return dp[0][0]for _ in range(int(input())):N,Y = map(int,input().split())A = list(map(int,input().split()))print(solve_mini(N,A,Y))