noip 2018 洛谷 P5021 赛道修建

题目：赛道修建

思路：
二分答案。
judge时，令节点1为根节点，dfs求解。
安利

代码：

#include<bits/stdc++.h>
using namespace std;

#define maxn 50000
#define read(x) scanf("%d",&x)

int cnt[maxn+5],f[maxn+5];

struct Edge{
	int u,v,w,f;
	Edge(){}
	Edge(int uu,int vv,int ww) {u=uu,v=vv,w=ww;}
	Edge(Edge oth,int ff) {u=oth.u,v=oth.v,w=oth.w,f=ff;}
	bool operator < (const Edge& oth) const {
		return f<oth.f;
	}
};

int n,m;

vector<Edge> g[maxn+5];

vector<Edge> chd[maxn+5];

void dfs(int x,int fa,int mid) {
	cnt[x]=f[x]=0;
	chd[x].clear();
	for(int i=0;i<g[x].size();i++) {
		Edge y=g[x][i];
		if(y.v==fa) continue;
		dfs(y.v,x,mid);
		f[y.v]+=y.w;
		if(f[y.v]>=mid) cnt[x]++;
		else chd[x].push_back(Edge(y,f[y.v]));
	}
	
	if(chd[x].size()==0) {
		return ;
	}
	
	sort(chd[x].begin(),chd[x].end());
	
	int L=0,R=chd[x].size()-1,mm=-1,dd=0;
	while(L<R) {
		if(chd[x][L].f+chd[x][R].f>=mid) cnt[x]++,dd++,L++,R--;
		else L++;		
	}
	
	L=0,R=chd[x].size()-1;
	while(L<=R) {
		int Mid=(L+R)/2;
		int flg=0;
		
		int l=0,r=chd[x].size()-1;
		if(l==Mid) l++;if(r==Mid) r--;
		while(l<r) {
			if(chd[x][l].f+chd[x][r].f>=mid) flg++,l++,r--;
			else l++;
			if(l==Mid) l++;if(r==Mid) r--;
		}
		
		if(flg==dd) L=Mid+1,mm=Mid;
		else R=Mid-1;
	}
	
	if(mm!=-1) f[x]=max(f[x],chd[x][mm].f);
	return ;
}

bool judge(int mid) {
	int ans=0;
	dfs(1,0,mid);
	for(int i=1;i<=n;i++) ans+=cnt[i];
	return ans>=m;
}

int main() {
	int L=0,R=0;
	
	read(n),read(m);
	for(int i=1;i<n;i++) {
		int x,y,z;
		read(x),read(y),read(z);R+=z;
		g[x].push_back(Edge(x,y,z)),g[y].push_back(Edge(y,x,z));
	}
	
	R/=m;int ans;
	while(L<=R) {
		int mid=(L+R)/2;
		if(judge(mid)) L=mid+1,ans=mid;
		else R=mid-1;
	}
	
	printf("%d",ans);
	
	return 0;
}