皇宫看守

http://www.elijahqi.win/archives/1180
题目背景

太平王世子事件后，陆小凤成了皇上特聘的御前一品侍卫。

题目描述

皇宫以午门为起点，直到后宫嫔妃们的寝宫，呈一棵树的形状；某些宫殿间可以互相望见。大内保卫森严，三步一岗，五步一哨，每个宫殿都要有人全天候看守，在不同的宫殿安排看守所需的费用不同。

可是陆小凤手上的经费不足，无论如何也没法在每个宫殿都安置留守侍卫。

帮助陆小凤布置侍卫，在看守全部宫殿的前提下，使得花费的经费最少。

输入输出格式

输入格式：

输入文件中数据表示一棵树，描述如下：

第1行 n，表示树中结点的数目。

第2行至第n+1行，每行描述每个宫殿结点信息，依次为：该宫殿结点标号i（0< i<=n），在该宫殿安置侍卫所需的经费k，该边的儿子数m，接下来m个数，分别是这个节点的m个儿子的标号r1，r2，…，rm。

对于一个n个结点的树，结点标号在1到n之间，且标号不重复。

输出格式：

输出文件仅包含一个数，为所求的最少的经费。

输入输出样例

输入样例#1：

6
1 30 3 2 3 4
2 16 2 5 6
3 5 0
4 4 0
5 11 0
6 5 0
输出样例#1：

25
说明

0 < n <= 1500

能被看到的意思是有边相连hh，给定了树形。dp[0/1/2][i]表示以i为跟的子树全覆盖所需的最小花费，0表示子树的根被儿子看，1表示子树的根被自己看，2表示子树的根被父亲看。
则转移为dp[0][i]=∑min{dp[0][x],dp[1][x]}+dp[1][y],x为i的儿子，x!=y,y为选定的儿子（枚举y）
dp[1][i]=∑min{dp[2][x],dp[1][x]}+a[i],x为i的儿子,a[i]为i的权值
dp[2][i]=∑min{dp[0][x],dp[1][x]},x为i的儿子
最后答案就是min{dp[rt][0],dp[rt][1]}

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#define inf 0x7f7f7f7f
#define N 1650
using namespace std;
inline char gc(){
    static char now[1<<16],*S,*T;
    if (S==T){T=(S=now)+fread(now,1,1<<16,stdin);if (S==T) return EOF;}
    return *S++;
}
inline int read(){
    int x=0;char ch=gc();
    while (ch<'0'||ch>'9') ch=gc();
    while (ch<='9'&&ch>='0'){x=x*10+ch-'0';ch=gc();}
    return x;
}
int root,n,cost[N],fa[N],d[N][3];vector<int> son[N];
void f(int x,int s){
    if(s==1||s==2){
        int ans=0;if (s==1) ans=cost[x];
        for (int i=0;i<son[x].size();++i){
            int son1=son[x][i];
            if (d[son1][0]==inf) f(son1,0);
            if (d[son1][1]==inf) f(son1,1);
            if (s!=2) {
                if (d[son1][2]==inf) f(son1,2);
                ans+=min(min(d[son1][0],d[son1][1]),d[son1][2]);
            }else ans+=min(d[son1][0],d[son1][1]);
        }
        d[x][s]=ans;
        return;
    }
    for (int i=0;i<son[x].size();++i){
        int son1=son[x][i];if (d[son1][1]==inf) f(son1,1);
        int ans=d[son1][1];
        for (int j=0;j<son[x].size();++j){
            int son2=son[x][j];if (son2==son1) continue;
            if (d[son2][0]==inf) f(son2,0);
            if (d[son2][1]==inf) f(son2,1);
            ans+=min(d[son2][0],d[son2][1]);
        }
        d[x][s]=min(d[x][s],ans);
    }


}
int main(){
    freopen("guard.in","r",stdin);
    n=read();
    for (int i=1;i<=n;++i){
        int x=read();cost[x]=read();int tmp1=0;
        int tmp=read();for (int j=1;j<=tmp;++j) tmp1=read(),son[x].push_back(tmp1),fa[tmp1]=x;
    }
    for (int i=1;i<=n;++i)
        if (fa[i]==0) {root=i;break;} 
    memset(d,0x7f,sizeof(d));
    f(root,1);f(root,0);
    printf("%d",min(d[root][1],d[root][0]));
    return 0;
}