30 Pts求助dalao(玄关)
查看原帖
30 Pts求助dalao(玄关)
867875
niveKmAI楼主2024/8/14 10:27
#include<bits/stdc++.h>
#define int long long
using namespace std;
int n,m,k,s,t;
int f[2][200005];
struct node {
	int v,w;
};
vector<node>g[200005];
vector<int>a;
int vis[200005];
void dfs(int x,int fa,int dis[]) {
	for(int i=0; i<g[x].size(); i++) {
		int v=g[x][i].v;
		if(v==fa) continue;
		int w=g[x][i].w;
		dis[v]=dis[x]+w;
		dfs(v,x,dis);
	}
}
bool check(int ans){
	if(f[0][t]<=ans){
		return 1;
	}
	int cnt=0;
//	int mx=0;
	for(int i=1; i<=n; i++) {
		if(f[0][i]>ans) continue;
		if(f[0][i]+k<=ans) {
			cnt++;
		}
		if(f[0][i]+f[1][i]<=ans) return 1; 
		int now=ans-f[0][i]-k;
		int pos=upper_bound(a.begin(),a.end(),now)-a.begin();
		for(int j=0; j<g[i].size(); j++){
			int v=g[i][j].v;
			if(f[1][v]<=a[pos-1]){
				cnt--;
			}
		}
		cnt+=pos-1;
	} 
	if(cnt>m){
		return 1;
	} 
	else{
		return 0;
	}
}
signed main() {
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);

	cin>>n>>m>>k>>s>>t;
	for(int i=1; i<n; i++) {
		int u,v,w;
		cin>>u>>v>>w;
		g[u].push_back({v,w});
		g[v].push_back({u,w});
	}
	dfs(s,0,f[0]);
	dfs(t,0,f[1]);
//	     
	for(int i=1; i<=n; i++) {
		a.emplace_back(f[1][i]);
	}
	sort(a.begin(),a.end());
	int ans=1e18;
	int l=1;
	int r=1e18;
	while(l<=r) {
		int mid=(l+r)/2;
		if(check(mid)) {
			r=mid-1;
			ans=min(mid,ans);
			//	cout<<ans<<"\n";
		} else {
			l=mid+1;
		}
	}
	cout<<min((long long)1e9,ans);
	return 0;
}
2024/8/14 10:27
加载中...