C - Valera and Elections CodeForces - 369C【DFS+思维】

题意：一棵树有n个节点，n-1条边，现在有些边是坏了，当你修理某个点的时候，从这个点出发到1节点的所有边都会修好，问至少需要修理几次。

思路：DFS

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+50;
struct aa
{
    int node,v;
};
vector <aa> road[maxn];
vector <int> ans; //存储答案
bool vis[maxn]; //记录该节点是否访问过

bool dfs(int node,int v) // node指当前节点,v是当前节点和父亲节点之间边的好坏，1代表好，2代表坏
{
    vis[node]=true; // 来判断是否访问过该节点
    bool rep=false; // 来判断当前节点是否需要修复
    bool flag=true; // 来判断是否到叶子

// 第一次dfs先枚举出1这个节点所能到的所有节点nextnode，如果nextnode没访问过，那么继续dfs,直到到叶子节点（flag==1），
如果叶子节点和叶子父亲边是坏的，那么叶子节点必须要修理，把叶子
节点的编号推进ans数组。如果叶子节点和父亲的边是好的，那么久不
用修，return false，回到上一个节点，如果上一个节点和父亲节
的边点是好的，就return false，否则坏的，把上一个节点推入ans
数组。然后在回溯的过程中，叶子的父辈都修理好了，然后回溯到至少有
2个儿子的节点上，再重复该操作。
    for(int i=0;i<road[node].size();i++)
    {
        int nextnode=road[node][i].node,tc=road[node][i].v;
        if(vis[nextnode]==false)
        {
            flag=false;
            if(dfs(nextnode,tc))
            {
                 rep=true;
            }
        }
    }
    if(rep) return true; // 如果当前节点，有下面的节点修复了，就return true，不需要修复该节点
    else if(flag && v==2)//如果当前节点是叶子节点 并且 叶子和父亲的边是坏的，那么推入ans数组，return true，该叶子节点到1节点都是true
    {ans.push_back(node);
        return true;

    }
    else if(v==2 && flag==false && rep==false)//如果该节点不是叶子节点，并且，这个节点的下面节点没修复过，并且这个节点和其父亲节点的边是坏的，那么推入ans数组，return true，从该节点到1的节点通通不需要修复。
    {
        ans.push_back(node);
        return true;
    }
    return false;
}

int main(void)
{
    memset(vis,false,sizeof(vis));
    int n;
    cin >> n;
    for(int i=1;i<=n-1;i++) // 输入n-1条边的情况
    {
        int a,b,v;
        scanf("%d%d%d",&a,&b,&v);
        aa temp;
        temp.node=b,temp.v=v;
        road[a].push_back(temp);//类似无向图
        temp.node=a;
        road[b].push_back(temp);
    }
    dfs(1,1);
    printf("%d\n",ans.size());
    for(int i=0;i<ans.size();i++)
        printf("%d ",ans[i]);
    puts("");
    return 0;
}