GYM 100247 E. Of Groups and Rights（dfs）

Description
给出一棵n个节点，以1为根的树，每个节点初始有三种状态：有权限，无权限，未设定，如果一个节点未设定，那么就继承其父亲的权限，如果其父亲未设定，其父亲就会继承其父亲的父亲的权限，以此类推，如果1节点未设定那么认为1节点无权限。定义叶子节点为用户，现在要求存最少的节点，给这些节点权限，其余节点全部未设定，要求根据这些节点以及上面的继承规则还原所有用户的权限
Input
第一行一整数n表示节点数，之后n个字符表示每个节点的权限，’+’表示有权限，’-‘表示无权限，’0’表示未设定，最后n-1行每行两个整数u,v表示树上一条边u->v(1<=n<=200000)
Output
输出一个整数m表示需要存的最少点数，之后升序输出这m个点的编号
Sample Input
10
+-+000++-+
1 2
1 5
1 9
2 6
2 8
5 4
5 10
9 3
9 7
Sample Output
3
5 8 9
Solution
首先一遍dfs让未设定的点继承权限，第二遍dfs统计任一子树中是否含有无权限的叶子节点，如果没有就接着往上回溯，否则就将该点的所有“没有无权限叶子节点”的儿子节点存到答案中表示这些点需要被保存
Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 222222
vector<int>g[maxn],ans;
int n,a[maxn];
char s[maxn];
void dfs1(int u)
{
    for(int i=0;i<g[u].size();i++)
    {
        int v=g[u][i];
        if(a[v]==0)a[v]=a[u];
        dfs1(v);
    }
}
int dfs2(int u)
{
    if(g[u].size()==0)return a[u];
    int res=1;
    for(int i=0;i<g[u].size();i++)
    {
        int v=g[u][i];
        int temp=dfs2(v);
        res=min(res,temp);
    }
    if(res==-1)
    {
        for(int i=0;i<g[u].size();i++)
        {
            int v=g[u][i];
            if(a[v]==1)ans.push_back(v);
        }
    }
    return a[u]=res;
}
int main()
{
    while(~scanf("%d",&n))
    {
        ans.clear();
        scanf("%s",s+1);
        memset(a,0,sizeof(a));
        for(int i=1;i<=n;i++)
        {
            if(s[i]=='+')a[i]=1;
            else if(s[i]=='-')a[i]=-1;
        }
        for(int i=1;i<=n;i++)g[i].clear();
        for(int i=1;i<n;i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            g[u].push_back(v);
        }
        if(a[1]==0)a[1]=-1;
        dfs1(1);
        if(dfs2(1)==1)ans.push_back(1);
        sort(ans.begin(),ans.end());
        int len=ans.size();
        printf("%d\n",len);
        for(int i=0;i<len;i++)
            printf("%d%c",ans[i],i==len-1?'\n':' ');
    }
    return 0;
}