Luogu P1273 有线电视网 树形DP

又重构了一下。。。当然当初的题一看就看懂了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

 

posted @ 2019-04-28 13:41  LuitaryiJack  阅读(155)  评论(0编辑  收藏  举报