Submission #1281031

#TimeUsernameProblemLanguageResultExecution timeMemory
1281031trinm01Race (IOI11_race)C++20
Compilation error
0 ms0 KiB
// #pragma GCC optimize("O3")
// #pragma GCC optimization("Ofast,unroll-loops")
// #pragma GCC target("avx2,bmi,bmi2,lzcnt,popcnt")
#include <bits/stdc++.h>
using namespace std;

#define int long long 
#define ll long long
#define FOR(i, l, r) for (int i = (l); i <= (r); i++)
#define FOD(i, r, l) for (int i = (r); i >= (l); i--)
#define fi first
#define se second
#define pii pair<int, int>

const ll mod = 1e9 + 7;
const int MAXN = 2e5 + 5;
const ll oo = 1e18 + 7;  
const int base = 10;

int n, k;
vector<pii> adj[MAXN];

int del[MAXN], sz[MAXN];
void count(int u, int p){
	sz[u]=1;
	for(auto [v, w]:adj[u]){
		if(v==p || del[v]) continue;
		count(v, u);
		sz[u]+=sz[v];
	}
}
int find(int u, int p, int m){
	for(auto [v, w]:adj[u]){
		if(v==p || del[v]) continue;
		if(sz[v]>m/2){
			return find(v, u, m);
		}
	}
	return u;
}
int ans=oo;
int d[MAXN], h[MAXN];
int mp[1000005];
void reset(int u, int p){
	if(d[u]<=1e6){
		mp[d[u]]=oo;
	}
	if(k-d[u]>=0){
		mp[k-d[u]]=oo;
	}
	for(auto [v, w]:adj[u]){
		if(v==p || del[v]) continue;
		h[v]=h[u]+1;
		d[v]=d[u]+w;
		reset(v, u);
	}
}
void dfs1(int u, int p){
	if(k-d[u]>=0){
		ans=min(ans, h[u]+mp[k-d[u]]);
	}
	for(auto [v, w]:adj[u]){
		if(v==p || del[v]) continue;
		h[v]=h[u]+1;
		d[v]=d[u]+w;
		dfs1(v, u);
	}
}
void dfs2(int u, int p){
	if(d[u]<=1e6)
		mp[d[u]]=min(mp[d[u]], h[u]);
	for(auto [v, w]:adj[u]){
		if(v==p || del[v]) continue;
		dfs2(v, u);
	}
}
void solve(int u){
	h[u]=d[u]=0;
	reset(u, 0);
	mp[0]=0;
	for(auto [v, w]:adj[u]){
		if(del[v]) continue;
		h[v]=h[u]+1;
		d[v]=d[u]+w;
		dfs1(v, u);
		dfs2(v, u);
	}
	
	
	reset(u, 0);
	mp[0]=0;
	FOD(i, (int)adj[u].size()-1, 0){
		auto [v, w]=adj[u][i];
		if(del[v]) continue;
		h[v]=h[u]+1;
		d[v]=d[u]+w;
		dfs1(v, u);
		dfs2(v, u);
	}
}
void centroid(int u, int p){
	count(u, 0);
	int root=find(u, 0, sz[u]);
	
	solve(root);
	
	del[root]=1;
	for(auto [v, w]:adj[root]){
		if(del[v]) continue;
		centroid(v, root);
	}
}

signed main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    
    
    // freopen("test.txt", "r", stdin);
    // freopen("o2.out", "w", stdout);

    if(fopen(".inp", "r")){
        freopen(".inp", "r", stdin);
        freopen(".out", "w", stdout);
    }

    cin >> n >> k;
    FOR(i, 1, n-1){
    	int u, v, c;
    	cin >> u >> v >> c;
    	u++;
    	v++;
    	adj[u].push_back({v, c});
    	adj[v].push_back({u, c});
    }
    
    centroid(1, 0);
    
    if(ans==oo) ans=-1;
    cout << ans;
    
    return 0;
}

Compilation message (stderr)

race.cpp: In function 'int main()':
race.cpp:124:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  124 |         freopen(".inp", "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~
race.cpp:125:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  125 |         freopen(".out", "w", stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~
/usr/bin/ld: /tmp/ccw4ueLs.o: in function `main':
grader.cpp:(.text.startup+0x0): multiple definition of `main'; /tmp/cccvWJYu.o:race.cpp:(.text.startup+0x0): first defined here
/usr/bin/ld: /tmp/ccw4ueLs.o: in function `main':
grader.cpp:(.text.startup+0x28): undefined reference to `best_path(int, int, int (*) [2], int*)'
collect2: error: ld returned 1 exit status