【HAOI2015】树上染色

这题思路好神仙啊，首先显然是树形dp，f[i][j]表示在以i为根的子树中选j个黑点对答案的贡献（并不是当前子树最大值），dp时只考虑i与儿子连边的贡献。此时（i，son[i]）产生的收益是（设子树大小为size[i]）子树上的黑点个数（j）与子树外的黑点个数（m - j）的乘积乘上这条边的边权（w[i]）加上子树上白点的个数（size[i] - j）乘以子树外白点的个数（n - m - size[i] + j）再乘以边权，这些贡献是在加入了根节点以后才产生的新的贡献，与子树上黑白点如何分配无关。

#include<iostream>
#include<cstring>
#include<cstdio>
#define int LL
#define LL long long
#define ma(x) memset(x,0,sizeof(x))
#define MAXN 3010
using namespace std;
struct edge
{int u,v,w,nxt;#define u(x) ed[x].u#define v(x) ed[x].v#define w(x) ed[x].w#define n(x) ed[x].nxt
}ed[2000000];
int first[MAXN],num_e;
#define f(x) first[x]
int f[MAXN][MAXN],du[MAXN],root;
int n,nk,size[MAXN],tmp[MAXN];
void dfs(int x,int fa)
{size[x]=1;for(int i=f(x);i;i=n(i))if(v(i)!=fa)dfs(v(i),x);for(int i=f(x);i;i=n(i))if(v(i)!=fa){ma(tmp);for(int j=0;j<=size[x]&&j<=nk;j++)for(int k=0;k<=size[v(i)]&&k+j<=nk;k++){tmp[j+k]=max(tmp[j+k],f[x][j]+f[v(i)][k]+k*(nk-k)*w(i)+(size[v(i)]-k)*(n-nk-size[v(i)]+k)*w(i));}for(int j=0;j<=nk;j++)f[x][j]=tmp[j];size[x]+=size[v(i)];}
}
inline void add(int u,int v,int w);
signed main()
{
//  freopen("in.txt","r",stdin);
//  freopen("2.in","r",stdin);scanf("%lld%lld",&n,&nk);int a,b,c;for(int i=1;i<n;i++){scanf("%lld%lld%lld",&a,&b,&c);du[a]++;du[b]++;add(a,b,c);add(b,a,c);}for(int i=1;i<=n;i++)if(du[i]==1){root=i;break;}dfs(root,0);cout<<f[root][nk]<<endl;
}
inline void add(int u,int v,int w)
{++num_e;u(num_e)=u;v(num_e)=v;w(num_e)=w;n(num_e)=f(u);f(u)=num_e;
}

转载于:https://www.cnblogs.com/Al-Ca/p/11198618.html

【HAOI2015】树上染色相关推荐

[BZOJ4033][HAOI2015]树上染色
4033: [HAOI2015]树上染色 Time Limit: 10 Sec Memory Limit: 256 MB Submit: 2108 Solved: 901 [Submit][Sta ...
Luogu P3177 [HAOI2015] 树上染色（树上背包）
整理的算法模板合集: ACM模板点我看算法全家桶系列!!! 实际上是一个全新的精炼模板整合计划 Luogu P3177 [HAOI2015] 树上染色有一棵点数为 NNN 的树,树边有边权.给你一 ...
[BZOJ4033][HAOI2015]树上染色(树形DP)
4033: [HAOI2015]树上染色 Time Limit: 10 Sec Memory Limit: 256 MB Submit: 2437 Solved: 1034 [Submit][St ...
bzoj4033: [HAOI2015]树上染色(树形dp)
4033: [HAOI2015]树上染色 Time Limit: 10 Sec Memory Limit: 256 MB Submit: 3269 Solved: 1413 [Submit][St ...
bzoj 4033: [HAOI2015]树上染色（树形DP）
4033: [HAOI2015]树上染色 Time Limit: 10 Sec Memory Limit: 256 MB Submit: 1786 Solved: 754 [Submit][Sta ...
洛谷 P3177 [HAOI2015]树上染色
题目链接题目描述有一棵点数为 \(N\) 的树,树边有边权.给你一个在 \(0~ N\) 之内的正整数 \(K\) ,你要在这棵树中选择 \(K\)个点,将其染成黑色,并将其他的\(N-K\)个 ...
[HAOI2015]树上染色（树形dp，树形背包）
链接:https://ac.nowcoder.com/acm/problem/19996 来源:牛客网题目描述有一棵点数为N的树,树边有边权.给你一个在0~N之内的正整数K,你要在这棵树中选择K个 ...
[HAOI2015]树上染色（树形背包）
有一棵点数为 N 的树,树边有边权.给你一个在 0~ N 之内的正整数 K ,你要在这棵树中选择 K个点,将其染成黑色,并将其他的N-K个点染成白色 . 将所有点染色后,你会获得黑点两两之间的距离加 ...
bzoj4033:[HAOI2015]树上染色
传送门我一开始想的是考虑每个点的颜色设的状态就是\(f[i][j]\)表示\(i\)子树里有\(j\)个黑点的\(i\)子树的收益最大值,后来发现无法转移那么考虑答案的统计,可以对于边统计答案 ...
bzoj4033 [HAOI2015]树上染色
题目:https://www.lydsy.com/JudgeOnline/problem.php?id=4033 重要的思路:与其考虑每一个点对的贡献,不如考虑每条边的贡献(==被经过了几次)! 树形 ...

【HAOI2015】树上染色

【HAOI2015】树上染色相关推荐

最新文章

热门文章