C

来源:互联网 发布:如何制作自己的软件 编辑:程序博客网 时间:2024/05/22 10:54

题意:一棵树有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;}