cf C. Valera and Elections

http://codeforces.com/contest/369/problem/C

先见边，然后dfs，在回溯的过程中，如果在这个点之后有多条有问题的边，就不选这个点，如果没有而且连接这个点的边还是有问题的边，这个点就是所求的点。

#include <cstdio>
#include <cstring>
#include <algorithm>
#define maxn 200000
using namespace std;

int head[maxn],e,cnt,n;
int ans[maxn];
bool vis[maxn];
int point[maxn];
struct node
{
    int u,v,w;
    int next;
} p[maxn];

void add(int u,int v,int w)
{
    p[e].u=u;
    p[e].v=v;
    p[e].w=w;
    p[e].next=head[u];
    head[u]=e++;
}

void cls()
{
    memset(head,-1,sizeof(head));
    memset(vis,false,sizeof(vis));
    memset(ans,0,sizeof(ans));
    e=0;
    cnt=0;
}

int dfs(int u,int flag)
{
    vis[u]=1;
    for(int i=head[u]; i!=-1; i=p[i].next)
    {
        int v=p[i].v,w=p[i].w;
        if(!vis[v])
        {
            ans[u]+=dfs(v,w);
        }
    }
    if(flag==2||ans[u]>=1)
    {
        if(ans[u]==0)
            point[cnt++]=u;
        return 1;
    }
    return 0;
}

int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        cls();
        for(int i=1; i<=n-1; i++)
        {
           int a,b,c;
           scanf("%d%d%d",&a,&b,&c);
           add(a,b,c);
           add(b,a,c);
        }
        dfs(1,0);
        printf("%d
",cnt);
        for(int i=0; i<cnt; i++)
        {
           if(i==0) printf("%d",point[i]);
           else printf(" %d",point[i]);
        }
        printf("
");
    }
    return 0;
}