bzoj 4033: [HAOI2015]T1(树形DP)

4033: [HAOI2015]T1

Time Limit: 10 Sec  Memory Limit: 256 MB

Submit: 819  Solved: 375

[Submit][Status][Discuss]

Description

有一棵点数为 N 的树,树边有边权。给你一个在 0~ N 之内的正整

数 K ,你要在这棵树中选择 K个点,将其染成黑色,并将其他 的

N-K个点染成白色 。 将所有点染色后,你会获得黑点两两之间的距

离加上白点两两之间的距离的和的受益。问受益最大值是多少。

Input

第一行包含两个整数 N, K 。

接下来 N-1 行每行三个正整数 fr, to, dis , 表示该树中存在一条长度

为 dis 的边 (fr, to) 。输入保证所有点之间是联通的。

Output

输出一个正整数,表示收益的最大值。

Sample Input

3 1

1 2 1

1 3 2

Sample Output

3

HINT

对于 100% 的数据, 0<=K<=N <=2000

Source

鸣谢bhiaibogf提供

[Submit][Status][Discuss]
题解:树形dp

f[i][j]表示从i的子树中选j个黑点的最大收益值。

对于一条边c[i],他的贡献就是(他子树内的黑点个数×子树外的黑点个数+子树内白点的个数×子树外白点的个数)×c[i],因为是黑点和白点之间两两连边,要到达这棵子树,那么就必须经过这条边,所以这条边被经过的次数是一定的。

那么如果转移呢?

f[x][j]=max(f[x][j],f[x][j-l]+f[v[i]][l](当前子树的贡献)+当前边的贡献)

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define LL long long
using namespace std;
LL f[2003][2003];
int tot,next[4003],point[4003],size[2003],v[4003],c[4003];
int n,k;
void add(int x,int y,int z)
{
	tot++; next[tot]=point[x]; point[x]=tot; v[tot]=y; c[tot]=z;
	tot++; next[tot]=point[y]; point[y]=tot; v[tot]=x; c[tot]=z;
}
void dfs(int x,int fa)
{
	f[x][0]=f[x][1]=0;
	size[x]=1;
	for (int i=point[x];i;i=next[i])
	 if (v[i]!=fa)
	 {
	 	dfs(v[i],x);
	 	size[x]+=size[v[i]];
	 	for (int j=min(k,size[x]);j>=0;j--)
	 	 for (int l=0;l<=min(j,size[v[i]]);l++)
	 	  {
	 	  	LL ans1=(LL)(l*(k-l))*c[i];
	 	  	LL ans2=(LL)(size[v[i]]-l)*(n-k-(size[v[i]]-l))*c[i];
	 	  	f[x][j]=max(f[x][j],f[x][j-l]+f[v[i]][l]+ans1+ans2);
	 	  }
	 }
}
int main()
{
	scanf("%d%d",&n,&k);
	memset(f,128,sizeof(f));
	for (int i=1;i<n;i++)
	 {
	 	int x,y,z;
	 	scanf("%d%d%d",&x,&y,&z);
	 	add(x,y,z);
	 }
	dfs(1,0);
	printf("%lld\n",f[1][k]);
} 
时间: 2024-10-07 04:01:22

bzoj 4033: [HAOI2015]T1(树形DP)的相关文章

BZOJ 4033 HAOI2015 T1 树形DP

题目大意:给定一棵树,你需要把其中的k个点染成黑色,使得黑色点两两之间的距离和+白色点两两之间的距离和最大,求最大值 题解戳这里 Orz ydcydc 看来我对于非线性的树形DP还是做得太少了QwQ #include <cstdio> #include <cstring> #include <iostream> #include <algorithm> #define M 2020 using namespace std; struct edge{ int

[BZOJ 4033] [HAOI2015] T1 【树形DP】

题目链接:BZOJ - 4033 题目分析 使用树形DP,用 f[i][j] 表示在以 i 为根的子树,有 j 个黑点的最大权值. 这个权值指的是,这个子树内部的点对间距离的贡献,以及 i 和 Father[i] 之间的边对答案的贡献(比如这条边对黑点对距离和的贡献就是子树内部的黑点数 * 子树外部的黑点数 * 这条边的权值). 然后DFS来求,枚举 i 的每个儿子 j,现在的 f[i][] 是包含了 [1, j-1] 子树,然后两重循环枚举范围是 [1, j - 1] 的子树总 Size 和

