結果

問題 No.1421 国勢調査 (Hard)
コンテスト
ユーザー snow39
提出日時 2021-03-05 23:25:18
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 47 ms / 2,000 ms
コード長 2,299 bytes
コンパイル時間 1,228 ms
コンパイル使用メモリ 115,128 KB
最終ジャッジ日時 2025-01-19 12:01:54
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <cmath>
#include <map>
#include <queue>
#include <iomanip>
#include <set>
#include <tuple>
#define mkp make_pair
#define mkt make_tuple
#define rep(i,n) for(int i = 0; i < (n); ++i)
#define all(v) v.begin(),v.end()
using namespace std;
typedef long long ll;
const ll MOD=1e9+7;
template<class T> void chmin(T &a,const T &b){if(a>b) a=b;}
template<class T> void chmax(T &a,const T &b){if(a<b) a=b;}

int Gauss(vector<vector<ll>> &A,bool is_extended=false){//XOR
    int R=A.size();
    int C=A[0].size();
    int rank=0;
    for(int j=0;j<C;j++){
        if(is_extended&&j+1==C) break;
        int pivot=-1;
        for(int i=rank;i<R;i++){
            if(A[i][j]){
                pivot=i;
                break;
            }
        }

        if(pivot>=0){
            if(pivot!=rank){
                for(int k=0;k<C;k++) swap(A[pivot][k],A[rank][k]);
            }

            for(int i=0;i<R;i++){
                if(i!=rank&&A[i][j]){
                    for(int k=0;k<C;k++) A[i][k]^=A[rank][k];
                }
            }
            rank++;
        }
    }
    return rank;
}

vector<ll> linear_equation(vector<vector<ll>> A,vector<ll> B){
    int R=A.size();
    int C=A[0].size();
    vector<vector<ll>> M(R,vector<ll> (C+1,0));
    rep(i,R) rep(j,C) M[i][j]=A[i][j];
    rep(i,R) M[i][C]=B[i];

    int rank=Gauss(M,true);

    vector<ll> result;
    for(int i=rank;i<R;i++) if(M[i][C]!=0) return result;

    result.resize(C,0);
    for(int i=0;i<rank;i++){
        int tar=-1;
        for(int j=0;j<C;j++){
            if(M[i][j]!=0){
                tar=j;
                break;
            }
        }
        if(tar==-1) continue;
        result[tar]=M[i][C];
    }
    return result;
}

int main(){
  cin.tie(0);
  ios::sync_with_stdio(false);

  int N,M;
  cin>>N>>M;
  vector<vector<ll>> A(M,vector<ll> (N,0));
  vector<ll> B(M);
  rep(q,M){
      int K;
      cin>>K;
      vector<int> X(K);
      rep(i,K) cin>>X[i];
      rep(i,K) X[i]--;
      int value;
      cin>>value;
      
      rep(i,K) A[q][X[i]]=1;
      B[q]=value;
  }

  vector<ll> ans=linear_equation(A,B);
  if(ans.empty()){
      cout<<-1<<endl;
      return 0;
  }
  for(auto n:ans) cout<<n<<"\n";


  return 0;
}
0