結果

問題 No.1420 国勢調査 (Easy)
コンテスト
ユーザー outline
提出日時 2021-03-05 22:23:35
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 4,353 bytes
コンパイル時間 1,722 ms
コンパイル使用メモリ 145,612 KB
最終ジャッジ日時 2025-01-19 11:25:56
ジャッジサーバーID
（参考情報） 		judge2 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 25 TLE * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
#include <cmath>
#include <queue>
#include <string>
#include <map>
#include <set>
#include <stack>
#include <tuple>
#include <deque>
#include <array>
#include <numeric>
#include <bitset>
#include <iomanip>
#include <cassert>
#include <chrono>
#include <random>
#include <limits>
#include <iterator>
#include <functional>
#include <sstream>
#include <fstream>
#include <complex>
#include <cstring>
#include <unordered_map>
#include <unordered_set>
using namespace std;

using ll = long long;
constexpr int INF = 1001001001;
constexpr int mod = 1000000007;
// constexpr int mod = 998244353;

template<class T>
inline bool chmax(T& x, T y){
    if(x < y){
        x = y;
        return true;
    }
    return false;
}
template<class T>
inline bool chmin(T& x, T y){
    if(x > y){
        x = y;
        return true;
    }
    return false;
}

struct StronglyConnectedComponents{
    using G = vector<vector<int>>;

    G graph;
    G rev;
    int n;
    int scc_sz = 0;
    G scc_graph;

    vector<int> scc_id;
    vector<int> vs;
    vector<bool> visited;

    StronglyConnectedComponents() = default;

    StronglyConnectedComponents(int V) : n(V) {
        graph.resize(n);
        rev.resize(n);
        scc_id.assign(n, -1);
        visited.assign(n, false);
    }

    StronglyConnectedComponents(G& g) : graph(g), n(g.size()) {
        rev.resize(n);
        scc_id.assign(n, -1);
        visited.assign(n, false);
        
        for(int u = 0; u < n; ++u){
            for(int v : graph[u])   rev[v].push_back(u);
        }
    }

    void resize(int V){
        n = V;
        graph.resize(n);
        rev.resize(n);
        scc_id.assign(n, -1);
        visited.assign(n, false);
    }

    void add_edge(int u, int v){
        graph[u].emplace_back(v);
        rev[v].emplace_back(u);
    }

    void dfs(int from){
        visited[from] = true;
        for(int to : graph[from]){
            if(!visited[to])  dfs(to);
        }
        vs.push_back(from);
    }

    void rdfs(int from, int k){
        scc_id[from] = k;
        for(int to : rev[from]){
            if(scc_id[to] == -1)    rdfs(to, k);
        }
    }

    void build(){
        for(int v = 0; v < n; ++v){
            if(!visited[v]) dfs(v);
        }
        reverse(vs.begin(), vs.end());

        for(int v : vs){
            if(scc_id[v] == -1)    rdfs(v, scc_sz++);
        }

        scc_graph.resize(scc_sz);
        for(int v = 0; v < n; ++v){
            scc_graph[scc_id[v]].push_back(v);
        }

        return;
    }

    int size() { return scc_sz; }

    vector<int> get_set(int k) { return scc_graph[k]; }

    int operator[](int v) { return scc_id[v]; }
};

struct TwoSAT {
    StronglyConnectedComponents scc;
    int n;
    vector<int> sat;

    TwoSAT(int V) : n(V) {
        scc.resize(n * 2);
        sat.resize(n);
    }

    // [false] [true]
    void add_edge(int i, bool f, int j, bool g){
        scc.add_edge(i * 2 + (f ? 0 : 1), j * 2 + (g ? 1 : 0));
        scc.add_edge(j * 2 + (g ? 0 : 1), i * 2 + (f ? 1 : 0));
    }

    // sat[v] = true  -> v = false
    // sat[v] = false -> v = true
    bool satisfiable(){
        scc.build();
        for(int i = 0; i < n; ++i){
            if(scc[i * 2] == scc[i * 2 + 1])    return false;
            sat[i] = scc[i * 2] < scc[i * 2 + 1];
        }
        return true;
    }

    vector<int> get_sat() { return sat; }
};

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

    int N, M;
    cin >> N >> M;
    vector<int> A(M), B(M), Y(M);
    for(int i = 0; i < M; ++i){
        cin >> A[i] >> B[i] >> Y[i];
        --A[i], --B[i];
    }

    vector<int> X(N);
    for(int k = 0; k < 30; ++k){
        TwoSAT ts(N);
        for(int i = 0; i < M; ++i){
            if(Y[i] >> k & 1){
                ts.add_edge(A[i], false, B[i], false);
                ts.add_edge(A[i], true, B[i], true);
            }
            else{
                ts.add_edge(A[i], false, B[i], true);
                ts.add_edge(A[i], true, B[i], false);
            }
        }
        if(!ts.satisfiable()){
            cout << -1 << endl;
            return 0;
        }
        for(int i = 0; i < N; ++i){
            if(!ts.sat[i])  X[i] += 1 << k;
        }
    }

    for(int i = 0; i < N; ++i)  cout << X[i] << '\n';

    return 0;
}
0