皇宫看守
Description
太平王世子事件后,陆小凤成了皇上特聘的御前一品侍卫。
皇宫以午门为起点,直到后宫嫔妃们的寝宫,呈一棵树的形状;某些宫殿间可以互相望见。大内保卫森严,三步一岗,五步一哨,每个宫殿都要有人全天候看守,在不同的宫殿安排看守所需的费用不同。
可是陆小凤手上的经费不足,无论如何也没法在每个宫殿都安置留守侍卫。
Input
帮助陆小凤布置侍卫,在看守全部宫殿的前提下,使得花费的经费最少。
Output
输入文件中数据表示一棵树,描述如下:
第1行 n,表示树中结点的数目。
第2行至第n+1行,每行描述每个宫殿结点信息,依次为:该宫殿结点标号i(0对于一个n(0 < n<=1500)个结点的树,结点标号在1到n之间,且标号不重复。
Sample Input
输出文件仅包含一个数,为所求的最少的经费。
Sample Output
6
1 30 3 2 3 4
2 16 2 5 6
3 5 0
4 4 0
5 11 0
6 5 0
解题思路
一如既往的树形DP。
#include<iostream>
#include<cstdio>
using namespace std;
long long n,a[1510],b[1510],hd[1510],tot,f[1510][4];
struct abc{
long long to,next;
}s[1000010];
void add(long long x,long long y)
{
s[++tot]=(abc){
y,hd[x]};
hd[x]=tot;
}
void dp(long long x)
{
for(long long i=hd[x];i;i=s[i].next)
{
long long y=s[i].to;
dp(y);
f[x][1]+=min(f[y][1],min(f[y][2],f[y][3]));
f[x][3]+=f[y][2];
}
f[x][1]+=a[x];
long long tmp=0;
for(long long i=hd[x];i;i=s[i].next)
{
long long y=s[i].to;
tmp+=min(f[y][1],f[y][2]);
}
f[x][2]=0x3f3f3f3f;
for(long long i=hd[x];i;i=s[i].next)
{
long long y=s[i].to;
f[x][2]=min(f[x][2],tmp-min(f[y][1],f[y][2])+f[y][1]);
}
}
int main()
{
cin>>n;
for(long long i=1;i<=n;i++)
{
long long x,k;
scanf("%lld",&x);
scanf("%lld%lld",&a[x],&k);
for(long long j=1;j<=k;j++)
{
long long y;
scanf("%lld",&y);
add(x,y);
b[y]=1;
}
}
for(long long i=1;i<=n;i++)
if(!b[i])
{
dp(i);
cout<<min(f[i][1],f[i][2])<<endl;
}
return 0;
}