結果

問題 No.1023 Cyclic Tour
ユーザー kyort0n
提出日時 2020-04-10 22:12:31
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 896 ms / 2,000 ms
コード長 4,024 bytes
コンパイル時間 2,726 ms
コンパイル使用メモリ 202,196 KB
実行使用メモリ 47,856 KB
最終ジャッジ日時 2024-09-15 20:37:13
合計ジャッジ時間 27,075 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<ll, ll> l_l;
typedef pair<int, int> i_i;
template<class T>
inline bool chmax(T &a, T b) {
    if(a < b) {
        a = b;
        return true;
    }
    return false;
}

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

const long double EPS = 1e-10;
const long long INF = 1e18;
const long double PI = acos(-1.0L);
//const ll mod = 1000000007;
struct UnionFind {
    vector<int> par;
    vector<int> rank;
    vector<ll> Size;
    UnionFind(int n = 1) {
        init(n);
    }

    void init(int n = 1) {
        par.resize(n + 1); rank.resize(n + 1); Size.resize(n + 1);
        for (int i = 0; i <= n; ++i) par[i] = i, rank[i] = 0, Size[i] = 1;
    }

    int root(int x) {
        if (par[x] == x) {
            return x;
        }
        else {
            int r = root(par[x]);
            return par[x] = r;
        }
    }

    bool issame(int x, int y) {
        return root(x) == root(y);
    }

    bool merge(int x, int y) {
        x = root(x); y = root(y);
        if (x == y) return false;
        if (rank[x] < rank[y]) swap(x, y);
        if (rank[x] == rank[y]) ++rank[x];
        par[y] = x;
        Size[x] += Size[y];
        return true;
    }

    ll size(int x){
        return Size[root(x)];
    }
};

typedef vector<vector<int>> UnWeightedGraph;

template< typename G >
struct StronglyConnectedComponents {
  const G &g;
  UnWeightedGraph gg, rg;
  vector< int > comp, order, used;

  StronglyConnectedComponents(G &g) : g(g), gg(g.size()), rg(g.size()), comp(g.size(), -1), used(g.size()) {
    for(int i = 0; i < g.size(); i++) {
      for(auto e : g[i]) {
        gg[i].emplace_back((int) e);
        rg[(int) e].emplace_back(i);
      }
    }
  }

  int operator[](int k) {
    return (comp[k]);
  }

  void dfs(int idx) {
    if(used[idx]) return;
    used[idx] = true;
    for(int to : gg[idx]) dfs(to);
    order.push_back(idx);
  }

  void rdfs(int idx, int cnt) {
    if(comp[idx] != -1) return;
    comp[idx] = cnt;
    for(int to : rg[idx]) rdfs(to, cnt);
  }

  void build(UnWeightedGraph &t) {
    for(int i = 0; i < gg.size(); i++) dfs(i);
    reverse(begin(order), end(order));
    int ptr = 0;
    for(int i : order) if(comp[i] == -1) rdfs(i, ptr), ptr++;

    t.resize(ptr);
    for(int i = 0; i < g.size(); i++) {
      for(auto &to : g[i]) {
        int x = comp[i], y = comp[to];
        if(x == y) continue;
        t[x].push_back(y);
      }
    }
  }
};


vector<l_l> e;

int main() {
    ll N, M;
    cin >> N >> M;
    UnionFind uni(N);
    for(int i = 0; i < M; i++) {
        ll a, b, c;
        cin >> a >> b >> c;
        a--;
        b--;
        if(c == 1) {
            if(!uni.merge(a, b)) {
                cout << "Yes" << endl;
                return 0;
            }
        } else {
            e.push_back({a, b});
        }
    }
    ll V = 0;
    map<int, int> inv;
    for(int i = 0; i < N; i++) {
        if(uni.root(i) == i) {
            inv[i] = V;
            V++;
        }
    }
    for(int i = 0; i < N; i++) {
        inv[i] = inv[uni.root(i)];
    }
    UnWeightedGraph g(V), t;
    set<l_l> st;
    for(auto tmp : e) {
        if(inv[tmp.first] == inv[tmp.second]) {
            cout << "Yes" << endl;
            return 0;
        }
        //cerr << inv[tmp.first] << " " << inv[tmp.second] << endl;
        if(st.count({inv[tmp.second], inv[tmp.first]})) {
            cout << "Yes" << endl;
            return 0;
        }
        st.insert({inv[tmp.first], inv[tmp.second]});
        g[inv[tmp.first]].push_back(inv[tmp.second]);
    }
    StronglyConnectedComponents<UnWeightedGraph> SCC(g);
    SCC.build(t);
    set<int> st2;
    for(int i = 0; i < V; i++) {
        //cerr << i << " " << SCC[i] << endl;
        if(st2.count(SCC[i])) {
            cout << "Yes" << endl;
            return 0;
        }
        st2.insert(SCC[i]);
    }
    cout << "No" << endl;
    return 0;
}
0