Codeforces 369 C Valera and Elections

Valera and Elections

题意：现在有n个候选人， 有n-1条路， 如果选择了这个候选人， 这个候选人就会将从自己这个城市到1号城市上所有坏的路都修复一下，现在求最小的候选人数目， 如果答案有多种情况输出任何一种都可以。

输入：x y k 表示x与y之间有一条双向通道， k==1 表示这路是好的， k == 2表示这条路是坏的。

题解： 从1开始dfs搜索， 往外走， 如果遇到坏的城市就标记一下， 然后继续往外走， 如果更远的地方有一个坏的路就用更远地方的城市候选人代替这个标记的这个人。

代码：

#include<bits/stdc++.h>
using namespace std;
#define fi first
#define se second
typedef pair<int, int> pll;
const int INF = 0x3f3f3f3f;
const int N = 1e5+5;
vector<pll> E[N];
bool vis[N];
int cnt = 0;
void dfs(int x, int n, int last)
{
    for(int i = 0; i < E[x].size(); i++)
    {
        pll tmp = E[x][i];
        if(tmp.fi == last) continue;
        if(tmp.se == 2)
        {
            cnt++;
            if(vis[n]) cnt--, vis[n] = 0;
            vis[tmp.fi] = 1;
            dfs(tmp.fi, tmp.fi, x);
        }
        else dfs(tmp.fi, n, x);
    }
}
int main()
{
    int n;
    scanf("%d",&n);
    int x, y, ok;
    for(int i = 1; i < n; i++)
    {
        scanf("%d%d%d",&x,&y,&ok);
        E[x].push_back(pll(y,ok));
        E[y].push_back(pll(x,ok));
    }
    memset(vis, 0, sizeof(vis));
    dfs(1,0,0);
    printf("%d\n", cnt);
    for(int i = 2; i <= n; i++)
        if(vis[i]) printf("%d ",i);
    return 0;
}