题目链接:http://codeforces.com/contest/815/problem/C
啊,怎么xjb剪剪枝就过了啊,最开始写了个一个5000^3的算法,然后就T了啊。。。。后来约束了一下背包的上界,就过了?题解说复杂度变成了O(n^2),考虑一下满二叉树,我感觉我这样是带着log的啊?竟然分析不出来复杂度了。。。
剩下的就是一个裸的树上背包了啊?敢写敢过?
dp[now][x][1]表示,在now节点,买x个,并且now可以使用优惠券的最小花费
dp[now][x][0]表示,在now节点,买x个,并且now不可以使用优惠券的最小花费
然后XJB转移就好了2333333
反正复杂度O(能过)
代码:
#include<bits/stdc++.h>
using namespace std;
const int MAXN=5005;
const int INF=0x3f3f3f3f;
int dp[MAXN][MAXN][2];
int c[MAXN],d[MAXN],mx[MAXN];
vector<int> E[MAXN];
int tmp[MAXN][2];
void dfssize(int now)
{
mx[now]=1;
for(int _=0;_<E[now].size();_++)
{
int v=E[now][_];
dfssize(v);
mx[now]+=mx[v];
}
}
void dfs(int now)
{
int M=1,sz=1;
dp[now][0][0]=0;
dp[now][1][0]=c[now];
dp[now][1][1]=c[now]-d[now];
for(int _=0;_<E[now].size();_++)
{
int v=E[now][_];
dfs(v);
for(int i=0;i<=mx[now];i++)
{
tmp[i][0]=dp[now][i][0];
tmp[i][1]=dp[now][i][1];
}
for(int i=mx[v];i>=0;i--)
{
for(int j=min(sz,mx[now]);j>=0;j--)
{
if(dp[now][j][0]!=INF)
{
tmp[j+i][0]=min(tmp[j+i][0],dp[now][j][0]+dp[v][i][0]);
M=max(M,j+i);
}
if(dp[now][j][1]!=INF)
{
tmp[j+i][1]=min(tmp[j+i][1],dp[now][j][1]+dp[v][i][0]);
tmp[j+i][1]=min(tmp[j+i][1],dp[now][j][1]+dp[v][i][1]);
M=max(M,j+i);
}
}
}
sz=max(M,sz);
for(int i=min(sz,mx[now]);i>=0;i--)
{
if(tmp[i][0]!=INF)
dp[now][i][0]=min(dp[now][i][0],tmp[i][0]);
if(tmp[i][1]!=INF)
dp[now][i][1]=min(dp[now][i][1],tmp[i][1]);
}
}
}
int n,b;
int main()
{
//freopen("in.txt","r",stdin);
//freopen("out.txt","w",stdout);
memset(dp,0x3f,sizeof(dp));
scanf("%d%d",&n,&b);
for(int i=1;i<=n;i++)
{
scanf("%d%d",&c[i],&d[i]);
if(i!=1)
{
int u;
scanf("%d",&u);
E[u].push_back(i);
}
}
dfssize(1);
dfs(1);
int ans=0;
for(int i=1;i<=n;i++)
{
if(dp[1][i][1]<=b||dp[1][i][0]<=b)
ans=i;
}
printf("%d\n",ans);
return 0;
}