結果

問題 No.2895 Zero XOR Subset
ユーザー Today03Today03
提出日時 2024-09-23 06:59:54
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 3,941 bytes
コンパイル時間 3,267 ms
コンパイル使用メモリ 257,688 KB
実行使用メモリ 10,956 KB
最終ジャッジ日時 2024-09-23 07:00:11
合計ジャッジ時間 11,949 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 47 ms
9,344 KB
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 AC 35 ms
9,468 KB
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 AC 37 ms
9,552 KB
testcase_11 AC 35 ms
9,452 KB
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 AC 35 ms
9,460 KB
testcase_18 WA -
testcase_19 WA -
testcase_20 AC 36 ms
9,528 KB
testcase_21 AC 36 ms
9,520 KB
testcase_22 WA -
testcase_23 AC 37 ms
9,472 KB
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 の拡大係数行列
// return: a のランク

template <int Row, int Col>
struct BitMatrix {
    int row, col;
    array<bitset<Col>, Row> val;
    BitMatrix() {
        row = Row;
        col = Col;
        for (int i = 0; i < row; i++) val[i].reset();
    }
};

template <int row, int col>
int rowReduction(BitMatrix<row, col>& a, vector<int>& where) {
    int rank = 0;
    for (int c = 0; c < col - 1; c++) {
        int pivot = rank;
        while (pivot < row && !a.val[pivot][c]) pivot++;
        if (pivot == row) continue;
        swap(a.val[pivot], a.val[rank]);
        where.push_back(c);
        for (int r = 0; r < row; r++) {
            if (r != rank && a.val[r][c]) {
                for (int i = 0; i < c; i++) a.val[r][i] = a.val[r][i] ^ a.val[rank][i];
            }
        }
        rank++;
        if (rank == row) break;
    }
    return rank;
}

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

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

    for (int r = rank; r < row; r++) {
        if (a2.val[r][col]) return false;
    }
    if (!where.empty() && where.back() == col) return false;

    for (int i = 0; i < rank; i++) x0.val[where[i]][0] = a2.val[i][col];

    // Ax=0 の解空間の基底
    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.val[r2][c];
        ker.push_back(x);*/
        bitset<col> x;
        x[c] = 1;
        for (int r2 = 0; r2 < r; r2++) x[where[r2]] = a2.val[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);
        }

        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];
    for (int i = 0; i < N; i++) {
        if (A[i] == 0) {
            cout << 1 << endl;
            cout << i + 1 << endl;
            return 0;
        }
    }

    BitMatrix<60, 200000> a;
    for (int i = 0; i < N; i++) {
        for (int j = 0; j < 60; j++) a.val[j][i] = (A[i] >> j) & 1;
    }

    /*vector<bool> x0;
    vector<vector<bool>> xs;*/
    BitMatrix<200000, 1> x0;
    vector<bitset<200000>> xs;
    BitMatrix<60, 1> b;
    for (int i = 0; i < N; i++) b.val[i][0] = 0;

    bool res = linearEquation<60, 200000>(a, b, 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);
    }

    if (ans.size() == 0) return cout << -1 << endl, 0;

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