Submission #948997

#	Time	Username	Problem	Language	Result	Execution time	Memory
948997		Irate	Sightseeing (NOI14_sightseeing)	C++14	0 / 25	2801 ms	165576 KiB

sightseeing

#include <bits/stdc++.h>
using namespace std;
const int mxM = 5e6 + 6;
const int mxN = 5e5 + 5;
struct Edge{
    int u, v, w;
    bool operator<(Edge &other){
        return w > other.w;
    }
};
struct DSU{
    vector<int>par;
    vector<set<int>>s;
    DSU(int n){
        par.assign(n + 1, -1);
        s.resize(n + 1);
        for(int i = 1;i <= n;++i){
            s[i].insert(i);
        }
    }
    int FindRep(int u){
        if(par[u] < 0)return u;
        return par[u] = FindRep(par[u]);
    }
    void Union(int u, int v){
        u = FindRep(u);
        v = FindRep(v);
        if(u == v)return;
        if(par[u] > par[v])swap(u, v);
        for(auto i : s[v])s[u].insert(i);
        par[u] += par[v];
        par[v] = u;
    }
};  
Edge edges[mxM];
int ans[mxN];
int main(){
    // ios_base::sync_with_stdio(0);
    // cin.tie(0);
    int n, m, q;
    cin >> n >> m >> q;
    for(int i = 0;i < m;++i){
        int u, v, w;
        cin >> u >> v >> w;
        edges[i] = {u, v, w};
    }
    DSU dsu(n);
    sort(edges, edges + m);
    for(int i = 0;i < n;++i){
        int u = edges[i].u, v = edges[i].v, w = edges[i].w;
        u = dsu.FindRep(u);
        v = dsu.FindRep(v);
        if((u == 1 && v == 1) || (u != 1 && v != 1)){
            dsu.Union(u, v);
        }
        else{
            if(u == 1){
                for(auto i : dsu.s[v])ans[i] = w;
            }
            else{
                for(auto i : dsu.s[u])ans[i] = w;
            }
            dsu.Union(u, v);
        }
    }
    while(q--){
        int node;
        cin >> node;
        cout << ans[node] << '\n';
    }
}

• Subtask 1: 0 / 4
• Subtask 2: 0 / 5
• Subtask 3: 0 / 6
• Subtask 4: 0 / 10