求助帖 TLE 了6个点
查看原帖
求助帖 TLE 了6个点
211144
waOooo楼主2020/4/28 18:16

感觉改的跟题解差不多一样的了,二分也试过其他方法,但是还是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;
}
2020/4/28 18:16
加载中...