結果

問題 No.1421 国勢調査 (Hard)
ユーザー snow39snow39
提出日時 2021-03-05 23:31:43
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 40 ms / 2,000 ms
コード長 2,423 bytes
コンパイル時間 1,435 ms
コンパイル使用メモリ 116,092 KB
実行使用メモリ 10,240 KB
最終ジャッジ日時 2024-04-16 11:38:40
合計ジャッジ時間 3,219 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 2 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 2 ms
5,376 KB
testcase_13 AC 2 ms
5,376 KB
testcase_14 AC 2 ms
5,376 KB
testcase_15 AC 2 ms
5,376 KB
testcase_16 AC 2 ms
5,376 KB
testcase_17 AC 2 ms
5,376 KB
testcase_18 AC 2 ms
5,376 KB
testcase_19 AC 2 ms
5,376 KB
testcase_20 AC 2 ms
5,376 KB
testcase_21 AC 2 ms
5,376 KB
testcase_22 AC 39 ms
10,240 KB
testcase_23 AC 39 ms
10,240 KB
testcase_24 AC 40 ms
10,240 KB
testcase_25 AC 39 ms
10,112 KB
testcase_26 AC 39 ms
10,112 KB
testcase_27 AC 40 ms
10,240 KB
testcase_28 AC 40 ms
10,112 KB
testcase_29 AC 39 ms
10,112 KB
testcase_30 AC 39 ms
10,240 KB
testcase_31 AC 40 ms
10,112 KB
権限があれば一括ダウンロードができます

ソースコード

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;}

// is_extended flag is used for solving a linear_equation.
template<class T>
int Gauss(vector<vector<T>> &M,bool is_extended=false){// xor
    int R=M.size();
    int C=M[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(M[i][j]){
                pivot=i;
                break;
            }
        }

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

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

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

    // matrix M will be transformed.
    int rank=Gauss(M,true);

    vector<T> result;
    for(int i=rank;i<R;i++) if(M[i][C]) 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]){
                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<int>> A(M,vector<int> (N,0));
  vector<int> 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<int> ans=linear_equation(A,B);
  if(ans.empty()){
      cout<<-1<<endl;
      return 0;
  }
  for(auto n:ans) cout<<n<<"\n";

  return 0;
}
0