NOIP2016(D1T2)天天爱跑步题解
首先声明这不是一篇算法独特的题解,仍然是“LCA+桶+树上差分”,但这篇题解是为了让很多很多看了很多题解仍然看不懂的朋友们看懂的,其中就包括我,我也在努力地把解题的“思维过程”呈现出来,希望能帮助到别人。实在是佩服那些考场AC的大牛,再次向你们献上敬意!
1. 第一步
- 首先可以初步判断这个题肯定要计算LCA,方法有倍增/Tarjan-DFS,我们就写个简单的倍增吧,使用链式前向星存储边。
- 选择1号结点开始dfs,别的结点也可以
- dfs过程中计算
fa[][]
数组(fa[x][i]
表示 \(x\) 结点的 \(2^i\) 代祖先是谁)和deep[]数组(deep[x]
表示结点 \(x\) 在树中的深度)
#include<bits/stdc++.h>
using namespace std;
const int SIZE=300000;
int n, m, tot, h[SIZE], deep[SIZE], fa[SIZE][20], w[SIZE]; //w[i]表示i结点出现观察员的时间
struct edge
{
int to, next;
}E[SIZE*2], e1[SIZE*2], e2[SIZE*2]; //边集数组e1,e2留待备用
void add(int x, int y) //加边函数
{
E[++tot].to=y;
E[tot].next=h[x];
h[x]=tot;
}
void dfs1(int x) //dfs的过程中完成“建树”,预处理fa[][]数组, 计算deep[]数组
{
for(int i=1; (1<<i)<=deep[x]; i++)
fa[x][i]=fa[fa[x][i-1]][i-1]; //x的2^i代祖宗就是x的2^{i-1}代祖宗的2^{i-1}代祖宗
for(int i=h[x]; i; i=E[i].next)
{
int y=E[i].to;
if(y==fa[x][0]) continue; //如果y是父结点,跳过
fa[y][0]=x;
deep[y]=deep[x]+1;
dfs1(y);
}
}
int get_lca(int x, int y) //计算x和y的最近公共祖先
{
if(x==y) return x; //没有这一行,遇到 lca(x, x) 这样的询问时会挂掉
if(deep[x]<deep[y]) swap(x, y); //保持x的深度大于y的深度
int t=log(deep[x]-deep[y])/log(2);
for(int i=t; i>=0; i--) //x向上跳到和y同样的深度
{
if(deep[fa[x][i]]>=deep[y])
x=fa[x][i];
if(x==y)
return x;
}
t=log(deep[x])/log(2);
for(int i=t; i>=0; i--) //x和y一起向上跳
{
if(fa[x][i]!=fa[y][i])
x=fa[x][i], y=fa[y][i];
}
return fa[x][0];
}
int main() //先把主函数写上一部分
{
scanf("%d%d", &n, &m);
for(int i=1; i<n; i++)
{
int u, v;
scanf("%d%d", &u, &v);
add(u, v);
add(v, u);
}
deep[1]=1;
fa[1][0]=1;
dfs1(1);
for(int i=1; i<=n; i++) scanf("%d", &w[i]);
/////////////////////////////////////////////////////////////
////////////////////////未完待续///////////////////////////
/////////////////////////////////////////////////////////////
return 0;
}
2. 第二步
大概分析一下,m个玩家对应m条路径,有了起点和终点的 lca 后,如果我们模拟这个过程:
直觉
- 从起点 \(S_i\) 跑到 \(LCA\) 在树长得很匀称的情况下为 \(O(lgn)\)
- 从起点 \(LCA\) 跑到 \(T_i\) 在树长得很匀称的情况下为 \(O(lgn)\)
- 因此,模拟一个玩家的跑步过程为 \(O(lgn)\),m个玩家为 \(O(mlgn)\)
- 理想情况下是可行的,但现实就是不理想
- 题目清楚告诉你,树会退化成一条链,因此模拟一个过程变成 \(O(n)\),总的就是。。。\(O(mn)\),必挂无疑
- 此法不是正解!
尝试
- 我们能不能改变模拟跑步的过程,从 \(O(n)\) 优化到 \(O(lgn)\) 呢?思前想后不可能,有 \(n\) 个观察员矗在那里,你可以对哪个视而不见?
- 路已走到尽头
转换
- 这时候需要放大招,转换思想!或许解决问题的思路压根就不是一个玩家一个玩家模拟,而是整体处理呢?
- 也就是说,我们不枚举每个运动员,而是枚举每个观察员i,看看哪些结点会为这个观察员i做贡献(刚好在\(w_i\)秒跑到他这儿)。
- 枚举观察员的过程就是DFS整颗树的过程,我们可以在 \(O(n)\) 内搞定!
- 对于观察员i,哪些人会为他做贡献呢?
深入分析
- 对于结点 \(P\), 如果他位于一条起点、终点分别为 \(s_i\), \(t_i\) 的跑步路径上,如何判断这名选手会不会为 \(P\) 作贡献呢?
- 分情况考虑
- 如果 \(P\) 是在从 \(s_i\) 到 \(LCA\) 的路上,如下图:
我们可以得出结论:当起点 $ s_i $ 满足 $ deep[s_i]=w[P]+deep[P] $时,起点 \(s_i\)会为 \(P\) 观察员做一个贡献(运动员从\(s_i\)出发,可以被\(P\)处的观察员在\(w[P]\)秒看到)
如果 \(P\) 是在从 \(LCA\) 到 \(t_i\) 的路上,如下图:
- 定义 \(dist[s_i, t_i]\)为从 \(s_i\)出发到\(t_i\)的路径长度,如果运动员从\(s_i\)出发,可以被\(P\)处的观察员在\(w[P]\)秒观察到,可以由上图得出以下式子:
- \(dist[s_i, t_i]-w[P]=deep[t_i]-deep[P]\),移项后得到:
- $ dist[s_i, t_i]-deep[t_i]=w[P]-deep[P] $
- 我们可以得出结论:当终点 $ t_i $ 满足 $ dist[s_i, t_i]-deep[t_i]=w[P]-deep[P] $时,终点 \(t_i\)会为 \(P\) 观察员做一个贡献
- 做一个重要的总结:上行过程中,满足条件的起点可以做贡献,下行过程中,满足条件的终点可以做贡献,但无论是哪一种情形,能对 \(P\) 做贡献的起点或终点一定都在以\(P\)为根的子树上,这使得可以在DFS回溯的过程中处理以任意节点为根的子树。
3. 第三步
如何统计子树贡献
- 递归以\(P\)为根的子树时,可以统计出其子树中所有的起点和终点对它的贡献
- 这里又需要转换
- 子树中有的起点和终点对\(P\)产生了贡献,有些不对其产生贡献但对\(P\)以外的结点产生了贡献
- 所以我们不能枚举每个点(子树根),找子树中哪些点对其产生贡献,这样复杂度就上去了
- 而是对于树上的任何一个起点和终点,把其产生的贡献放在桶里面,回溯到子树根的时候再到桶里面查询结果
- 有人产生疑问了,也是很多人看不懂这里桶用法的地方,疑问如图:
- \(c\)点产生贡献放在桶的\(deep[c]\)位置,计算\(b\)点获得的贡献时当然是从\(bucket1[deep[b]+w[b]]\)位置获取,于是得到1个贡献,你发现\(a\)结点也是用的同一个桶,这个还好,因为\(c\)确实给他做了贡献,可是\(e\)点呢?他是不应该获得贡献的!既然我会给和我无关的结点做贡献,那么其它无关的结点难免也会给我做贡献!
- 问题总结一下,对于一个点\(P\)来说,究竟哪些点在桶里面产生的贡献才是有效的。
- 答案是:以\(P\)为根递归整颗子树过程中在桶内产生的差值才是有效的
还要考虑一种情况
- 先看图:
- 看懂了吗?对于以\(P\)为根的内部路径(不经过\(P\)),这条路径的起点和终点产生的贡献是不应该属于\(P\)的
- 所以dfs过程中,在统计当前结点作为起点和终点所产生的贡献后,继而计算出当前结点作为“根”上的差值后,在回溯过程中,一定要减去以当前结点为\(LCA\)的起点、终点在桶里产生的贡献,这部分贡献在离开这个子树后就没有意义了。
代码说明
e1,tot1,h1,add1
是使用链式前向星的方法存储每个结点作为终点对应的路径集合e2,tot2,h2,add2
是使用链式前向星的方法存储每个结点作为LCA对应的路径集合b1,b2
是两组桶,分别用于上行阶段和下行阶段的贡献统计js[SIZE]
用于统计以每个结点作为起点的路径条数dist[SIZE], s[SIZE], t[SIZE]
用于统计m条路径对应的长度,起点和终点信息ans[SIZE]
存储最后输出的答案,是每个结点观察员看到的人数
int tot1, tot2, h1[SIZE], h2[SIZE];
void add1(int x, int y)
{
e1[++tot1].to=y;
e1[tot1].next=h1[x];
h1[x]=tot1;
}
void add2(int x, int y)
{
e2[++tot2].to=y;
e2[tot2].next=h2[x];
h2[x]=tot2;
}
int b1[SIZE*2], b2[SIZE*2], js[SIZE], dist[SIZE], s[SIZE], t[SIZE], ans[SIZE];
void dfs2(int x)
{
int t1=b1[w[x]+deep[x]], t2=b2[w[x]-deep[x]+SIZE]; //递归前先读桶里的数值,t1是上行桶里的值,t2是下行桶的值
for(int i=h[x]; i; i=E[i].next) //递归子树
{
int y=E[i].to;
if(y==fa[x][0]) continue;
dfs2(y);
}
b1[deep[x]]+=js[x]; //上行过程中,当前点作为路径起点产生贡献,入桶
for(int i=h1[x]; i; i=e1[i].next) //下行过程中,当前点作为路径终点产生贡献,入桶
{
int y=e1[i].to;
b2[dist[y]-deep[t[y]]+SIZE]++;
}
ans[x]+=b1[w[x]+deep[x]]-t1+b2[w[x]-deep[x]+SIZE]-t2; //计算上、下行桶内差值,累加到ans[x]里面
for(int i=h2[x]; i; i=e2[i].next) //回溯前清除以此结点为LCA的起点和终点在桶内产生的贡献,它们已经无效了
{
int y=e2[i].to;
b1[deep[s[y]]]--; //清除起点产生的贡献
b2[dist[y]-deep[t[y]]+SIZE]--; //清除终点产生的贡献
}
}
int main()
{
////////////////重复部分跳过////////////
////////////////文末提供完整代码////////
for(int i=1; i<=m; i++) //读入m条询问
{
scanf("%d%d", &s[i], &t[i]);
int lca=get_lca(s[i], t[i]); //求LCA
dist[i]=deep[s[i]]+deep[t[i]]-2*deep[lca]]; //计算路径长度
js[s[i]]++; //统计以s[i]为起点路径的条数,便于统计上行过程中该结点产生的贡献
add1(t[i], i); //第i条路径加入到以t[i]为终点的路径集合中
add2(lca, i); //把每条路径归到对应的LCA集合中
if(deep[lca]+w[lca]==deep[s[i]]) ans[lca]--; //见下面的解释
}
dfs2(1); //dfs吧!
for(int i=1; i<=n; i++) printf("%d ", ans[i]);
return 0;
}
一些重要补充
- 上述代码中有一行未加解释
if(deep[lca]+w[lca]==deep[s[i]]) ans[lca]--;
- 考虑路径是这样的,如图:
这个图可能不太好懂,意思是:
如果路径起点或终点刚好为LCA且LCA处是可观察到运动员的,那么我们在上行统计过程中和下行统计过程中都会对该LCA产生贡献,这样就重复计数一次!
好在这种情况很容易发现,我们提前预测到,对相应的结点进行
ans[x]--
即可。此外,在使用第二个桶时,下标是
w[x]-deep[x]
会成为负数,所以使用第二个桶时,下标统一+SIZE
,向右平移一段区间,防止下溢。
4. 结束
我不知道自己说清楚没有,但愿大家不要拍砖头!下面是完整代码
#include<bits/stdc++.h>
using namespace std;
const int SIZE=300000;
int n, m, tot, h[SIZE], deep[SIZE], fa[SIZE][20], w[SIZE];
struct edge
{
int to, next;
}E[SIZE*2], e1[SIZE*2], e2[SIZE*2];
void add(int x, int y)
{
E[++tot].to=y;
E[tot].next=h[x];
h[x]=tot;
}
int tot1, tot2, h1[SIZE], h2[SIZE];
void add1(int x, int y)
{
e1[++tot1].to=y;
e1[tot1].next=h1[x];
h1[x]=tot1;
}
void add2(int x, int y)
{
e2[++tot2].to=y;
e2[tot2].next=h2[x];
h2[x]=tot2;
}
void dfs1(int x)
{
for(int i=1; (1<<i)<=deep[x]; i++)
fa[x][i]=fa[fa[x][i-1]][i-1];
for(int i=h[x]; i; i=E[i].next)
{
int y=E[i].to;
if(y==fa[x][0]) continue;
fa[y][0]=x;
deep[y]=deep[x]+1;
dfs1(y);
}
}
int get_lca(int x, int y)
{
if(x==y) return x;
if(deep[x]<deep[y]) swap(x, y);
int t=log(deep[x]-deep[y])/log(2);
for(int i=t; i>=0; i--)
{
if(deep[fa[x][i]]>=deep[y])
x=fa[x][i];
if(x==y)
return x;
}
t=log(deep[x])/log(2);
for(int i=t; i>=0; i--)
{
if(fa[x][i]!=fa[y][i])
x=fa[x][i], y=fa[y][i];
}
return fa[x][0];
}
int b1[SIZE*2], b2[SIZE*2], js[SIZE], dist[SIZE], s[SIZE], t[SIZE], l[SIZE], ans[SIZE];
void dfs2(int x)
{
int t1=b1[w[x]+deep[x]], t2=b2[w[x]-deep[x]+SIZE];
for(int i=h[x]; i; i=E[i].next)
{
int y=E[i].to;
if(y==fa[x][0]) continue;
dfs2(y);
}
b1[deep[x]]+=js[x];
for(int i=h1[x]; i; i=e1[i].next)
{
int y=e1[i].to;
b2[dist[y]-deep[t[y]]+SIZE]++;
}
ans[x]+=b1[w[x]+deep[x]]-t1+b2[w[x]-deep[x]+SIZE]-t2;
for(int i=h2[x]; i; i=e2[i].next)
{
int y=e2[i].to;
b1[deep[s[y]]]--;
b2[dist[y]-deep[t[y]]+SIZE]--;
}
}
int main()
{
scanf("%d%d", &n, &m);
for(int i=1; i<n; i++)
{
int u, v;
scanf("%d%d", &u, &v);
add(u, v);
add(v, u);
}
deep[1]=1;
fa[1][0]=1;
dfs1(1);
for(int i=1; i<=n; i++) scanf("%d", &w[i]);
for(int i=1; i<=m; i++)
{
scanf("%d%d", &s[i], &t[i]);
int lca=get_lca(s[i], t[i]);
dist[i]=deep[s[i]]+deep[t[i]]-2*deep[lca];
js[s[i]]++;
add1(t[i], i);
add2(lca, i);
if(deep[lca]+w[lca]==deep[s[i]]) ans[lca]--;
}
dfs2(1);
for(int i=1; i<=n; i++) printf("%d ", ans[i]);
return 0;
}
NOIP2016(D1T2)天天爱跑步题解的更多相关文章
- 「NOIP2016」天天爱跑步 题解
(声明:图片来源于网络) 「NOIP2016」天天爱跑步 题解 题目TP门 题目 题目描述 小c同学认为跑步非常有趣,于是决定制作一款叫做<天天爱跑步>的游戏.<天天爱跑步>是 ...
- 【NOIP2016】天天爱跑步 题解(LCA+桶+树上差分)
题目链接 题目大意:给定一颗含有$n$个结点的树,每个结点有一个权值$w$.给定$m$条路径,如果一个点与路径的起点的距离恰好为$w$,那么$ans[i]++$.求所有结点的ans. 题目分析 暴力的 ...
- LOJ #2359. 「NOIP2016」天天爱跑步(倍增+线段树合并)
题意 LOJ #2359. 「NOIP2016」天天爱跑步 题解 考虑把一个玩家的路径 \((x, y)\) 拆成两条,一条是 \(x\) 到 \(lca\) ( \(x, y\) 最近公共祖先) 的 ...
- UOJ261 【NOIP2016】天天爱跑步 LCA+动态开点线段树
UOJ261 [NOIP2016]天天爱跑步 Description 小c同学认为跑步非常有趣,于是决定制作一款叫做<天天爱跑步>的游戏.天天爱跑步是一个养成类游戏,需要玩家每天按时上线, ...
- NOIP2016天天爱跑步 题解报告【lca+树上统计(桶)】
题目描述 小c同学认为跑步非常有趣,于是决定制作一款叫做<天天爱跑步>的游戏.«天天爱跑步»是一个养成类游戏,需要玩家每天按时上线,完成打卡任务. 这个游戏的地图可以看作一一棵包含 nn个 ...
- [NOIP2016]天天爱跑步 题解(树上差分) (码长短跑的快)
Description 小c同学认为跑步非常有趣,于是决定制作一款叫做<天天爱跑步>的游戏.<天天爱跑步>是一个养成类游戏,需要 玩家每天按时上线,完成打卡任务.这个游戏的地图 ...
- 【NOIP2016】天天爱跑步
题目描述 小c同学认为跑步非常有趣,于是决定制作一款叫做<天天爱跑步>的游戏.«天天爱跑步»是一个养成类游戏,需要玩家每天按时上线,完成打卡任务. 这个游戏的地图可以看作一一棵包含 个结点 ...
- Noip 2016 天天爱跑步 题解
[NOIP2016]天天爱跑步 时间限制:2 s 内存限制:512 MB [题目描述] 小C同学认为跑步非常有趣,于是决定制作一款叫做<天天爱跑步>的游戏.<天天爱跑步>是 ...
- UOJ261 【NOIP2016】天天爱跑步
本文版权归ljh2000和博客园共有,欢迎转载,但须保留此声明,并给出原文链接,谢谢合作. 本文作者:ljh2000作者博客:http://www.cnblogs.com/ljh2000-jump/转 ...
随机推荐
- SuperSocket 案例
SuperSocket 是一个轻量级, 跨平台而且可扩展的 .Net/Mono Socket 服务器程序框架.你无须了解如何使用 Socket, 如何维护 Socket 连接和 Socket 如何工作 ...
- MT【311】三角递推数列
已知数列$\{a_n\}$满足$a_1=\dfrac{1}{2},a_{n+1}=\sin\left(\dfrac{\pi}{2}a_n\right),S_n$ 为$\{a_n\}$的前$n$项和,求 ...
- MT【277】华中科技大学理科实验班选拔之三次方程
(2015华中科技大学理科实验班选拔)已知三次方程$x^3+ax^2+bx+x=0$有三个实数根.(1)若三个实根为$x_1,x_2,x_3$,且$x_1\le x_2\le x_3,a,b$为常数, ...
- Linux 源码安装 Python3
下载源码包https://www.python.org/downloads/ 解压(以3.64版本为例)wget https://www.python.org/ftp/python/3.6.4/Pyt ...
- 工作笔记:/bin/bash^M: 坏的解释器: 没有那个文件或目录 问题解决
问题原因: 由于windows上换行符为CR LF而在Linux下是 LF 导致出现上述问题 解决方案 1. 在windows下 可以使用nodepad打开该shell文件,然后将shell文件中的格 ...
- ONI无法启动: Uh oh! Unable to launch Neovim...
问题描述 在终端中是可以打开nvim的,ONI无法正确找到位置 解决方法 修改配置文件,指定nvim的路径 终端中输入which nvim定位所在位置,这里返回的结果是/usr/local/bin/n ...
- 【ZJOI2007】粒子运动
若此段起始点为(stx,sty),速度为(vx,vy),设碰撞时间为t,则(stx+vx·t)²+(sty+vy·t)²=r² → stx²+vx²·t²+2·stx·vx·t+sty²+vy²·t² ...
- django 前端模板继承显示model中使用choices的字段
比如model中的一个class Need class Need(models.Model): """ 任务 """ party_a=mod ...
- 斯坦福大学公开课机器学习: machine learning system design | error analysis(误差分析:检验算法是否有高偏差和高方差)
误差分析可以更系统地做出决定.如果你准备研究机器学习的东西或者构造机器学习应用程序,最好的实践方法不是建立一个非常复杂的系统.拥有多么复杂的变量,而是构建一个简单的算法.这样你可以很快地实现它.研究机 ...
- linux ------ 使用 screen 后 SSH 断开后程序依旧能在后台运行
为什么ssh断开后你运行的进程会退出呢? 因为所有进程都得有个父进程.当你ssh到一个服务器上时,打开的shell就是你所有执行命令的父进程. 当你断开ssh连接时,你的命令的父进程就没了.如果处理不 ...