結果

問題 No.2442 線形写像
ユーザー lam6er
提出日時 2025-03-26 15:43:54
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 179 ms / 2,000 ms
コード長 1,049 bytes
コンパイル時間 305 ms
コンパイル使用メモリ 82,108 KB
実行使用メモリ 109,316 KB
最終ジャッジ日時 2025-03-26 15:43:59
合計ジャッジ時間 3,225 ms
ジャッジサーバーID
(参考情報)
judge5 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff #

def main():
    import sys
    input = sys.stdin.read().split()
    ptr = 0
    N = int(input[ptr])
    ptr +=1
    size = 1 << N
    A = []
    for _ in range(size):
        A.append(int(input[ptr]))
        ptr +=1

    if N ==0:
        if A[0] ==0:
            print("Yes")
        else:
            print("No")
        return

    # Build the basis
    basis = []
    for x in range(size):
        current = x
        for b in basis:
            current = min(current, current ^ b)
        if current !=0:
            basis.append(current)
            # Sort in descending order to maintain highest bit first
            basis.sort(reverse=True)

    # Check each element
    is_good = True
    for x in range(size):
        current_x = x
        xor_sum = 0
        for b in basis:
            if current_x & b:
                xor_sum ^= A[b]
                current_x ^= b
        if current_x !=0 or xor_sum != A[x]:
            is_good = False
            break

    print("Yes" if is_good else "No")

if __name__ == "__main__":
    main()
0