hdu 4003

来源:互联网 发布:2016淘宝排名靠前技巧 编辑:程序博客网 时间:2024/06/06 03:29

树形DP,f[j][i]表示有j个robot遍历i及其子树并且在此子树返回地球所需的最小值,但还有k个robot遍历子树i及其子树,再返回父亲节点m个机器人这种情况,但是可以证明这种情况一定不会是最优解,所以不予考虑。还有一个地方就是f[0][i]表示用0个robot遍历意思就是说用n个机器人遍历此子树,再返回,可知n=1,道理和前面那种情况的道理一样。在求f[i][j]是用的是分组背包的思想,刚开始多用了一个数组g[][],来实现这个过程,后来看了其他人的代码,想了一想,可以用滚动数组优化,用f[i][j]就可以实现这个过程了

#include <iostream>#include <cstdio>#include <cstring>#define LL long longusing namespace std;const int maxn=10000+10;const LL inf=0x3f3f3f3f3f3f3f3f;int e[2*maxn],head[maxn],cost[maxn*2],next[maxn*2],vis[maxn];int n,s,k,tot;LL f[maxn][11],g[maxn][11];void dp(int x){    int i=head[x];    vis[x]=1;    if(next[i]==-1&&x!=s) return;//注意x!=s这一条件因为根可能只有一个子树    int p,q,w,tv;    int j;    for(;i!=-1;i=next[i])    {        tv=e[i];        if(!vis[tv])        {            dp(tv);            for(j=k;j>=0;j--)            {                f[x][j]=f[x][j]+f[tv][0]+2*cost[i];                for(q=1;q<=j;q++) f[x][j]=min(f[x][j],f[x][j-q]+f[tv][q]+q*cost[i]);            }        }    }}int main(){    while(~scanf("%d%d%d",&n,&s,&k))    {        tot=0;        memset(head,-1,sizeof(head));        memset(vis,0,sizeof(vis));        memset(f,0,sizeof(f));        int i,u,v,c;        for(i=0;i<n-1;i++)        {            scanf("%d%d%d",&u,&v,&c);            e[tot]=v;            cost[tot]=c;            next[tot]=head[u];            head[u]=tot++;            e[tot]=u;            cost[tot]=c;            next[tot]=head[v];            head[v]=tot++;        }        dp(s);        printf("%I64d\n",f[s][k]);    }    return 0;}


原创粉丝点击