@description@

给定一个 N 个点的树,编号为 1, 2, ..., N。第 i 条边连接 ai 与 bi。

再给定一个长度为 N 的 01 串,第 i 个字符表示 i 号点上是否有一个棋子。

保证至少有一个点有棋子。

你可以执行如下操作若干次:

选择两个距离 >= 2 的棋子,将这两个棋子往靠近对方的方向分别移动 1 条边。

是否可以通过若干次操作将所有棋子集中于一个点。如果可以,求出最小操作次数。

Constraints

2≤N≤2000

|S|=N,只包含 0, 1 且至少有一个 1。

1≤ai,bi≤N(ai≠bi) 且形成一棵树。

Input

输入形式如下:

N

S

a1 b1

a2 b2

:

aN−1 bN−1

Output

如果不可能集中所有棋子,输出 -1;

否则输出最小操作次数。

Sample Input 1

7

0010101

1 2

2 3

1 4

4 5

1 6

6 7

Sample Output 1

3

@solution@

N 这么小,显然可以枚举最后集中的点,然后进行判断。

考虑规定了集中的点后,怎么快速判断是否合法。我们尝试去找操作下的不变量。

显然操作要分成两类:祖先关系与非祖先关系。

祖先关系的操作会将一个点拉低,一个点抬高;另一种操作会同时将两个点抬高。

那么深度和的奇偶性是不会变的。

同时发现两种操作下,深度和要么不变,要么减少 2。而当深度和 = 0 时所有点集中于一点。

那么我们可以忽略祖先操作,只剩下非祖先操作,那么深度和每次严格减少 2。只要可行,最小操作次数 = 深度和 / 2。

当然需要稍微证明一下这个的合法性:祖先关系的操作总可以与下一个操作交换。于是可以将所有祖先操作挪到最后。然后就没用了。

怎么判断是否可行呢?考虑根,如果只对根的不同子树进行操作都可以合法的话自然合法。

求出根的每棵子树的深度和。相当于每次操作选出两棵子树,它们的深度和分别减一。

若干正整数 a1, a2, ..., ak,每次操作同时选两个数减一,问是否可以全部减成 0。这其实是一个很经典的问题。

解决方法是比较 sum - max{ai} 与 max{ai}。如果 sum - max{ai} < max{ai},显然最后会剩下 max{ai} - (sum - max{ai});否则,一定可以消到只剩 0 或 1(取决于 sum{ai} 的奇偶性)。

假如只操作根就是这样。

假如还可以操作其他子树内的点,我们显然应该递归去 max{ai} 的那一棵子树,看最后能够得到最小深度和。

再将最终得到的最小深度和拿去上面的判定条件判断一下,就可以知道当前能够剩下的最小深度和。

@accepted code@

#include<cstdio>
#include<algorithm>
using namespace std;
const int MAXN = 2000;
const int INF = (1<<30);
struct edge{
int to; edge *nxt;
}edges[2*MAXN + 5], *adj[MAXN + 5], *ecnt=&edges[0];
void addedge(int u, int v) {
edge *p = (++ecnt);
p->to = v, p->nxt = adj[u], adj[u] = p;
p = (++ecnt);
p->to = u, p->nxt = adj[v], adj[v] = p;
}
int a[MAXN + 5], N;
int f[MAXN + 5], s[MAXN + 5];
void dfs(int x, int fa) {
s[x] = a[x], f[x] = 0;
for(edge *p=adj[x];p;p=p->nxt) {
if( p->to == fa ) continue;
dfs(p->to, x), s[x] += s[p->to];
f[x] += f[p->to] + s[p->to];
}
}
int get(int x, int fa) {
int mx = 0;
for(edge *p=adj[x];p;p=p->nxt) {
if( p->to == fa ) continue;
if( mx == 0 || f[p->to] > f[mx] ) mx = p->to;
}
if( mx == 0 ) return 0;
int k = get(mx, x) + s[mx];
if( f[x] - f[mx] - s[mx] >= k ) return f[x] & 1;
else return k - (f[x] - f[mx] - s[mx]);
}
char S[MAXN + 5];
int main() {
scanf("%d%s", &N, S + 1);
for(int i=1;i<=N;i++) a[i] = S[i] - '0';
for(int i=1;i<N;i++) {
int u, v; scanf("%d%d", &u, &v);
addedge(u, v);
}
int ans = INF;
for(int j=1;j<=N;j++) {
dfs(j, 0);
if( get(j, 0) == 0 )
ans = min(ans, f[j]/2);
}
if( ans == INF ) printf("-1\n");
else printf("%d\n", ans);
}

@details@

惊了,这一场 AGC 的过题人数 E > F > D。

其实 E 还是比较容易往正解想。而且就算不会那个经典的模型,你也可以使用树形 dp 来搞定。

@atcoder - AGC034E@ Complete Compress的更多相关文章

  1. 「AGC034E」 Complete Compress

    「AGC034E」 Complete Compress 显然可以枚举根. 然后把某两棵棋子同时往深度浅的方向提,即对不存在祖先关系的两个棋子进行操作. 如果能到达那么就更新答案. 问题转化为如何判定能 ...

  2. AT4995-[AGC034E] Complete Compress【树形dp】

    正题 题目链接:https://www.luogu.com.cn/problem/AT4995 题目大意 \(n\)个点的一棵树,上面有一些棋子,每次可以选择两个棋子移动到他们之间的路径上相邻的点上, ...

  3. [atAGC034E]Complete Compress

    先考虑枚举最后的点,并以其为根 首先,操作祖先-后代关系是没有意义的,因为以后必然有一次操作会操作祖先使其返回原来的位置,那么必然不如操作后代和那一个点(少一次操作) 考虑某一次操作,总深度和恰好减2 ...

  4. WC2021 题目清单

    Day2 上午 <IOI题型与趣题分析> 来源 题目 完成情况 备注 IOI2002 Day1T1 Frog 已完成 IOI2002 Day1T2 Utopia IOI2002 Day1T ...

  5. 多校联训 DP 专题

    [UR #20]跳蚤电话 将加边变为加点,方案数为 \((n-1)!\) 除以一个数,\(dp\) 每种方案要除的数之和即可. 点击查看代码 #include<bits/stdc++.h> ...

  6. 【AtCoder】AGC034

    AGC034 刷了那么久AtCoder我发现自己还是只会ABCE(手动再见 A - Kenken Race 大意是一个横列,每个点可以跳一步或者跳两步,每个格子是空地或者石头,要求每一步不能走到石头或 ...

  7. 30个HTML初学者建议

    The most difficult aspect of running Nettuts+ is accounting for so many different skill levels. If w ...

  8. use zlib lib to compress or decompress file

    If you want to compress or decompress file when writing C++ code,you can choose zlib library,that's ...

  9. AtCoder Beginner Contest 103

    https://beta.atcoder.jp/contests/abc103 A - Task Scheduling Problem Time Limit: 2 sec / Memory Limit ...

随机推荐

  1. ES6--反引号的使用

    /*动态初始退出登出框话模态框*/ /*动态的初始化退出登陆模态框 反引号ES6语法 * 为什么在使用字符串格式直接创建模态框 * 1.不能在html页面中创建模板,因为如果换一个页面就没有对应的模板 ...

  2. CI框架--URL路径跳转与传值

    CI框架使用URL的前提是需要加载辅助函数$this->load->helper('url');当然我建议大家将所有需要加载的东西写在构造方法内,这样就不需每个控制器每个方法都去调用一次了 ...

  3. Markdown文档使用

    Markdown使用 一.markdown标题:1级-6级 一级 #空格 二级 ##空格 三级 ###空格 ... 六级 ######空格 二.代码块 print("hello world! ...

  4. 关于display:flex;兼容写法

    display: -moz-box; /* Firefox */ display: -ms-flexbox; /* IE10 */ display: -webkit-box; /* Safari */ ...

  5. 如何获取SpringBoot项目的applicationContext对象

    ApplicationContext对象是Spring开源框架的上下文对象实例,在项目运行时自动装载Handler内的所有信息到内存.传统的获取方式有很多种,不过随着Spring版本的不断迭代,官方也 ...

  6. 使用Httpclient 完美解决服务端跨域问题

    项目需求: jsonp是从前台js的角度考虑,通过Ajax调用springMVC的接口.同一个ip.同一个网络协议.同一个端口,三者都满足就是同一个域,否则就是跨域问题了.首页广告需要一个轮播的效果, ...

  7. 只需一步,DLA开启TableStore多元索引查询加速!

    一.背景介绍 Data Lake Analytics(简称DLA)在构建第一天就是支持直接关联分析Table Store(简称OTS)里的数据,实现存储计算分离架构,满足用户基于SQL接口分析Tabl ...

  8. WPF DrawingVisual详解

    在WPF中,如果需要绘制大量图形元素,并且对性能要求严苛的话,最好使用DrawingVisual,当然,你也可以选用 Path类和比Path类更轻量级的Geometry(几何形状)来实现你的需求,但是 ...

  9. 学习笔记(3)---安装SVM问题及解决方法

    1. LibSVM下载,安装 下载地址:http://www.csie.ntu.edu.tw/~cjlin/libsvm/,最新的版本是3.17 2. 入门 [heart_scale_label,he ...

  10. DirectX11笔记(五)--Direct3D渲染1--VERTICES AND INPUT LAYOUTS

    原文:DirectX11笔记(五)--Direct3D渲染1--VERTICES AND INPUT LAYOUTS 版权声明:本文为博主原创文章,未经博主允许不得转载. https://blog.c ...