bzoj 4033: [HAOI2015]树上染色

4033: [HAOI2015]树上染色 Time Limit: 10 Sec  Memory Limit: 256 MBSubmit: 1886  Solved: 805 Description 有一棵点数为N的树,树边有边权.给你一个在0~N之内的正整数K,你要在这棵树中选择K个点,将其染成黑色,并 将其他的N-K个点染成白色.将所有点染色后,你会获得黑点两两之间的距离加上白点两两之间距离的和的收益. 问收益最大值是多少. Input 第一行两个整数N,K. 接下来N-1行每行三个正整数fr

BZOJ 1017--魔兽地图(树形DP&amp;完全背包)

1017: [JSOI2008]魔兽地图DotR Time Limit: 30 Sec  Memory Limit: 162 MBSubmit: 2674  Solved: 1041[Submit][Status][Discuss] Description DotR (Defense of the Robots) Allstars是一个风靡全球的魔兽地图,他的规则简单与同样流行的地图DotA (Defense of the Ancients) Allstars.DotR里面的英雄只有一个属性——

BZOJ 1596--电话网络(树形DP)

1596: [Usaco2008 Jan]电话网络 Time Limit: 10 Sec  Memory Limit: 162 MBSubmit: 1139  Solved: 534[Submit][Status][Discuss] Description Farmer John决定为他的所有奶牛都配备手机,以此鼓励她们互相交流.不过,为此FJ必须在奶牛们居住的N(1 <= N <= 10,000)块草地中选一些建上无线电通讯塔,来保证任意两块草地间都存在手机信号.所有的N块草地按1..N 顺

bzoj 4033 树上染色 - 树形动态规划

有一棵点数为N的树,树边有边权.给你一个在0~N之内的正整数K,你要在这棵树中选择K个点,将其染成黑 色,并将其他的N-K个点染成白色.将所有点染色后,你会获得黑点两两之间的距离加上白点两两之间距离的和的 收益.问收益最大值是多少. Input 第一行两个整数N,K. 接下来N-1行每行三个正整数fr,to,dis,表示该树中存在一条长度为dis的边(fr,to). 输入保证所有点之间是联通的. N<=2000,0<=K<=N Output 输出一个正整数,表示收益的最大值. Sampl

BZOJ 1040: [ZJOI2008]骑士( 树形dp )

这是一个森林中, 每棵树上都有一个环...每棵树单独处理, 找出环上任意一条边断开, 限制一下这条边两端点的情况, 然后就可以树dp了.. ------------------------------------------------------------------------ #include<cstdio> #include<algorithm> #include<cstring> #include<cctype> using namespace

BZOJ 3829 Poi2014 FarmCraft 树形DP+贪心

题目大意:给定一棵树,从1号节点出发对树进行欧拉遍历,每到达一个点这个点就开始装MC,每个点装MC的时间不同,最后回到1号节点装MC,求所有人都能联机的最少时间 令f[x]为对第x个节点进行欧拉遍历的时间,g[x]为对第x个节点进行欧拉遍历并完成所有节点的装机的最小时间 那么在每个节点以什么顺序遍历每棵子树呢? 我们发现装机多出来的时间 即g[x]-f[x]可以用来遍历其它子树 那么显然要从g[x]-f[x]大的子树开始遍历 因此对每个节点的子树按照g[x]-f[x]递减排个序即可 时间复杂度O

BZOJ 3037 创世纪 树形DP

题目大意:给定一张有向图,每个点有且仅有一条出边,要求若一个点x扔下去,至少存在一个保留的点y,y的出边指向x,求最多扔下去多少个点 首先原题的意思就是支配关系 我们反向考虑 求最少保留的点 要求一个点若扔出去 则必须存在一个保留的点指向它 于是这就是最小支配集 不过由于是有向图 所以一个点要么选择 要么被子节点支配 所以就只剩下2个状态了 设f[x]为以x为根的子树选择x的最小支配集 g[x]为不选择x的最小支配集 然后由于是基环树林 所以我们选择一个环上的点 拆掉它的出边 设这个点为x 出边