結果

問題 No.1023 Cyclic Tour
ユーザー carrot46carrot46
提出日時 2020-04-10 22:31:42
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,276 bytes
コンパイル時間 1,880 ms
コンパイル使用メモリ 175,648 KB
実行使用メモリ 18,944 KB
最終ジャッジ日時 2024-09-15 21:00:57
合計ジャッジ時間 11,030 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 39 WA * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <iomanip>
using namespace std;
#define reps(i,s,n) for(int i = s; i < n; i++)
#define rep(i,n) reps(i,0,n)
#define Rreps(i,n,e) for(int i = n - 1; i >= e; --i)
#define Rrep(i,n) Rreps(i,n,0)
#define ALL(a) a.begin(), a.end()
#define fi first
#define se second
typedef long long ll;
typedef vector<ll> vec;
typedef vector<vec> mat;

ll N,M,H,W,K,Q,A,B;
string S;
const ll MOD = 998244353;
//const ll MOD = (1e+9) + 7;
const ll INF = 1LL<<60;
typedef pair<ll, ll> P;
class union_find{
        vector<long> parent, tree_size;
    public:
        union_find(unsigned long _n) : parent(_n), tree_size(_n, 1){
            for(unsigned long i = 0; i < _n; ++i)parent[i] = i;
        }
        ll find(ll x){
            while(parent[x] != x)x = parent[x] = parent[parent[x]];
            return x;
        }
        void merge(ll a, ll b){
            a = find(a);
            b = find(b);
	    if(a==b) return;
            if(tree_size[a] < tree_size[b])swap(a, b);
            tree_size[a] += tree_size[b];
            parent[b] = a;
            return;
        }
	bool same(ll a, ll b){
	    a = find(a);
	    b = find(b);
	    return a == b;
	}
	ll p(ll a){return find(a);};
    };


mat G(int(1e+5) + 1, vec(0));
vec used(int(1e+5) + 10, 0);
bool dfs(int v, int k){
    used[v] = k;
    for(ll to : G[v]){
        if(used[to] == 0) {
            if(dfs(to, k)) return true;
        }else if(used[to] == k){
            return true;
        }
    }
    return false;
}
int main() {
    cin>>N>>M;
    union_find uf(N);
    vec a(M), b(M), c(M);
    rep(i,M) {
        cin>>a[i]>>b[i]>>c[i];
        --a[i];
        --b[i];
        if(c[i] == 1){
            if(uf.same(a[i], b[i])){
                cout<<"Yes"<<endl;
                return 0;
            }
            uf.merge(a[i], b[i]);
        }
    }
    vec num_in(N, 0);
    rep(i,M){
        if(c[i] == 2){
            ll pa = uf.p(a[i]), pb = uf.p(b[i]);
            if(uf.same(pa, pb)){
                cout<<"Yes"<<endl;
                return 0;
            }
            ++num_in[pb];
            G[pa].push_back(pb);
        }
    }
    int k = 0;
    rep(i,N){
        if(dfs(i, ++k)){
            cout<<"Yes"<<endl;
            return 0;
        }
    }
    cout<<"No"<<endl;
}
0