[bzoj1812][ioi2005]riv（树上dp）

来源：互联网发布：酒鬼蔷薇圣斗知乎编辑：程序博客网时间：2024/05/22 08:20

【题目链接】http://www.lydsy.com/JudgeOnline/problem.php?id=1812
【解题思路】转化为二叉树更方便。可以从Bytetown向后dp同时记录每个点在Bytetown方向上的最近锯木厂以满足无后效性。
【呆马】
f[i][j][k]表示第i棵子树，最近锯木厂为j，子树内可再建锯木厂为k

#include<cstdio>#include<algorithm>#include<cmath>#include<cstdlib>#include<iostream>#include<cstring>const int inf=2*(1e9);using namespace std;struct st{int to,next,v;} e[101];int n,cnt,k,x,y,i,d[101],v[101],f[101][101][51],l[101],r[101],fi[101];void add(int x,int y,int z){e[++cnt].to=y; e[cnt].next=fi[x]; e[cnt].v=z; fi[x]=cnt;}void dfs(int x){    for (int i=fi[x];i;i=e[i].next)    {        d[e[i].to]=d[x]+e[i].v;        dfs(e[i].to);    }}int dp(int i,int j,int k){    if (f[i][j][k]>=0) return f[i][j][k];    f[i][j][k]=inf;    for (int p=0;p<=k;p++)    {        int t=v[i]*(d[i]-d[j]);        if (l[i]) t+=dp(l[i],j,p);        if (r[i]) t+=dp(r[i],j,k-p);        f[i][j][k]=min(f[i][j][k],t);        if (p<k)        {            t=0;            if (l[i]) t+=dp(l[i],i,p);            if (r[i]) t+=dp(r[i],j,k-1-p);            f[i][j][k]=min(f[i][j][k],t);        }    }    return f[i][j][k];}int main(){        scanf("%d%d\n",&n,&k);        for (i=1;i<=n;i++)        {            scanf("%d%d%d\n",&v[i],&x,&y);            add(x,i,y);            if (!l[x]) l[x]=i;            else r[i]=l[x],l[x]=i;        }        dfs(0);        memset(f,-1,sizeof(f));        printf("%d",dp(0,0,k));}

0 0