結果

問題 No.1640 簡単な色塗り
ユーザー 🍮かんプリン🍮かんプリン
提出日時 2021-08-07 23:10:58
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 378 ms / 2,000 ms
コード長 3,994 bytes
コンパイル時間 1,953 ms
コンパイル使用メモリ 187,476 KB
実行使用メモリ 57,408 KB
最終ジャッジ日時 2024-06-29 17:02:55
合計ジャッジ時間 18,491 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 1 ms
5,376 KB
testcase_02 AC 1 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 241 ms
45,564 KB
testcase_05 AC 259 ms
47,300 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 1 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 192 ms
29,732 KB
testcase_11 AC 156 ms
25,012 KB
testcase_12 AC 134 ms
22,476 KB
testcase_13 AC 299 ms
43,664 KB
testcase_14 AC 320 ms
43,484 KB
testcase_15 AC 100 ms
17,284 KB
testcase_16 AC 126 ms
20,432 KB
testcase_17 AC 243 ms
36,836 KB
testcase_18 AC 20 ms
6,436 KB
testcase_19 AC 127 ms
22,396 KB
testcase_20 AC 180 ms
29,020 KB
testcase_21 AC 148 ms
24,292 KB
testcase_22 AC 14 ms
5,376 KB
testcase_23 AC 199 ms
31,140 KB
testcase_24 AC 49 ms
10,944 KB
testcase_25 AC 133 ms
22,916 KB
testcase_26 AC 223 ms
33,756 KB
testcase_27 AC 85 ms
16,532 KB
testcase_28 AC 286 ms
41,820 KB
testcase_29 AC 219 ms
33,544 KB
testcase_30 AC 25 ms
9,728 KB
testcase_31 AC 232 ms
51,696 KB
testcase_32 AC 204 ms
43,236 KB
testcase_33 AC 121 ms
30,308 KB
testcase_34 AC 170 ms
40,748 KB
testcase_35 AC 124 ms
28,876 KB
testcase_36 AC 26 ms
9,984 KB
testcase_37 AC 36 ms
13,312 KB
testcase_38 AC 197 ms
42,172 KB
testcase_39 AC 66 ms
21,516 KB
testcase_40 AC 74 ms
19,600 KB
testcase_41 AC 154 ms
35,060 KB
testcase_42 AC 77 ms
22,208 KB
testcase_43 AC 83 ms
27,268 KB
testcase_44 AC 77 ms
24,572 KB
testcase_45 AC 54 ms
19,964 KB
testcase_46 AC 27 ms
10,240 KB
testcase_47 AC 17 ms
8,448 KB
testcase_48 AC 205 ms
48,732 KB
testcase_49 AC 7 ms
5,504 KB
testcase_50 AC 2 ms
5,376 KB
testcase_51 AC 1 ms
5,376 KB
testcase_52 AC 356 ms
54,856 KB
testcase_53 AC 378 ms
57,408 KB
07_evil_01.txt AC 877 ms
93,836 KB
07_evil_02.txt AC 1,463 ms
139,032 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

/**
 *   @FileName	a.cpp
 *   @Author	kanpurin
 *   @Created	2021.08.07 23:08:20
**/

#include "bits/stdc++.h" 
using namespace std; 
typedef long long ll;



struct StronglyConnectedComponents {
private:
    vector<vector<int>> g, rg;
    vector<int> order; 
    vector<bool> visited;
    int V;
    void dfs(int v) {
        if (visited[v]) return;
        visited[v] = true;
        for (int u : g[v]) dfs(u);
        order.push_back(v);
    }
    void rdfs(int v, int c) {
        if (comp[v] != -1) return;
        comp[v] = c;
        for (int u : rg[v]) rdfs(u, c);
    }
public:
    vector<vector<int>> Graph; 
    vector<int> comp; 
    StronglyConnectedComponents(int v) : V(v) {
        g.resize(v);
        rg.resize(v);
        visited.resize(v);
        comp.resize(v);
    }
    
    void add_vertex(int n = 1) {
        V += n;
        g.resize(V);
        rg.resize(V);
        visited.resize(V);
        comp.resize(V);
    }
    void add_edge(int from, int to) {
        g[from].push_back(to);
        rg[to].push_back(from);
    }
    void build() {
        Graph = vector<vector<int>>();
        order = vector<int>();
        visited.assign(V, false);
        comp.assign(V, -1);
        for (int i = 0; i < V; i++) dfs(i);
        reverse(order.begin(), order.end());
        int number = 0;
        for (int i = 0; i < V; i++) {
            if (comp[order[i]] == -1) {
                rdfs(order[i], number++);
            }
        }
        Graph.resize(number);
        for (int i = 0; i < V; i++) {
            for (int v : g[i]) {
                if (comp[i] == comp[v]) continue;
                Graph[comp[i]].push_back(comp[v]);
            }
        }
    }
};

struct TwoSAT {
private:
    int _n;
    std::vector<bool> _answer;
    StronglyConnectedComponents scc;
    void at_most_one_naive(const vector<int> &x, const vector<bool> &t) {
        assert(x.size() == t.size());
        int _m = x.size();
        for (int i = 0; i < _m-1; i++) {
            for (int j = i+1; j < _m; j++) {
                add_clause(x[i],!t[i],x[j],!t[j]);
            }
        }
    }
public:
    TwoSAT():_n(0),scc(0){}
    explicit TwoSAT(int n) : _n(n), _answer(n), scc(2*n) {}
    void add_clause(int i, bool f, int j, bool g) {
        assert(0 <= i && i < _n);
        assert(0 <= j && j < _n);
        scc.add_edge(2 * i + (f ? 0 : 1), 2 * j + (g ? 1 : 0));
        scc.add_edge(2 * j + (g ? 0 : 1), 2 * i + (f ? 1 : 0));
    }
    
    void at_most_one(const vector<int> &x, const vector<bool> &t) {
        assert(x.size() == t.size());
        int _m = x.size();
        if (_m <= 6) {
            at_most_one_naive(x,t);
            return;
        }
        int _nn = _n;
        _n += _m;
        scc.add_vertex(_m*2);
        _answer.resize(_n);
        for (int i = 0; i < _m; i++) {
            add_clause(_nn+i,false,x[i],!t[i]);
            if (i) {
                add_clause(_nn+i,false,_nn+i-1,true);
                add_clause(x[i],!t[i],_nn+i-1,true);
            }
        }
    }
    bool satisfiable() {
        scc.build();
        auto id = scc.comp;
        for (int i = 0; i < _n; i++) {
            if (id[2 * i] == id[2 * i + 1]) return false;
            _answer[i] = id[2 * i] < id[2 * i + 1];
        }
        return true;
    }
    std::vector<bool> answer() { return _answer; }
};
int main() {
    int n;cin >> n;
    TwoSAT ts(n);
    vector<int> a(n),b(n);
    vector<vector<int>> x(n);
    vector<vector<bool>> t(n);
    for (int i = 0; i < n; i++) {
        cin >> a[i] >> b[i];
        x[a[i]-1].push_back(i);
        x[b[i]-1].push_back(i);
        t[a[i]-1].push_back(true);
        t[b[i]-1].push_back(false);
    }
    for (int i = 0; i < n; i++) {
        ts.at_most_one(x[i],t[i]);
    }
    if (!ts.satisfiable()) {
        puts("No");
    }
    else {
        puts("Yes");
        auto ans = ts.answer();
        for (int i = 0; i < n; i++) {
            cout << (ans[i]?a[i]:b[i]) << endl;
        }
    }
    return 0;
}
0