P1084 [NOIP2012 提高组] 疫情控制洛谷省选/NOI-题解

此题解非洛谷题解

嗯嗯嗯，好的，看到这个题目，我们知道他非常的应景，如今，疫情防范仍需注意，还请各位戴好口罩，做好防护！！

言归正传，凡事预则立，不预则废，让我们看一下题目吧

[NOIP2012 提高组] 疫情控制

题目描述

H 国有 n 个城市，这 n 个城市用 n-1 条双向道路相互连通构成一棵树，1 号城市是首都，也是树中的根节点。

H 国的首都爆发了一种危害性极高的传染病。当局为了控制疫情，不让疫情扩散到边境城市（叶子节点所表示的城市），决定动用军队在一些城市建立检查点，使得从首都到边境城市的每一条路径上都至少有一个检查点，边境城市也可以建立检查点。但特别要注意的是，首都是不能建立检查点的。

现在，在 H 国的一些城市中已经驻扎有军队，且一个城市可以驻扎多个军队。一支军队可以在有道路连接的城市间移动，并在除首都以外的任意一个城市建立检查点，且只能在一个城市建立检查点。一支军队经过一条道路从一个城市移动到另一个城市所需要的时间等于道路的长度（单位：小时）。

请问最少需要多少个小时才能控制疫情。注意：不同的军队可以同时移动。

输入格式

第一行一个整数 $ n$，表示城市个数。

接下来的 $n-1$ 行，每行 3 个整数，$u,v,w$，每两个整数之间用一个空格隔开，表示从城市 u 到城市 v 有一条长为 $w$ 的道路。数据保证输入的是一棵树，且根节点编号为 $1$。

接下来一行一个整数 $m$，表示军队个数。

接下来一行 m 个整数，每两个整数之间用一个空格隔开，分别表示这 m个军队所驻扎的城市的编号。

输出格式

一个整数，表示控制疫情所需要的最少时间。如果无法控制疫情则输出 $-1$。

样例 #1

样例输入 #1

4 
1 2 1 
1 3 2 
3 4 3 
2 
2 2

样例输出 #1

3

提示

【输入输出样例说明】

第一支军队在 $2$ 号点设立检查点，第二支军队从 $2$ 号点移动到$ 3$ 号点设立检查点，所需时间为 $3$ 个小时。

【数据范围】

保证军队不会驻扎在首都。

• 对于 $20\%$ 的数据，$2\le n\le 10$；
• 对于 $40\%$ 的数据，$2\le n\le 50$，$0<w<10^5$；
• 对于 $60\%$ 的数据，$2\le n\le 1000$，$0<w<10^6$；
• 对于 $80\%$ 的数据，$2\le n\le 10^4$；
• 对于 $100\%$ 的数据，$2\le m\le n\le 5\times 10^4$，$0<w<10^9$。

NOIP 2012 提高组 第二天 第三题

好的，可以看到，最后一句话说明这道题也不简单，我们还得认真对待，
由题目可知这一道题是让我们输出控制疫情的最短时间，我们可以用if else来做最后的收官，整体代码使用二分，dfs思路
来看代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<cmath>
#include<algorithm>
#define ll long long
using namespace std;
const int N=6e4;
int n,m,t,tt=0,att=0,btt=0,ctt=0;
int d[N],qq[N],f[N][20];
int ver[2*N],ee[2*N],NN[2*N],ff[N];
bool ok,sta[N],asd[N];
ll ans,meme[N],ned[N],dist[N][20];
pair<ll,int> h[N];
queue<int> q;
void aa(int x,int y,int z)
{
	ver[++tt]=y,ee[tt]=z,NN[tt]=ff[x],ff[x]=tt;
}//1、输入
void bfs()
{
	q.push(1);
	d[1]=1;
	while(q.size())
	{
		int x=q.front();q.pop();
		for(int i=ff[x];i;i=NN[i])
		{
			int y=ver[i];
			if(d[y])
				continue;
			d[y]=d[x]+1;
			f[y][0]=x,dist[y][0]=ee[i];
			for(int j=1;j<=t;j++)
			{
				f[y][j]=f[f[y][j-1]][j-1];
				dist[y][j]=dist[y][j-1]+dist[f[y][j-1]][j-1];
			}
			q.push(y);
		}
	}
}//倍增预处理
bool dfs(int x)
{
	bool pson=0;
	if(sta[x])
		return 1;
	for(int i=ff[x];i;i=NN[i])
	{
		int y=ver[i];
		if(d[y]<d[x])
			continue;
		pson=1;
		if(!dfs(y))
			return 0;
	}
	if(!pson)
		return 0;
	return 1;
}//dfs寻找路径未被驻扎的叶子节点
bool check(ll lim)
{
	memset(sta,0,sizeof(sta));
	memset(meme,0,sizeof(meme));
	memset(ned,0,sizeof(ned));
	memset(h,0,sizeof(h));
	memset(asd,0,sizeof(asd));
	att=0,btt=0,ctt=0;//初始化
	for(int i=1;i<=m;i++)
	{
		ll x=qq[i],cnt=0;
		for(int j=t;j>=0;j--)
			if(f[x][j]>1 && cnt+dist[x][j]<=lim)
			{
				cnt+=dist[x][j];
				x=f[x][j];
			}
		if(f[x][0]==1 && cnt+dist[x][0]<=lim)
			h[++ctt]=make_pair(lim-cnt-dist[x][0],x);
		else
			sta[x]=1;
	}//上移军队
	for(int i=ff[1];i;i=NN[i])
		if(!dfs(ver[i]))
			asd[ver[i]]=1;//dfs寻找路径未被驻扎的叶子节点
	sort(h+1,h+ctt+1);
	for(int i=1;i<=ctt;i++)
		if(asd[h[i].sss] && h[i].first<dist[h[i].sss][0])
			asd[h[i].sss]=0;
		else
			meme[++att]=h[i].first;//对根节点的需要被驻扎的子节点进行初步处理
	for(int i=ff[1];i;i=NN[i])
		if(asd[ver[i]])
			ned[++btt]=dist[ver[i]][0];//找到仍需要被驻扎的节点并存储
	if(att<btt)
		return 0;
	sort(meme+1,meme+att+1),sort(ned+1,ned+btt+1);
	int i=1,j=1;
	while(i<=btt && j<=att)
		if(meme[j]>=ned[i])
		{
			i++,j++;
		}
		else
			j++;
	if(i>btt)
		return 1;
	return 0;//贪心匹配
}
int main()
{
	ll l=0,r=0,mid;
	cin>>n;
	t=log2(n)+1;
	for(int i=1;i<n;i++)
	{
		int x,y,z;
		scanf("%d%d%d",&x,&y,&z);
		aa(x,y,z),aa(y,x,z);
		r+=z;
	}
	bfs();倍增预处理
	cin>>m;
	for(int i=1;i<=m;i++)
		scanf("%d",&qq[i]);
	while(l<=r)
	{
		mid=(l+r)>>1;
		if(check(mid))
		{
			r=mid-1;
			ans=mid;
			ok=1;
		}
		else
			l=mid+1;
	}//二分
	if(!ok)
		cout<<-1;
	else
		cout<<ans;//输出
	return 0;
}

蟹蟹观看
来个小故事嘻嘻嘻