洛谷P1600 天天爱跑步(差分 LCA 桶)
题意
Sol
一步一步的来考虑
:直接的暴力
链的情况:维护两个差分数组,分别表示从左向右和从右向左的贡献,
:统计每个点的子树内有多少起点即可
:同样还是差分的思想,由于每个点 能对其产生的点的深度是相同的(假设为),那么访问该点时记录下的数量,将结束时的数量与其做差即可
满分做法和上面类似,我们考虑把每个点的贡献都转换到子树内统计
对于每次询问,拆为两种(从下到上 / 从上到下)
从上往下需要满足的条件:
从下往上需要满足的条件:
#include<bits/stdc++.h>
#define Pair pair<int, int>
#define MP make_pair
#define fi first
#define se second
using namespace std;
const int MAXN = 1e6 + 10, mod = 1e9 + 7, B = 20;
inline int read() {
char c = getchar(); int x = 0, f = 1;
while(c < '0' || c > '9') {if(c == '-') f = -1; c = getchar();}
while(c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
return x * f;
}
int N, M, ans[MAXN], dep[MAXN], top[MAXN], son[MAXN], siz[MAXN], fa[MAXN], S[MAXN],
T[MAXN], w[MAXN], tmp[MAXN], num2[MAXN], sum1[MAXN], sum2[MAXN], Lca[MAXN];
int *num1;//上 -> 下
vector<int> up[MAXN], da[MAXN], dc[MAXN];
vector<int> v[MAXN];
void dfs(int x, int _fa) {
dep[x] = dep[_fa] + 1; siz[x] = 1; fa[x] = _fa;
for(int i = 0, to; i < v[x].size(); i++) {
if((to = v[x][i]) == _fa) continue;
dfs(to, x);
siz[x] += siz[to];
if(siz[to] > siz[son[x]]) son[x] = to;
}
}
void dfs2(int x, int topf) {
top[x] = topf;
if(!son[x]) return ;
dfs2(son[x], topf);
for(int i = 0, to; i < v[x].size(); i++)
if(!top[to = v[x][i]]) dfs2(to, to);
}
int LCA(int x, int y) {
while(top[x] ^ top[y]) {
if(dep[top[x]] < dep[top[y]]) swap(x, y);
x = fa[top[x]];
}
return dep[x] < dep[y] ? x : y;
}
void Deal(int s, int t, int id) {// from s to t
int lca = LCA(s, t); Lca[id] = lca;
up[lca].push_back(s);//from down to up
int dis = dep[s] + dep[t] - 2 * dep[lca];
sum2[s]++;
da[t].push_back(dep[t] - dis);//increase
dc[lca].push_back(dep[t] - dis);//decrase
}
void Find(int x) {
int t1 = num1[dep[x] - w[x]], t2 = num2[dep[x] + w[x]];// 1: 从上往下 2:从下往上
for(int i = 0, to; i < v[x].size(); i++) {
if((to = v[x][i]) == fa[x]) continue;
Find(to);
}
num2[dep[x]] += sum2[x];
for(int i = 0; i < da[x].size(); i++) num1[da[x][i]]++;
ans[x] += num2[dep[x] + w[x]] - t2 + num1[dep[x] - w[x]] - t1;
for(int i = 0; i < up[x].size(); i++) num2[dep[up[x][i]]]--;
for(int i = 0; i < dc[x].size(); i++) num1[dc[x][i]]--;
}
int main() {
//freopen("a.in", "r", stdin); freopen("a.out", "w", stdout);
num1 = tmp + (int)3e5 + 10;
N = read(); M = read();
for(int i = 1; i <= N - 1; i++) {
int x = read(), y = read();
v[x].push_back(y); v[y].push_back(x);
}
dep[0] = -1; dfs(1, 0); dfs2(1, 1);
//for(int i = 1; i <= N; i++, puts("")) for(int j = 1; j <= N; j++) printf("%d %d %d\n", i, j, LCA(i, j));
for(int i = 1; i <= N; i++) w[i] = read();
for(int i = 1; i <= M; i++) S[i] = read(), T[i] = read(), Deal(S[i], T[i], i);
Find(1);
for(int i = 1; i <= M; i++) if(dep[S[i]] - dep[Lca[i]] == w[Lca[i]]) ans[Lca[i]]--;
for(int i = 1; i <= N; i++) printf("%d ", ans[i]);
return 0;
}
作者:自为风月马前卒
本文版权归作者和博客园共有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文连接,否则保留追究法律责任的权利。
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 开发中对象命名的一点思考
· .NET Core内存结构体系(Windows环境)底层原理浅谈
· C# 深度学习:对抗生成网络(GAN)训练头像生成模型
· .NET 适配 HarmonyOS 进展
· .NET 进程 stackoverflow异常后,还可以接收 TCP 连接请求吗?
· 本地部署 DeepSeek:小白也能轻松搞定!
· 基于DeepSeek R1 满血版大模型的个人知识库,回答都源自对你专属文件的深度学习。
· 在缓慢中沉淀,在挑战中重生!2024个人总结!
· 大人,时代变了! 赶快把自有业务的本地AI“模型”训练起来!
· Tinyfox 简易教程-1:Hello World!
2017-11-06 树上倍增求LCA及例题
2017-11-06 素数的筛法
2017-11-06 Catalan卡特兰数入门
2017-11-06 整理一下自己手撸的博客
2017-11-06 字符串hash入门
2017-11-06 洛谷P3375 【模板】KMP字符串匹配
2017-11-06 存一下自己的blog代码