又重构了一下。。。当然当初的题一看就看懂了QAQ
设f[i][j]表示以i为根的子树,有j个客户的最大收益
方程:f[u][j+k]=max(f[u][j+k],f[u][j]+f[v][k]-w(u,v)) 0<=j<=lastsize,0<=k<=v.size
#include<cstdio> #include<iostream> #include<cstring> #define R register int const int N=3010; using namespace std; inline int g() { R ret=0,fix=1; register char ch; while(!isdigit(ch=getchar())) fix=ch=='-'?-1:fix; do ret=ret*10+(ch^48); while(isdigit(ch=getchar())); return ret*fix; } int n,m,cnt; int vr[N*N],nxt[N*N],w[N*N],fir[N],a[N],f[N][N]; inline void add(int u,int v,int ww) {vr[++cnt]=v,w[cnt]=ww,nxt[cnt]=fir[u],fir[u]=cnt;} inline int dfs(int u) { if(u>n-m) {f[u][1]=a[u]; return 1;} R ret=0; for(R i=fir[u];i;i=nxt[i]) { R v=vr[i],sz=dfs(v),ww=w[i]; for(R j=ret;j>=0;--j) for(R k=sz;k>=0;--k) f[u][j+k]=max(f[u][j+k],f[u][j]+f[v][k]-ww); ret+=sz; } return ret; } signed main() { n=g(),m=g(); memset(f,0xbf,sizeof(f)); for(R u=1;u<=n-m;++u) { R sz=g(); for(R j=1,v,w;j<=sz;++j) v=g(),w=g(),add(u,v,w); } for(R i=n-m+1;i<=n;++i) a[i]=g(); for(R i=1;i<=n;++i) f[i][0]=0; dfs(1); for(R i=m;i;--i) if(f[1][i]>=0) {printf("%d\n",i); break;} }
2019.04.28