結果

問題 No.2895 Zero XOR Subset
ユーザー Today03Today03
提出日時 2024-09-24 14:48:33
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 3,333 bytes
コンパイル時間 3,677 ms
コンパイル使用メモリ 264,196 KB
実行使用メモリ 10,624 KB
最終ジャッジ日時 2024-09-24 14:48:57
合計ジャッジ時間 22,301 ms
ジャッジサーバーID
(参考情報)
judge2 / judge5
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,812 KB
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
testcase_32 WA -
testcase_33 WA -
testcase_34 WA -
testcase_35 WA -
testcase_36 WA -
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const int INF = 1e9 + 10;
const ll INFL = 4e18;

/*
    F_2 上の連立線形方程式
    ref: https://qiita.com/drken/items/a14e9af0ca2d857b85c3
*/

// 掃き出し法
// vector<vector<bool>> a: 連立方程式 Ax=b の拡大係数行列
// vector<int> where: ピボットとなる変数を記録するための配列
// return: a のランク
int rowReduction(vector<vector<bool>>& a, vector<int>& where) {
    int row = a.size(), col = a.front().size();
    int rank = 0;
    for (int c = 0; c < col - 1; c++) {
        int pivot = rank;
        while (pivot < row && !a[pivot][c]) pivot++;
        if (pivot == row) continue;
        swap(a[pivot], a[rank]);
        where.push_back(c);
        for (int r = 0; r < row; r++) {
            if (r != rank && a[r][c]) {
                // A[r]^=A[c]
                for (int i = 0; i < col; i++) a[r][i] = a[r][i] ^ a[rank][i];
            }
        }
        rank++;
    }
    return rank;
}

// 連立線形方程式 Ax=b を解く
// x0: 特殊解(b=0 の場合は自明解になる)
// ker: Ax=0 の解空間の基底
// 一般解は x0 と解空間の基底の任意の線形結合で表される
bool linearEquation(vector<vector<bool>> a, vector<bool> b, vector<bool>& x0, vector<vector<bool>>& ker) {
    int row = a.size(), col = a.front().size();
    assert(b.size() == row);
    vector<vector<bool>> a2 = a;
    for (int i = 0; i < row; i++) a2[i].push_back(b[i]);

    vector<int> where;
    int rank = rowReduction(a2, where);

    for (int r = rank; r < row; r++) {
        if (a2[r].back()) return false;
    }

    x0 = vector<bool>(col, false);
    for (int i = 0; i < rank; i++) x0[where[i]] = a2[i].back();

    /*int r=0;
    for(int c=0;c<col;c++){
        if(r<rank&&c==where[r]){
            r++;
            continue;
        }
        vector<bool>x(col);
        x[c]=true;
        for(int r2=0;r2<r;r2++)x[where[r2]]=a2[r2][c];
        ker.push_back(x);
        return true;
    }*/
    {
        // いわゆる noshi 基底
        vector<ll> a3(row);
        for (int r = 0; r < row; r++) {
            for (int c = 0; c < col; c++) a3[r] |= (ll)a[r][c] << c;
        }
        vector<ll> basis;
        for (ll e : a3) {
            for (ll b : basis) e = min(e, e ^ b);
            if (e) basis.push_back(e);
            if (basis.size() > 0) break;
        }

        ker = vector<vector<bool>>(basis.size(), vector<bool>(col, false));
        for (int r = 0; r < (int)basis.size(); r++) {
            for (int c = 0; c < col; c++) ker[r][c] = basis[r] >> c & 1;
        }
    }

    return rank;
}

int main() {
    int N;
    cin >> N;
    vector<ll> A(N);
    for (int i = 0; i < N; i++) cin >> A[i];
    const int L = 60;

    vector<vector<bool>> a(L, vector<bool>(N));
    for (int i = 0; i < N; i++) {
        for (int j = 0; j < L; j++) {
            a[j][i] = (A[i] >> j) & 1;
        }
    }

    vector<bool> x0;
    vector<vector<bool>> xs;

    bool res = linearEquation(a, vector<bool>(L, false), x0, xs);
    if (!res || xs.size() == 0) return cout << -1 << endl, 0;

    vector<int> ans;
    for (int i = 0; i < N; i++) {
        if (xs[0][i]) ans.push_back(i + 1);
    }

    cout << ans.size() << endl;
    for (int x : ans) cout << x << ' ';
    cout << endl;
}
0