poj 1463 树形DP

一道简单的树形DP，可以这么想：

每个节点有可能选择也有可能不选择，而且各个子树相互独立，我用DP[i][0]表示节点i不选择所需要的最少士兵，DP[i][1]表示i节点选择所需要的最少士兵。假如j是i的儿子，因此状态转移方程可以写成如下形式：

DP[i][0]+=DP[j][1] ;

DP[i][1]+=min(DP[j][0],DP[j][1]);

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
struct node
{
    int i,next;
}S[15001];
int head[1501],dp[1501][2],N;
int add(int s,int w)
{
    S[N].i=w; 
    S[N].next=head[s];
    return N++;
}
inline int min(int a,int b)
{
    return a > b ? b : a;
}
int dfs(int i)
{
    int j,k;
    dp[i][0]=0;  dp[i][1]=1;

    for(j=head[i];j;j=S[j].next)
    {
        k=S[j].i;
        dfs(k);
        dp[i][0]+=dp[k][1];
        dp[i][1]+=min(dp[k][0],dp[k][1]);
    }

    return min(dp[i][0],dp[i][1]);
}

int main()
{
    int i,j,m,n,s,w,root;
    while(scanf("%d",&n)!=EOF)
    {
        N=1; root=-1;
        memset(head,0,sizeof(head));
        for(i=0;i<n;i++)
        {
            scanf("%d:(%d)",&s,&m);
            if(root==-1) root=s;
            for(j=0;j<m;j++)
            {
                scanf("%d",&w);
                head[s]=add(s,w);
            }
        }
        printf("%d\n",dfs(root));
    }
    return 0;
}