您的位置:首页 > 其它

hdu 5932: Backpack on Tree 贪心+dp

2016-10-11 18:05 316 查看
        因为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 20005
using namespace std;

int n,m,cnt,tot,fa
,lim[N<<4],sx
,pnt[N<<4],nxt[N<<4]; ll sy
,ans[N<<4],f
[16],g
[16];
struct node{ int x,y; }a
,b
;
struct graph{
int fst
;
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
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: