「CF375E」Red and Black Tree「树形DP」
题意
给定一个结点颜色红或黑的树,问最少进行多少次交换黑、红结点使得每个红结点离最近的黑结点距离\(\leq x\)。
\(1\leq n \leq 500, 1 \leq x \leq 10^9\)
题解
不是红黑树
据说可有单纯形做,这里讲的还是树形dp的做法
考虑交换比较费劲,直接在\(n\)个结点里选\(m\)个(\(m\)是黑结点个数),如果原来是红结点产生\(1\)的代价
\(dp[u][k][a]\)表示\(u\)子树内选了\(k\)个黑点,\(u\)选择的黑结点是\(a\)(这里为了方便,不定义为最近的黑结点)的最小代价(若\(a\)在子树外,不考虑\(a\)的代价)。这个状态相当于树形背包加一维。
考虑转移:新加入了子树\(v\)。原信息\(dp[u][su][a]\),子树信息\(dp[v][su][b]\)
- \(a=b\):加起来转移到\(dp[u][su + sv][a]\)上。
- \(b\)在\(v\)子树中:此时a一定不在\(v\)子树中。如果在子树中只要考虑\(a = b\),上面考虑过了。加起来转移到\(dp[u][su + sv][a]\)。(实现时直接对于每个子树外的\(a\)加上最小dp值就好)
- \(b\)在\(v\)子树外:这时候\(a\not =b\)一定是不优的。\(a\)在\(v\)子树内,把b改\(a\)不会更差;a也不在\(v\)子树中,让\(a=b\)或\(b=a\)一定有一种不会更差。所以不转移。
注意使用short类型防止MLE。
#include <algorithm>
#include <cstdio>
using namespace std;
const int N = 505;
struct Edge {
short v, nxt; int w;
} e[N * 2];
short hd[N], sz[N], pos[N], w[N], n, m, ec;
short dl[N], dr[N], idx, dp[N][N][N], t[N][N];
int x, dis[N][N];
void clr() {
fill(hd + 1, hd + n + 1, -1); ec = 0;
}
void add(short u, short v, int w) {
e[ec] = (Edge) {v, hd[u], w}; hd[u] = ec ++;
}
void dfs0(int *d, int u, int fa = 0) {
if(!fa) d[u] = 0;
for(int i = hd[u]; ~ i; i = e[i].nxt) {
Edge &p = e[i];
if(p.v != fa) {
d[p.v] = min(x + 1, d[u] + p.w);
dfs0(d, p.v, u);
}
}
}
void dfs1(int u, int fa = 0) {
dl[u] = ++ idx; pos[idx] = u;
for(int i = hd[u]; ~ i; i = e[i].nxt) {
int v = e[i].v;
if(v != fa) dfs1(v, u);
}
dr[u] = idx;
}
void dfs(int u, int fa = 0) {
for(int v = 1; v <= n; v ++) {
if(dis[u][v] <= x && u != v) {
dp[u][0][v] = 0;
}
}
dp[u][1][u] = w[u] ^ 1; sz[u] = 1;
for(short i = hd[u]; ~ i; i = e[i].nxt) {
int v = e[i].v;
if(v == fa) continue ;
dfs(v, u);
for(short s = min(sz[v] + sz[u], (int) m); s >= 0; s --)
for(short a = 1; a <= n; a ++)
t[s][a] = n + 1;
for(short su = min(sz[u], m); su >= 0; su --) {
for(short sv = min(sz[v], m); sv >= 0; sv --) {
short mv = n + 1;
for(int a = dl[v]; a <= dr[v]; a ++) mv = min(mv, dp[v][sv][pos[a]]);
for(int a = 1; a <= n; a ++) {
t[su + sv][a] = min((int) t[su + sv][a], dp[u][su][a] + dp[v][sv][a]);
if(a < dl[v] || a > dr[v]) {
t[su + sv][pos[a]] = min((int) t[su + sv][pos[a]], dp[u][su][pos[a]] + mv);
}
}
}
}
sz[u] += sz[v];
for(short s = min(sz[u], m); s >= 0; s --)
for(short a = 1; a <= n; a ++)
dp[u][s][a] = t[s][a];
}
}
int main() {
scanf("%hd%d", &n, &x); clr();
for(int i = 1; i <= n; i ++) {
scanf("%hd", w + i);
m += w[i];
}
for(int u, v, w, i = 1; i < n; i ++) {
scanf("%d%d%d", &u, &v, &w);
add(u, v, w); add(v, u, w);
}
for(int i = 1; i <= n; i ++) dfs0(dis[i], i);
for(int i = 0; i <= n; i ++)
for(int j = 0; j <= n; j ++)
for(int k = 0; k <= n; k ++)
dp[i][j][k] = n + 1;
dfs1(1); dfs(1);
short ans = n + 1;
for(int i = 1; i <= n; i ++)
ans = min(ans, dp[1][m][i]);
printf("%hd\n", ans > n ? -1 : ans);
return 0;
}
「CF375E」Red and Black Tree「树形DP」的更多相关文章
- 「树形结构 / 树形DP」总结
Codeforces 686 D. Kay and Snowflake 要求$O(n)$求出以每个节点为根的重心. 考虑对于一个根节点$u$,其重心一定在[各个子树的重心到$u$]这条链上.这样就能够 ...
- HDU5834 Magic boy Bi Luo with his excited tree(树形DP)
题目 Source http://acm.hdu.edu.cn/showproblem.php?pid=5834 Description Bi Luo is a magic boy, he also ...
- BZOJ-3227 红黑树(tree) 树形DP
个人认为比较好的(高端)树形DP,也有可能是人傻 3227: [Sdoi2008]红黑树(tree) Time Limit: 10 Sec Memory Limit: 128 MB Submit: 1 ...
- Codeforces Round #263 (Div. 2) D. Appleman and Tree(树形DP)
题目链接 D. Appleman and Tree time limit per test :2 seconds memory limit per test: 256 megabytes input ...
- 2017 Multi-University Training Contest - Team 1 1003&&HDU 6035 Colorful Tree【树形dp】
Colorful Tree Time Limit: 6000/3000 MS (Java/Others) Memory Limit: 131072/131072 K (Java/Others)T ...
- 「树形DP」洛谷P2607 [ZJOI2008]骑士
P2607 [ZJOI2008]骑士 题面: 题目描述 Z 国的骑士团是一个很有势力的组织,帮会中汇聚了来自各地的精英.他们劫富济贫,惩恶扬善,受到社会各界的赞扬. 最近发生了一件可怕的事情,邪恶的 ...
- 「HAOI2015树上染色」「树形DP」
其实我还不大会树形DP 此题就当练手叭,缕一下思路就好 题目链接 BZOJ4033 题目大意就是给一棵树,对一部分点染成黑色,剩下的为白色,问所有同色点距离之和....... 简明扼要的题意,然额我不 ...
- NFLSOJ #10317. -「2020联考北附2」三千世界(找等价表达+树形 dp)
题面传送门 出题人可能原本感觉没啥难度的 T2 竟然变成了防 AK 题,奇迹奇迹( 首先带着这个 \(\max\) 肯定不太好处理,考虑找出 \(f(S)\) 的等价表达.我们考虑以 \(1\) 为根 ...
- HDU 6035---Colorful Tree(树形DP)
题目链接 Problem Description There is a tree with n nodes, each of which has a type of color represented ...
随机推荐
- js取得对象的类名constructor.name
constructor 属性返回对创建此对象的数组函数的引用. 语法 object.constructor 例子 new Array().constructor//ƒ Array() { [nativ ...
- VS.NET(C#)--1.4项目与解决方案
项目与解决方案 项目 除创建网站,VS2005可创建项目.然后把项目放入解决方案中.VS2005可编译很多类型项目,分别是: 1.Windows应用程序 --在用戶计算机上运行的客户端应用程序,可显示 ...
- 笔记 - C#从头开始构建编译器 - 1
视频与PR:https://github.com/terrajobst/minsk/blob/master/docs/episode-01.md 作者是 Immo Landwerth(https:// ...
- VBA循环(十一)
当需要多次执行一段代码时,就可以使用循环语句. 一般来说,语句是按顺序执行的:函数中的第一个语句首先执行,然后是第二个,依此类推. 编程语言提供了各种控制结构,允许更复杂的执行路径. 循环语句允许多次 ...
- vue动态绘制四分之三圆环
参照网上的一个案例“参照的为绘制的是一个动态的圆环”,现在我的需求是改编成四分之三的圆环实现效果: 样式展示 canvas绘图基本操作设置就可以参考源代码链接:原文:https://blog.csdn ...
- laravel withCount 统计关联数量
roleModel定义关联 hasmany public function users(){ return $this->hasMany('App\Models\Users', 'role_i ...
- python 元类理解
原文来自:https://segmentfault.com/a/1190000011447445 学懂元类,你只需要知道两句话: 道生一,一生二,二生三,三生万物 我是谁?我从哪来里?我要到哪里去? ...
- Java中的经典算法之快速排序(Quick Sort)
Java中的经典算法之快速排序(Quick Sort) 快速排序的思想 基本思想是:通过一趟排序将要排序的数据分割成独立的两部分,其中一部分的所有数据都比另外一部分的所有数据都要小, 然后再按此方法对 ...
- Java eclipse等常见问题汇总
1.Eclipse 注释模板设置 2.eclipse注释字体大小显示不一样大
- Oracle-查看sql运行状况
查看占io较大的正在运行的session SELECT se.sid, se.serial#, pr.SPID, se.username, se.status, se.terminal, se.pro ...