Travel Plan

题目描述

DP

按dfs序处理出前缀和后缀背包数组。
设的是f[v]表示纪念品价格的和至少为v所需的最小支出。
询问时two pointer即可。
关于空间可以用unsigned int。
然后我被卡常了？？？

#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fd(i,a,b) for(i=a;i>=b;i--)
#define min(a,b) (a<b?a:b)
#define max(a,b) (a>b?a:b)
using namespace std;
typedef long long ll;
const int maxn=1000+10,maxv=50000+10;
const int inf=2000000050;
int pre[maxn][maxv],suf[maxn][maxv],t,r;
int h[maxn],go[maxn*2],next[maxn*2],dfn[maxn],nfd[maxn],size[maxn],v[maxn],c[maxn];
int sta[80];
int i,j,k,l,n,m,tot,top,euler,ans,mx;
int read(){
    ll x=0,f=1;
    char ch=getchar();
    while (ch<'0'||ch>'9'){
        if (ch=='-') f=-1;
        ch=getchar();
    }
    while (ch>='0'&&ch<='9'){
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x*f;
}
void add(int x,int y){
    go[++tot]=y;
    next[tot]=h[x];
    h[x]=tot;
}
void dfs(int x,int y){
    dfn[x]=++top;
    nfd[top]=x;
    int t=h[x];
    size[x]=1;
    while (t){
        if (go[t]!=y){
            dfs(go[t],x);
            size[x]+=size[go[t]];
        }
        t=next[t];
    }
}
void write(int x){
    if (!x){
        putchar('0');
        putchar('\n');
        return;
    }
    top=0;
    while (x){
        sta[++top]=x%10;
        x/=10;
    }
    while (top) putchar('0'+sta[top--]);
    putchar('\n');
}
int main(){
    freopen("plan.in","r",stdin);freopen("plan.out","w",stdout);
    n=read();
    fo(i,1,n-1){
        j=read();k=read();
        add(j,k);add(k,j);
    }
    fo(i,1,n) v[i]=read(),mx+=v[i],c[i]=read();
    dfs(1,0);
    fo(i,1,mx) pre[0][i]=inf;
    fo(i,1,n){
        fo(j,0,mx) pre[i][j]=pre[i-1][j];
        t=nfd[i];
        k=v[t];r=c[t];
        fd(j,mx,k) pre[i][j]=min(pre[i][j],pre[i][j-k]+r);
        fd(j,mx-1,0) pre[i][j]=min(pre[i][j],pre[i][j+1]);
    }
    fo(i,1,mx) suf[n+1][i]=inf;
    fd(i,n,1){
        fo(j,0,mx) suf[i][j]=suf[i+1][j];
        t=nfd[i];
        k=v[t];r=c[t];
        fd(j,mx,k) suf[i][j]=min(suf[i][j],suf[i][j-k]+r);
        fd(j,mx-1,0) suf[i][j]=min(suf[i][j],suf[i][j+1]);
    }
    m=read();
    while (m--){
        r=read();t=read();
        l=dfn[r]-1;r=dfn[r]+size[r];
        ans=0;
        k=0;
        fd(j,mx,0){
            //if (pre[l][j]>t) continue;
            while (k<mx&&(ll)pre[l][j]+(ll)suf[r][k+1]<=t) k++;
            if (pre[l][j]+suf[r][k]<=t){
                ans=max(ans,j+k);
            }
        }
        write(ans);
    }
}