[洛谷P4657][题解][CEOI2017]Chase

2021-12-26 15:35:56 阅读：116 来源： 互联网

一道有趣的树上 dp 题。
先考虑固定起点怎么做，容易发现以起点为根时每次选点的贡献是儿子的权值和。
那我们想得知全部的怎么办？需要进行麻烦的换根吗？
其实，我们还按照刚刚的 dp 方式，同时记录向上和向下的路径即可，设 \(f_{i,j,0/1}\) 和 \(g_{i,j,0/1}\) 表示从 \(i\) 子树内上来或从 \(i\) 下去，选了 \(j\) 个，当前选不选的最大贡献。
那么怎么统计答案呢？我们首先解决正确性：常见套路是加入当前儿子的贡献之前就统计答案，可以避免重复。然后解决效率：看起来要枚举点数上限做一个卷积一样复杂度巨大的东西，其实我们只需要记录前缀最大值即可避免此问题。
写的时候需要注意的一些细节：区分两个 dp 数组和转移的点、防止碰到不合法状态等等。

const int N=100010;
int n,m,w[N],f[N][110][2],g[N][110][2];
//f/g[i][j][0/1]:rT/xa, diVb, Vgasu6H, zx2 b'zx2
struct Edge {
  int to,nxt;
}e[N<<1];
int hd[N],cn;
il void ade(int u,int v){
  e[++cn].to=v,e[cn].nxt=hd[u],hd[u]=cn;
}
int ans;
#define Max(x,y) x=max(x,y)
void DFS(int u,int ff){
  int sum=0;
  for(int i=hd[u];i;i=e[i].nxt){
    int v=e[i].to;
    if(v!=ff)DFS(v,u),sum+=w[v];
  }
  f[u][1][1]=sum+w[ff];
  f[u][0][1]=g[u][0][1]=-INF;
  for(int i=hd[u];i;i=e[i].nxt){
    int v=e[i].to;
    if(v==ff)continue;
    int m0=0,m1=0;
    for(int j=0;j<=m;j++){
      Max(m0,max(f[u][j][0],f[u][j][1]));
      Max(m1,max(g[u][j][0],g[u][j][1]+w[ff]-w[v]));
      Max(ans,m0+max(g[v][m-j][0],g[v][m-j][1]));
      Max(ans,m1+max(f[v][m-j][0],f[v][m-j][1]));
    }
    for(int j=1;j<=m;j++){
      Max(f[u][j][0],max(f[v][j][0],f[v][j][1]));
      Max(g[u][j][0],max(g[v][j][0],g[v][j][1]));
      Max(f[u][j][1],max(f[v][j-1][0],f[v][j-1][1])\
      +sum-w[v]+w[ff]);
      Max(g[u][j][1],max(g[v][j-1][0],g[v][j-1][1])+sum);
    }
  }
}
signed main(){
  Read(n),Read(m);
  for(int i=1;i<=n;i++)Read(w[i]);
  for(int i=1,u,v;i<n;i++){
    Read(u),Read(v),ade(u,v),ade(v,u);
  }
  DFS(1,0);
  printf("%lld\n",ans);
  KafuuChino HotoKokoa
}

标签：起点,P4657,洛谷,int,题解,贡献,zx2,110,dp
来源： https://www.cnblogs.com/juruoajh/p/15733087.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

[洛谷P4657][题解][CEOI2017]Chase