[JLOI2014] 松鼠的新家(重链剖分+线段树)

题目传送门https://www.luogu.com.cn/problem/P3258好好的一道紫题为什么降绿了......

解题思路

可以先重链剖分一遍,给每个点编上一个号。

然后将树上问题转化为序列问题,那么再次想到线段树。

对于修改 间的路径,我们可以这样操作:

跳到同一条重链上,在跳的过程中,逐次对每个重链进行区间操作。

思路比较简单,但代码有点小长......

代码

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
#define int long long
int n,q,rt,mod;
int a[300001];
vector<int> g[300001];
int dep[300001];
int fa[300001];
int siz[300001];
int son[300001];
int id[300001],top[300001],cnt;
void dfs1(int x,int father,int deep)
{
	dep[x]=deep;
	fa[x]=father;
	siz[x]=1;
	int mxson=-1;
	for(auto y:g[x])
	{
		if(y==father)continue;
		dfs1(y,x,deep+1);
		siz[x]+=siz[y];
		if(siz[y]>mxson)
		{
			mxson=siz[y];
			son[x]=y;
		}
	}
}
void dfs2(int x,int tf)
{
	id[x]=++cnt;
	top[x]=tf;
	if(!son[x])return;
	dfs2(son[x],tf);
	for(auto y:g[x])
	{
		if(y==fa[x]||y==son[x])continue;
		dfs2(y,y);
	}
}
struct tree{
	int sum,l,r,add;
}tr[4000001];
void build(int u,int l,int r)
{
	tr[u]={0,l,r,0};
	if(l==r)
	{
		tr[u].sum=0;
		return;
	}
	int mid=l+r>>1;
	build(u*2,l,mid);
	build(u*2+1,mid+1,r);
	tr[u].sum=tr[u*2].sum+tr[u*2+1].sum;
}
void push_down(int u)
{
	if(tr[u].add)
	{
		tr[u*2].sum+=(tr[u*2].r-tr[u*2].l+1)*tr[u].add;
		tr[u*2].add+=tr[u].add;
		tr[u*2+1].sum+=(tr[u*2+1].r-tr[u*2+1].l+1)*tr[u].add;
		tr[u*2+1].add+=tr[u].add;
		tr[u].add=0;
	}
}
void push_up(int u)
{
	tr[u].sum=tr[u*2].sum+tr[u*2+1].sum;
}
void change(int u,int l,int r,int d)
{
	if(l<=tr[u].l&&tr[u].r<=r)
	{
		tr[u].sum+=(tr[u].r-tr[u].l+1)*d;
		tr[u].add+=d;
		return;
	}
	push_down(u);
	int mid=tr[u].l+tr[u].r>>1;
	if(l<=mid)
		change(u*2,l,r,d);
	if(r>mid)
		change(u*2+1,l,r,d);
	push_up(u);
}
int query(int u,int l,int r)
{
	if(l<=tr[u].l&&tr[u].r<=r)
	{
		return tr[u].sum;
	}
	push_down(u);
	int mid=tr[u].l+tr[u].r>>1;
	int res=0;
	if(l<=mid)
		res+=query(u*2,l,r);
	if(r>mid)
		res+=query(u*2+1,l,r);
	return res;
}
void update(int x,int y)
{
	while(top[x]!=top[y])
	{
		if(dep[top[x]]<dep[top[y]])
		{
			swap(x,y);
		}
		change(1,id[top[x]],id[x],1);
		x=fa[top[x]];
	}
	if(dep[x]>dep[y])
		swap(x,y);
	change(1,id[x],id[y],1);
	
}
signed main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
	}
	
	int u,v;
	for(int i=1;i<n;i++)
	{
		cin>>u>>v;
		g[u].push_back(v);
		g[v].push_back(u);
	}
	dfs1(1,0,1);
	dfs2(1,1);
	build(1,1,n);
	for(int i=1;i<n;i++)
	{
		update(a[i],a[i+1]);
		change(1,id[a[i+1]],id[a[i+1]],-1);
	}
	
	for(int i=1;i<=n;i++)
	{
		cout<<query(1,id[i],id[i])<<"\n";
	}
}
相关推荐
SweetCode3 分钟前
裴蜀定理:整数解的奥秘
数据结构·python·线性代数·算法·机器学习
ゞ 正在缓冲99%…16 分钟前
leetcode76.最小覆盖子串
java·算法·leetcode·字符串·双指针·滑动窗口
xuanjiong17 分钟前
纯个人整理,蓝桥杯使用的算法模板day2(0-1背包问题),手打个人理解注释,超全面,且均已验证成功(附带详细手写“模拟流程图”,全网首个
算法·蓝桥杯·动态规划
Zhichao_9734 分钟前
【UE5 C++课程系列笔记】33——商业化Json读写
c++·ue5
惊鸿.Jh36 分钟前
【滑动窗口】3254. 长度为 K 的子数组的能量值 I
数据结构·算法·leetcode
明灯L37 分钟前
《函数基础与内存机制深度剖析:从 return 语句到各类经典编程题详解》
经验分享·python·算法·链表·经典例题
碳基学AI43 分钟前
哈尔滨工业大学DeepSeek公开课:探索大模型原理、技术与应用从GPT到DeepSeek|附视频与讲义免费下载方法
大数据·人工智能·python·gpt·算法·语言模型·集成学习
补三补四1 小时前
机器学习-聚类分析算法
人工智能·深度学习·算法·机器学习
独好紫罗兰1 小时前
洛谷题单3-P5718 【深基4.例2】找最小值-python-流程图重构
开发语言·python·算法
云边有个稻草人1 小时前
【C++】第八节—string类(上)——详解+代码示例
开发语言·c++·迭代器·string类·语法糖auto和范围for·string类的常用接口·operator[]