Submission #1046057

#	Time	Username	Problem	Language	Result	Execution time	Memory
1046057		ezzzay	Closing Time (IOI23_closing)	C++17	8 / 100	80 ms	35512 KiB

closing

#include<bits/stdc++.h>
using namespace std;
#define ff first
#define ss second
#define pb push_back
#define ll long long
#include <vector>
const int MN=3e5;
vector<pair<int,ll>>v[MN];
ll distx[MN],disty[MN];
void dfsx(int a, int p){
    for(auto t:v[a]){
        if(p==t.ff)continue;
        distx[t.ff]=distx[a]+t.ss;
        dfsx(t.ff,a);
    }
}
void dfsy(int a, int p){
    for(auto t:v[a]){
        if(p==t.ff)continue;
        disty[t.ff]=disty[a]+t.ss;
        dfsy(t.ff,a);
    }
}
int max_score(int N, int X, int Y, long long K,
              std::vector<int> U, std::vector<int> V, std::vector<int> W)
{
    for(int i=0;i<N;i++){
        disty[i]=0;
        distx[i]=0;
    }
    for(int i=0;i<N;i++){
        v[i].clear();
    }
    ll k=K;
    for(int i=0;i<N-1;i++){
        int a=U[i];
        int b=V[i];
        ll c=W[i];
        v[a].pb({b,c});
        v[b].pb({a,c});
    }
    dfsx(X,-1);
    dfsy(Y,-1);
    vector<ll>vx,vy;
    for(int i=0;i<N;i++){
        vx.pb(distx[i]);
        vy.pb(disty[i]);
    }
    sort(vx.begin(),vx.end());
    sort(vy.begin(),vy.end());
    for(int i=1;i<N;i++){
        vy[i]+=vy[i-1];
    }
    for(int i=1;i<N;i++){
        //vy[i]+=vy[i-1];
    }
    ll s=0;
    ll ans=0;
    for(ll i=0;i<N;i++){
        s+=vx[i];
        if(s>k)break;
        auto it=upper_bound(vy.begin(),vy.end(),k-s+1);
        it--;
        int g=it-vy.begin();
       
        ans=max(ans,i+g+2);
        ////cout<<vy[it];
      //  return 0;
        
    }
    return ans;
}

• Subtask 0: 0 / 0
• Subtask 1: 8 / 8
• Subtask 2: 0 / 9
• Subtask 3: 0 / 12
• Subtask 4: 0 / 14
• Subtask 5: 0 / 9
• Subtask 6: 0 / 11
• Subtask 7: 0 / 10
• Subtask 8: 0 / 10
• Subtask 9: 0 / 17