hdu 5932: Backpack on Tree 贪心+dp

来源：互联网发布：数据库工程师历年真题编辑：程序博客网时间：2024/05/21 12:47

因为yls的要求就开了这道题。据说这道题目只有一个人当场过，而且yls还说是动态树合并导致我走上不归路

由于走上了动态树合并的不归路，我开始就在想直接用平衡树维护每个点的答案，还想了维护差分以后的答案什么的结果显然gg了。

然后我就考虑如何快速求出根节点的答案。认真思考了一下发现就是把它按照重量分别维护5棵平衡树（~~没错我还是陷入平衡树无法自拔~~），然后首先二分一下单位价值，直到它差不多选满，把重量比较大价值比较小的扔掉，然后剩下的把每种颜色的比较小的单位价值高的拿出来跑一下dp大概就差不多了。

yls说网上有题解然后我就去网上搜了一下就找到了o(╯□╰)o。然后发现和我做法差不多，但是窝突然发现他居然可以直接按照单位价值排序

排序？排序！

一口血喷出来。

然后把扔掉的做一个dp，拿进来的做一个dp预处理就好了。闸值设在15就能过了，只是不知道会不会被叉掉

AC代码如下：

#include<bits/stdc++.h>#define up(x,y) (x<(y)?x=(y):0)#define dn(x,y) (x>(y)?x=(y):0)#define ll long long#define N 20005using namespace std;int n,m,cnt,tot,fa[N],lim[N<<4],sx[N],pnt[N<<4],nxt[N<<4]; ll sy[N],ans[N<<4],f[N][16],g[N][16];struct node{ int x,y; }a[N],b[N];struct graph{    int fst[N];    void add(int x,int y){        pnt[++tot]=y; nxt[tot]=fst[x]; fst[x]=tot;    }    void clr(){ memset(fst,0,sizeof(int)*(n+1)); }}g1,g2;int read(){    int x=0; char cr=getchar();    while (cr<'0' || cr>'9') cr=getchar();    while (cr>='0' && cr<='9'){ x=x*10+cr-'0'; cr=getchar(); }    return x;}void dfs(int x){    int i,y; b[++cnt]=a[x];    for (i=g1.fst[x]; i; i=nxt[i]){        y=pnt[i];        if (y!=fa[x]){ fa[y]=x; dfs(y); }    }}bool cmp(node u,node v){ return (ll)u.y*v.x>(ll)v.y*u.x; }ll solve(int t){    if (sx[cnt]<t) return -1;    int l=0,r=cnt,mid;    while (l<r){        mid=l+r+1>>1;        if (sx[mid]<=t) l=mid; else r=mid-1;    }    ll tmp=-1; int i; t-=sx[l];    for (i=0; i+t<=15; i++) up(tmp,sy[l]-g[l][i]+f[l+1][i+t]);    return tmp;}int main(){    int cas=read(),ri;    for (ri=1; ri<=cas; ri++){        printf("Case #%d:\n",ri);        n=read();        int i,j,k,x,y;        tot=0; g1.clr(); g2.clr();        for (i=1; i<n; i++){            x=read(); y=read();            g1.add(x,y); g1.add(y,x);        }        for (i=1; i<=n; i++){ a[i].x=read(); a[i].y=read(); }        memset(fa,0,sizeof(fa));        cnt=0; dfs(1);        m=read();        for (i=1; i<=m; i++){            x=read(); lim[i]=read(); g2.add(x,i);        }        for (i=1; i<=n; i++) if (g2.fst[i]){            cnt=0; dfs(i);            sort(b+1,b+cnt+1,cmp);            for (j=1; j<=cnt; j++){                sx[j]=sx[j-1]+b[j].x; sy[j]=sy[j-1]+b[j].y;            }            memset(f[cnt+1],195,sizeof(f[cnt+1])); f[cnt+1][0]=0;            memset(g[0],60,sizeof(g[0])); g[0][0]=0;            for (j=cnt; j; j--){                memcpy(f[j],f[j+1],sizeof(f[j]));                for (k=15; k>=b[j].x; k--) up(f[j][k],f[j][k-b[j].x]+b[j].y);            }            for (j=1; j<=cnt; j++){                memcpy(g[j],g[j-1],sizeof(g[j]));                for (k=15; k>=b[j].x; k--) dn(g[j][k],g[j][k-b[j].x]+b[j].y);            }            for (j=g2.fst[i]; j; j=nxt[j]) ans[pnt[j]]=solve(lim[pnt[j]]);        }        for (i=1; i<=m; i++) printf("%lld\n",ans[i]);    }    return 0;}

by lych

2016.10.11

0 0