Codeforces 1143C

给一棵树,每个节点有一个值,0或1,设当前节点为u ,若u的值为1 并且 所有u的儿子节点的值为1的话,u就可以删除,求删除的节点编号,按从小到大的顺序排列。若不能删除一个输出-1。
一开始想dfs,到叶子节点时输出路径,然而调了半天了没写对。这题的思路在于从反面思考,若当前节点的值为0,则该节点不能被删除,并且其父亲节点(如果有)也不能被删除。

#include<bits/stdc++.h>
using namespace std;
#define forn(i,n) for(int i = 0;i<int(n);i++)
typedef long long LL;
int n,ch[101010],root,u,x,flag;
int main(){
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	//freopen("data.in","r",stdin);
	//freopen("data.out","w",stdout);
	cin>>n;
	for(int i = 1;i<=n;i++){
		cin>>u >> x;
		if(!x) {
			ch[i] = 1;
			if(u != -1) ch[u] = 1;
		}
	}
	for(int i = 1;i<=n;i++)	if(!ch[i]){
		flag = 1; cout<<i<<' ';
	}
	if(!flag) cout<<-1;
	return 0; 
} 

猜你喜欢

转载自blog.csdn.net/winhcc/article/details/88927585
今日推荐