Submission #28590106

---

Source Code Expand

#include <atcoder/all>
using namespace atcoder;
// using mint = modint998244353;
// using mint = modint1000000007;
#include <bits/stdc++.h>
#define all(v) (v).begin(), (v).end()
#define rall(v) (v).rbegin(), (v).rend()
#define rep(i,n) for (int i = 0; i < (n); ++i)
#define rep2(i,k,n) for (int i = (k); i < (n); ++i)
using namespace std;
using ll = long long;
using P = pair<ll,ll>;
// using P = pair<int,int>;
using vi = vector<int>;
using vl = vector<ll>;
using vvi = vector<vector<int>>;
using vvl = vector<vector<ll>>;

// const ll INF = (ll)2e18+9;
const int INF = (int)2e9+7;
// const ll MOD = (ll)1e9+9;
template<typename T>
void chmin(T &a, T b) { a = min(a, b); }
template<typename T>
void chmax(T &a, T b) { a = max(a, b); }

template<typename T>
void print(vector<T> v) {
    int n = v.size();
    rep(i,n) {
        if (i == 0) cout << v[i];
        else cout << ' ' << v[i];
    }
    cout << endl;
}

struct Edge {
    ll from, to, cost;
    int type; // type = 0: original, type = 1: query
    int idx;
};

bool operator<(const Edge &a, const Edge &b) {
    return a.cost < b.cost;
}

void solve() {
    int n, m, q;
    cin >> n >> m >> q;

    vector<Edge> edges;
    // 元の辺の候補
    rep(i,m) {
        ll a, b, c;
        cin >> a >> b >> c;
        a--, b--;
        edges.push_back({a, b, c, 0, i});
    }

    // query による辺の候補
    rep(i, q) {
        ll u, v, w;
        cin >> u >> v >> w;
        u--, v--;
        edges.push_back({u, v, w, 1, i});
    }
    sort(all(edges));

    // minimum spanning tree を構築
    vi ans(q, 0);
    dsu uf(n);
    for (auto edge : edges) {
        auto [from, to, cost, type, idx] = edge;
        if (!uf.same(from, to)) {
            if (type == 1) { // query の辺のとき
                ans[idx] = 1;
            } else { // 元の辺のとき
                uf.merge(from, to);
            }
        }
    }

    rep(i,q) cout << (ans[i] ? "Yes" : "No") << endl;
}

int main() {
    solve();
    return 0;
}

Submission Info

Judge Result