Submission #18369924

---

Source Code Expand

#include <bits/stdc++.h>
#include <atcoder/all>
#define INF 1e9
using namespace std;

#define REPR(i,n) for(int i=(n); i >= 0; --i)
#define FOR(i, m, n) for(int i = (m); i < (n); ++i)
#define REP(i, n) for(int i=0, i##_len=(n); i<i##_len; ++i)
#define ALL(a)  (a).begin(),(a).end()
#define endl "\n"

template<class T>bool chmin(T &a, const T &b) { if (b<a) { a=b; return true; } return false; }
template<class T>bool chmax(T &a, const T &b) { if (a<b) { a=b; return true; } return false; }
typedef long long ll;

struct Edge {
    int u,v,c;
};
void solve() {
    int N,M; cin >> N >> M;
    auto kruskal = [&]()->vector<set<pair<int,int>>> {
        // 辺集合
        vector<Edge> edges(M);
        REP(i,M) {
            int v1,v2,c; cin >> v1 >> v2 >> c;
            v1--; v2--;
            edges[i] = Edge{v1,v2,c};
        }

        atcoder::dsu uf(N);
        // key < val とする。
        vector<set<pair<int,int>>> g(N,set<pair<int,int>>());
        for(const auto &edge: edges) {
            // 木構造になっていたら終わり
            if(uf.size(0) == N) break;
            if(uf.same(edge.u,edge.v)) continue;
            // swap したいのでコピー
            g[edge.u].emplace(edge.v,edge.c);
            g[edge.v].emplace(edge.u,edge.c);
        }
        return g;
    };
    auto tree = kruskal();
    // rootノードを決める
    int root = -1;
    REP(i,N) if(!tree[i].empty()) {
        root = i;
        break;
    }
    assert(root != -1);
    // 良い書き方を作成していく
    auto bfs = [&]() {
        vector<int> vals(N,-1);
        vals[root] = 1;
        // pos, parent val
        queue<int> q;
        q.push(root);
        while(!q.empty()) {
            auto pos = q.front(); q.pop();
            for(const auto &child: tree[pos]) {
                auto [next,next_label] = child;
                if(vals[next] != -1) continue;
                if(next_label != vals[pos])vals[next] = next_label;
                else vals[next] = (next_label%N)+1;
                q.push(next);
            }
        }

        for(const auto &it: vals) cout << it << endl;
    };

    bfs();
}

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

Submission Info

Judge Result