【NOIP2018模拟赛2018.10.22】咖啡供应

题目

题解

–是贪心呢
先bfs一遍
从深度最深的节点开始处理
尽量向上走最远
但是这道题卡的很死，不能直接跑dfs
要用一种标记的方法
如果一个点被儿子中的咖啡馆范围覆盖，赋一个正值，表示还可以向周围覆盖a距离
如果没有被覆盖，赋一个负值，表示离他儿子中最远的没有覆盖的点的距离为-a
显然当一个节点的值为-k，就要在这里建一个咖啡馆，并把值赋为k
但是从儿子向父亲转移的时候，如果有正有负，要判断正的是否可以覆盖负的
而且，如果一个节点的值是从儿子的正值减为0与表示负数的0意义不一样，要特判
最后再判断1号节点是否为负就行了

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
const int MAXN=1e6+5;

int n,k;
int head[MAXN],next[MAXN*2],to[MAXN*2],cnt;
int bfn[MAXN],tot;
int f[MAXN],b[MAXN];
int v[MAXN];
bool a[MAXN];
int ans;

void add(int u,int v){
	cnt++;
	next[cnt]=head[u];
	to[cnt]=v;
	head[u]=cnt;
}

void bfs(){
	queue<int>q;
	q.push(1);
	bfn[1]=++tot;
	b[tot]=1;
	while(q.size()){
		int x=q.front();
		q.pop();
		for(int i=head[x];i;i=next[i]){
			int y=to[i];
			if(bfn[y])
				continue;
			bfn[y]=++tot;
			b[tot]=y;
			f[y]=x;
			q.push(y);
		}
	}
}

int better(int a,int b){
	if(a<b)
		swap(a,b);
	if(abs(a)>=abs(b))
		return a;
	else
		return b;
}

int main(){
//	freopen("coffee.in","r",stdin);
//	freopen("coffee.out","w",stdout);
	cin>>n>>k;
	for(int i=1;i<n;i++){
		int u,v;
		scanf("%d%d",&u,&v);
		add(u,v);
		add(v,u);
	}
	bfs();
	for(int i=n;i>=1;i--){
		int x=b[i];
		if(v[x]==-k){
			ans++;
			v[x]=k;
		}
		if(v[x]==0&&a[x])
			v[f[x]]=better(v[f[x]],0);
		else{
			v[f[x]]=better(v[f[x]],v[x]-1);
			if(!v[f[x]])
				a[f[x]]=1;
		}
	}
	if(v[1]<0||(!v[1]&&!a[1]))
		ans++;
	cout<<ans;
	return 0;
}