SGU 321 The Spy Network 树的贪心

题意：给定n（1<=n<=200000）个节点的树，每条边的属性是0（almost protected）或者1（protected），虽然是有向边但是题目保证每条边都有到达

         根节点发的路径，现在想问最小改变那些边可以使得所有的点到根节点的路径上1的个数>=一半。

题解：首先dfs一遍得到每一个节点还需要1的个数，维护每个节点的及其子节点最多需要1的个数，然后再dfs如果当前节点及其子节点还需要那么当前的边

         如果是0变成1记录下来，这里体现的是贪心的过程，因为改变边的状态的深度越小影响越大。

Sure原创，转载请注明出处。

#pragma comment (linker , "/STACK:1024000000,1024000000")
#include <iostream>
#include <cstdio>
#include <memory.h>
#define MAX(a , b) ((a) > (b) ? (a) : (b))
using namespace std;
const int maxn = 200002;
struct node
{
    int id,v,w;
    int next;
}edge[maxn];
int head[maxn],need[maxn],ans[maxn];
int n,idx,top;

void init()
{
    memset(head,-1,sizeof(head));
    idx = top = 0;
    return;
}

void addedge(int ii,int u,int v,int w)
{
    edge[idx].id = ii;
    edge[idx].v = v;
    edge[idx].w = w;
    edge[idx].next = head[u];
    head[u] = idx++;
    return;
}

void read()
{
    char str[10];
    int u,v;
    for(int i=1;i<n;i++)
    {
        scanf("%d %d %s",&v,&u,str);
        if(str[0] == 'a')
        {
            scanf("%s",str);
            addedge(i,u,v,0);
        }
        else addedge(i,u,v,1);
    }
    return;
}

void make(int st,int dep,int have)
{
    need[st] = (dep + 1) / 2 - have;
    for(int i=head[st];i != -1;i=edge[i].next)
    {
        make(edge[i].v , dep + 1 , have + edge[i].w);
        need[st] = MAX(need[st] , need[edge[i].v]);
    }
    return;
}

void dfs(int st,int num)
{
    for(int i=head[st];i != -1;i=edge[i].next)
    {
        if(num < need[edge[i].v])
        {
            if(edge[i].w == 0)
            {
                ans[top++] = edge[i].id;
                dfs(edge[i].v , num+1);
            }
            else dfs(edge[i].v , num);
        }
    }
    return;
}

void solve()
{
    make(1,0,0);
    dfs(1,0);
    printf("%d\n",top);
    for(int i=0;i<top;i++)
    {
        if(i) printf(" ");
        printf("%d",ans[i]);
    }
    puts("");
    return;
}

int main()
{
    while(~scanf("%d",&n))
    {
        init();
        read();
        solve();
    }
    return 0;
}