樹形結構+貪心思維 賽道修建(洛谷 P5021)

賽道修建

題目大意:

在一顆樹上找到m條路徑,求這m條路徑長度的最小值最大;

每條邊只能經過一次;


二分路徑長度d非常顯然;

就是怎麼求樹上路徑長度大於二分d的個數非常難;

首先知道每條邊只能用一次,所以每個點到其父親結點的邊也只能走一次,所以每個點往上走的話,只能帶一條邊,所以要儘量保證帶的這條邊最大;

當樹是兩層時(只有一個根和很多兒子結點),要湊出更多的長度大於d的路徑,可以先由小到大排序,d-a[i],然後在i+1–n中找到第一個大於d-a[i]這個差值的數,如果找的到說明這兩個點可以連接,路徑長度大於d;然後刪除這兩個點,最後維護一下這個根的點的mx[p];

一層一層的處理就行;

最好還是用multiset進行刪除,查找,非常方便,但是特別注意查找要用成員函數,要不然T到起飛;

#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=se.lower_bound(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,0,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=0,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;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章