Submission #1014960

#	Time	Username	Problem	Language	Result	Execution time	Memory
1014960		ttamx	Drivers (BOI24_drivers)	C++17	100 / 100	128 ms	15556 KiB

Main

#include<bits/stdc++.h>

using namespace std;

const int N=2e5+5;

int n,m,q;
int p[N];
vector<tuple<int,int,int>> a;
vector<tuple<int,int,int,int>> qr;
bool ans[N];

int fp(int u){
    return p[u]=u==p[u]?u:fp(p[u]);
}

int main(){
    cin.tie(nullptr)->sync_with_stdio(false);
    cin >> n >> m >> q;
    a.resize(m);
    for(auto &[w,u,v]:a)cin >> u >> v >> w;
    for(int i=0;i<q;i++){
        int u,v,w;
        cin >> u >> v >> w;
        qr.emplace_back(w,u,v,i);
    }
    sort(a.rbegin(),a.rend());
    sort(qr.begin(),qr.end());
    iota(p,p+n+1,0);
    for(auto [w,u,v,i]:qr){
        while(!a.empty()&&get<0>(a.back())<=w){
            auto [w,u,v]=a.back();
            a.pop_back();
            p[fp(u)]=fp(v);
        }
        ans[i]=fp(u)==fp(v);
    }
    for(int i=0;i<q;i++)cout << (ans[i]?"TAIP":"NE") << "\n";
}

• Subtask 1: 10 / 10
• Subtask 2: 11 / 11
• Subtask 3: 11 / 11
• Subtask 4: 23 / 23
• Subtask 5: 45 / 45