제출 #865986

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
865986		yeediot	Race (IOI11_race)	C++14	0 / 100	2 ms	12084 KiB

race

#include<bits/stdc++.h>
using namespace std;
#define F first
#define S second
#define all(x) x.begin(),x.end()
#define pii pair<long long,long long>
#define pb push_back
#define sz(x) (long long)(x.size())
#define chmin(x,y) x=min(x,y)
#define chmax(x,y) x=max(x,y)
#define vi vector<long long>
#define vp vector<pii>
#define vvi vector<vi>
//Don't open the standings during contests.
const long long mxn=2e5+5;
vector<pii>adj[mxn];
long long cnt[mxn];
bool vis[mxn];
int n,k;
void cnt_sz(long long v,long long pa){
    cnt[v]=1;
    for(auto [u,d]:adj[v]){
        if(vis[u] or u==pa)continue;
        cnt_sz(u,v);
        cnt[v]+=cnt[u];
    }
}
long long find(long long v,long long pa,long long tar){
    for(auto [u,d]:adj[v]){
        if(vis[u] or u==pa)continue;
        if(cnt[u]*2>tar)return find(u,v,tar);
    }
    return v;
}
long long ans=8e12;
vector<pii>dis;
void dfs(long long v,long long pa,long long l,long long len){
    if(l>k)return;
    dis.pb({l,len});
    for(auto [u,d]:adj[v]){
        if(vis[u] or u==pa)continue;
        dfs(u,v,l+1,len+d);
    }
}
long long f[mxn];
void cd(long long v){
    cnt_sz(v,v);
    v=find(v,v,cnt[v]);
    vis[v]=1;
    f[0]=0;
    vector<long long>undo;
    for(auto [u,d]:adj[v]){
        if(vis[u])continue;
        dis.clear();
        dfs(u,v,1,d);
        for(long long j=0;j<sz(dis);j++){
            chmin(ans,f[k-dis[j].S]+dis[j].F);
           // cout<<u<<' '<<dis[j].S<<' '<<dis[j].F<<' '<<f[k-dis[j].S]<<' '<<ans<<'\n';
        }
        for(long long j=0;j<sz(dis);j++){
            undo.pb(dis[j].S);
            chmin(f[dis[j].S],dis[j].F);
        }
    }
    for(auto x:undo)f[x]=8e12;
    for(auto [u,d]:adj[v]){
        if(vis[u])continue;
        cd(u);
    }
}
int best_path(int N,int K,int H[][2],int L[]){
    n=N;
    k=K;
    for(long long i=0;i<N-1;i++){
        adj[H[i][0]].pb({H[i][1],L[i]});
        adj[H[i][1]].pb({H[i][0],L[i]});
    }
    for(long long i=0;i<=K;i++){
        f[i]=8e12;
    }
    cd(0);
    return (ans==8e12?-1:ans);
}
//signed main(){
//    ios::sync_with_stdio(0);
//    cin.tie(0);cout.tie(0);
//    int n,k;
//    cin>>n>>k;
//    int h[n-1][2];
//    int l[n-1];
//    for(long long i=0;i<n-1;i++){
//        cin>>h[i][0]>>h[i][1];
//    }
//    for(long long i=0;i<n-1;i++){
//        cin>>l[i];
//    }
//    cout<<best_path(n,k,h,l);
//}
 /*
 input:
  11 12
  0 1
  0 2
  2 3
  3 4
  4 5
  0 6
  6 7
  6 8
  8 9
  8 10
  3
  4
  5
  4
  6
  3
  2
  5
  6
  7
*/

컴파일 시 표준 에러 (stderr) 메시지

race.cpp: In function 'void cnt_sz(long long int, long long int)':
race.cpp:22:14: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   22 |     for(auto [u,d]:adj[v]){
      |              ^
race.cpp: In function 'long long int find(long long int, long long int, long long int)':
race.cpp:29:14: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   29 |     for(auto [u,d]:adj[v]){
      |              ^
race.cpp: In function 'void dfs(long long int, long long int, long long int, long long int)':
race.cpp:40:14: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   40 |     for(auto [u,d]:adj[v]){
      |              ^
race.cpp: In function 'void cd(long long int)':
race.cpp:52:14: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   52 |     for(auto [u,d]:adj[v]){
      |              ^
race.cpp:66:14: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   66 |     for(auto [u,d]:adj[v]){
      |              ^

• Subtask 1: 0 / 9
• Subtask 2: 0 / 12
• Subtask 3: 0 / 22
• Subtask 4: 0 / 57