Submission #721788

#TimeUsernameProblemLanguageResultExecution timeMemory
721788alvingogoBridges (APIO19_bridges)C++14
0 / 100
3077 ms3436 KiB
#include <bits/stdc++.h>
#pragma GCC optimize("Ofast")
#define AquA cin.tie(0);ios_base::sync_with_stdio(0);
#define fs first
#define sc second
#define p_q priority_queue
using namespace std;

struct DSU{
    vector<int> bo,ss;
    void init(int x){
        bo.resize(x);
        iota(bo.begin(),bo.end(),0);
        ss.resize(x,1);
    }
    int find(int x){
        return bo[x]==x?x:bo[x]=find(bo[x]);
    }
    void merge(int x,int y){
        x=find(x);
        y=find(y);
        if(x==y){
            return;
        }
        if(ss[x]<ss[y]){
            swap(x,y);
        }
        ss[x]+=ss[y];
        bo[y]=x;
    }
};
int main(){
    AquA;
    int n,m;
    cin >> n >> m;
    vector<pair<pair<int,int>,int> > e(m);
    for(int i=0;i<m;i++){
        cin >> e[i].fs.fs >> e[i].fs.sc >> e[i].sc;
        e[i].fs.fs--;
        e[i].fs.sc--;
    }
    sort(e.begin(),e.end(),[&](auto a,auto b){return a.sc>b.sc;});
    int q;
    cin >> q;
    vector<int> ans(q);
    vector<pair<int,pair<int,int> > > gg(q);
    for(int i=0;i<q;i++){
        cin >> gg[i].fs >> gg[i].fs >> gg[i].sc.sc;
        gg[i].fs--;
        gg[i].sc.fs=i;
    }
    sort(gg.begin(),gg.end(),[&](auto a,auto b){return a.sc.sc>b.sc.sc;});
    int l=0;
    DSU dsu;
    dsu.init(n);
    for(auto h:gg){
        while(l<m && e[l].sc>=h.sc.sc){
            dsu.merge(e[l].fs.fs,e[l].fs.sc);
        }
        ans[h.sc.fs]=dsu.ss[dsu.find(h.fs)];
    }
    for(int i=0;i<q;i++){
        cout << ans[i] << '\n';
    }
    return 0;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...