P10723 [GESP202406 七级] 黑白翻转

[GESP202406 七级] 黑白翻转

题目描述

小杨有一棵包含 $n$ 个节点的树，这棵树上的任意一个节点要么是白色，要么是黑色。小杨认为一棵树是美丽树当且仅当在删除所有白色节点之后，剩余节点仍然组成一棵树。

小杨每次操作可以选择一个白色节点将它的颜色变为黑色，他想知道自己最少要执行多少次操作可以使得这棵树变为美丽树。

输入格式

第一行包含一个正整数 $n$，代表树的节点数。

第二行包含 $n$ 个非负整数 $a_1,a_2,\dots ,a_n$，其中如果 $a_i=0$，则节点 $i$ 的颜色为白色，否则为黑色。

之后 $n-1$ 行，每行包含两个正整数 $x_i,y_i$，代表存在一条连接节点 $x_i$ 和 $y_i$ 的边。

输出格式

输出一个整数，代表最少执行的操作次数。

样例 #1

样例输入 #1

5
0 1 0 1 0
1 2
1 3
3 4
3 5

样例输出 #1

2

提示

样例解释

将节点 $1$ 和 $3$ 变为黑色即可使这棵树变为美丽树，此时删除白色节点 $5$，剩余黑色节点仍然组成一棵树。

数据范围

子任务编号	数据点占比	$n$	$a_i$	特殊条件
$1$	$30\%$	$\le 10^5$	$0\le a_i\le 1$	树的形态为一条链
$2$	$30\%$	$\le 10^5$	$0\le a_i\le 1$	只有两个节点颜色为黑色
$3$	$40\%$	$\le 10^5$	$0\le a_i\le 1$

对于全部数据，保证有 $1\le n\le 10^5$，$0\le a_i\le 1$。

思路

题目数据范围在$10^5$，所以排除普通邻接矩阵做法，但是，有一种叫vector的东西是动态大小，可以直接直接当邻接表去存，不会内存超。
实际上，想要删除所有白色点后还能成为一棵树必须使得删去的所有数字必须是叶子节点或者一次或者多次删除叶子节点后是叶子节点。
在双向存储的无向图中，叶子节点度为1。因此可以使用一个数组存度，外加一个$ans$来存储白点的数量，不影响的，直接将$ans$减去1，表示不用把这个白点转成黑点。
找到一个白色叶子结点之后，从这个节点开始广度优先搜索，搜索到的所有白色节点直接可以去掉。
注意：n==1时需要特判。

代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define ld long double
#define FOR(x,a,b,c) for(int x=a;x<=b;x+=c)
#define MFOR(x,a,b,c) for(int x=a;x>=b;x-=c)
#define MPFOR(x,a,b,c) for(int x=a;a<=b;x*=c)
const int N3=1e3+10;
const int N=1e6+10;
const long double esp=1e-8;
bool f[N];
/*/
map<ll,int> a;
queue<int> a;
stack<int> a;
/*/
int gcd(int a,int b){
	int c=a%b;
	while(a%b!=0){
			b=c;
		a=b;
		c=a%b;
	}
	return b;
}
int lcm(int x,int y){
	return (x*y)/gcd(x,y);
}
void p(int n){
	f[1]=1;
	f[0]=1;
	for(int i=2;i*i<=n;i++){
		if(f[i]) continue;
		for(int j=i*i;j<=n;j+=i){
			f[j]=1;
		}
	}
}
//前面的基本没用
vector<int> mp[100005];//存图
queue<int> q;
int n;
int typ[100005];//黑白
bool vis[100005];
int d[100005];
int ans;
int main(){
	//freopen(".in","r",stdin);
	//freopen(".out","w",stdout);
	scanf("%d",&n);
	if(n==1){
		printf("0");
		return 0;
	}
	FOR(i,1,n,1){
		scanf("%d",&typ[i]);
		if(typ[i]==0) ans++;
	}
	FOR(i,1,n-1,1){
		int x,y;
		scanf("%d %d",&x,&y);
		mp[x].push_back(y);
		mp[y].push_back(x);
		d[x]++;
		d[y]++;
	}
	FOR(i,1,n,1){
		if(d[i]==1&&typ[i]==0){//这里表示白色叶子节点
			q.push(i);
			ans-=1;//可以不用变成黑色
			vis[i]=1;
		}
	}
	while(!q.empty()){//广度优先搜索，感觉深搜也可以
		int t=q.front();
		q.pop();
		int k=mp[t].size();
		FOR(i,0,k-1,1){
			d[mp[t][i]]--;
			if(typ[mp[t][i]]==0&&d[mp[t][i]]==1&&!vis[mp[t][i]]){
				vis[mp[t][i]]=1;
				q.push(mp[t][i]);
				ans--;
			}
		}		
	}
	printf("%d",ans);
	//fclose(stdin);
	//fclose(stdout);
	return 0;
}