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 222222vector<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;}