感觉改的跟题解差不多一样的了,二分也试过其他方法,但是还是TLE 6个点;
加O2也一样TLE 6 个点。
求写过这题的大佬看看;
代码很短很好理解的.......
#include<bits/stdc++.h>
#define LL long long
#define pa pair<int,int>
#define ls k<<1
#define rs k<<1|1
#define inf 0x3f3f3f3f
using namespace std;
const int N=50100;
const int M=50100;
const LL mod=10007;
int n,m,head[N],cnt;
int mx[N],sum;
struct Node{
int to,nex,w;
}edge[N*2];
void add(int p,int q,int w){edge[cnt].w=w,edge[cnt].to=q,edge[cnt].nex=head[p],head[p]=cnt++;}
void dfs(int p,int fa,int d){
if(sum>=m) return;
multiset<int>se;
for(int i=head[p];~i;i=edge[i].nex){
int q=edge[i].to;
if(q!=fa){
dfs(q,p,d);
int len=mx[q]+edge[i].w;
if(len>=d) sum++;
else se.insert(len);
}
}
int len=0;//儿子最大路径
while(!se.empty()){
int b=*se.begin();
se.erase(se.begin());
multiset<int>::iterator pos=lower_bound(se.begin(),se.end(),d-b);
if(pos!=se.end()) se.erase(pos),sum++;
else len=max(b,len);
}
mx[p]=len;
}
bool judge(int p){
sum=0;
dfs(1,-1,p);
if(sum>=m) return true;
return false;
}
int main(){
memset(head,-1,sizeof(head));
scanf("%d%d",&n,&m);
int ss=0;
for(int i=1;i<n;i++){
int a,b,l;scanf("%d%d%d",&a,&b,&l);
add(a,b,l),add(b,a,l),ss+=l;
}
int l=1,r=ss/m,ans=1;
while(l<=r){
int d=(l+r)>>1;
if(judge(d)) ans=d,l=d+1;
else r=d-1;
}
printf("%d\n",ans);
return 0;
}