题目:赛道修建
思路:
二分答案。
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;
}