首页 > 其他分享 >[BZOJ4771] 七彩树 题解

[BZOJ4771] 七彩树 题解

时间:2024-12-25 12:09:08浏览次数:3  
标签:rt 七彩 BZOJ4771 la int 题解 void dep return

好题,又学两个思路。


先把问题变简单一点,去掉深度限制,那么有两种做法:

  • 经典的前驱后继转化到二维数点。

  • 颜色相同的点按 \(dfs\) 序排序,每个点 \(+1\),相邻两点 \(lca-1\)。转化为区间求和。

第二种相对实现简单。

假如加上深度,我们可以离线问题,按深度顺序加点。

要在线的话,只需要将线段树改为主席树即可。

时间复杂度 \(O(\sum (n+m)\log n)\)。

除了刚才说到的主席树外,本题在实现过程中还需要用到单调栈(记录自己插入时已经插入且颜色相同的前驱后继)。

#include<bits/stdc++.h>
using namespace std;
const int N=2e5+5,M=3e7+5;
int n,tot,low[N],id;
int m,q,dfn[N],a[N];
int tp,sk[N],ij,rt[N];
int t,dep[N],st[N][20];
int fs[N],pr[N],nx[N];
int ls[M],rs[M],sm[M];
vector<int>g[N],cl[N];
int cmp(int x,int y){
	return dfn[x]<dfn[y];
}int cmd(int x,int y){
	return dep[x]<dep[y];
}int cmb(int x,int y){
	return dep[x]!=dep[y]?cmd(x,y):x<y;
}void dfs(int x,int fa){
	fs[x]=++ij,dfn[x]=++id;
	dep[st[ij][0]=x]=dep[fa]+1;
	for(auto y:g[x])
		dfs(y,x),st[++ij][0]=x;
	low[x]=id;
}int minn(int x,int y){
	return dep[x]>dep[y]?y:x;
}void ST(){
	for(int i=0;i<19;i++)
		for(int j=1;j<=ij-(1<<(i+1))+1;j++)
			st[j][i+1]=minn(st[j][i],st[j+(1<<i)][i]);
}int rmq(int l,int r){
	int k=log2(r-l+1),x=r-(1<<k)+1;
	return minn(st[l][k],st[x][k]);
}int lca(int x,int y){
	if(fs[x]>fs[y]) swap(x,y);
	return rmq(fs[x],fs[y]);
}void add(int &x,int y,int l,int r,int k,int v){
	sm[x=++tot]=sm[y]+v;
	if(l==r) return;int mid=(l+r)/2;
	if(k<=mid) add(ls[x],ls[y],l,mid,k,v),rs[x]=rs[y];
	else add(rs[x],rs[y],mid+1,r,k,v),ls[x]=ls[y];
}int sum(int x,int y,int l,int r,int L,int R){
	if(L>R) return 0;
	if(L<=l&&r<=R) return sm[x]-sm[y];
	int mid=(l+r)/2,re=0;
	if(L<=mid) re=sum(ls[x],ls[y],l,mid,L,R);
	if(R>mid) re+=sum(rs[x],rs[y],mid+1,r,L,R);
	return re;
}void solve(){
	for(int i=1;i<=n;i++)
		pr[i]=nx[i]=0,g[i].clear(),cl[i].clear();
	cin>>n>>q,tot=id=ij=0;int la=0;
	for(int i=1,x;i<=n;i++)
		cin>>x,a[i]=i,cl[x].push_back(i);
	for(int i=2,x;i<=n;i++)
		cin>>x,g[x].push_back(i);
	dfs(1,0),ST();
	sort(a+1,a+n+1,cmd),m=dep[a[n]];
	for(int i=1,k;i<=n;i++){
		k=cl[i].size(),sort(cl[i].begin(),cl[i].end(),cmp);
		for(int j=0;j<k;j++){
			while(tp&&cmb(cl[i][j],cl[i][sk[tp]])) tp--;
			if(tp) pr[cl[i][j]]=cl[i][sk[tp]];sk[++tp]=j;
		}while(tp) sk[tp--]=0;
		for(int j=k-1;~j;j--){
			while(tp&&cmb(cl[i][j],cl[i][sk[tp]])) tp--;
			if(tp) nx[cl[i][j]]=cl[i][sk[tp]];sk[++tp]=j;
		}while(tp) sk[tp--]=0;
	}for(int i=1;i<=n;i++){
		int nw=dep[a[i]];rt[nw]=rt[dep[a[i-1]]];
		add(rt[nw],rt[nw],1,n,dfn[a[i]],1);
		if(nx[a[i]]) add(rt[nw],rt[nw],1,n,dfn[lca(a[i],nx[a[i]])],-1);
		if(pr[a[i]]) add(rt[nw],rt[nw],1,n,dfn[lca(a[i],pr[a[i]])],-1);
		if(nx[a[i]]&&pr[a[i]])
			add(rt[nw],rt[nw],1,n,dfn[lca(pr[a[i]],nx[a[i]])],1);
	}while(q--){
		int x,d;cin>>x>>d,x^=la,d^=la;
		la=sum(rt[min(dep[x]+d,m)],rt[dep[x]-1],1,n,dfn[x],low[x]);
		cout<<la<<"\n";
	}
}int main(){
	ios::sync_with_stdio(0);
	cin.tie(0),cout.tie(0);
	cin>>t;
	while(t--) solve();
	return 0;
}

标签:rt,七彩,BZOJ4771,la,int,题解,void,dep,return
From: https://www.cnblogs.com/chang-an-22-lyh/p/18630099/bzoj4771-qi_cai_shu-tj

相关文章

  • CF2043C 题解
    CF2043C题解题意给定一个除了\(-1,1\)之外,最多存在一个\(x,x\in[-10^9,10^9]\)的数的序列,求其子段和的所有可能值,从小到大输出。分析很容易就去思考如何从这个特殊的\(x\)入手。于是先排除这个特例,考虑全都是\(1,-1\)的情形,那么顺序从左到右不断加入\(a_i\),可以发现......
  • P6779 [Ynoi2009] rla1rmdq 题解
    Description给定一棵\(n\)个节点的树,树有边权,与一个长为\(n\)的序列\(a\)。定义节点\(x\)的父亲为\(fa(x)\),根\(rt\)满足\(fa(rt)=rt\)。定义节点\(x\)的深度\(dep(x)\)为其到根简单路径上所有边权和。有\(m\)次操作:1lr:对于\(l\lei\ler\),\(a_i\lef......
  • 【Web】2024“国城杯”网络安全挑战大赛决赛题解(全)
    最近在忙联通的安全准入测试,很少有时间看CTF了,今晚抽点时间回顾下上周线下的题(期末还没开始复习......
  • 题解 P9885【[Qingdao18A] Sequence and Sequence】
    具体数学还在发力!题目描述考虑下列两个序列\(P\)和\(Q\)。我们用\(P(i)\)表示序列\(P\)中的第\(i\)个元素,用\(Q(i)\)表示序列\(Q\)中的第\(i\)个元素:序列\(P\)是一个已排序的序列,其中,对于所有\(k\in\mathbb{Z^+}\),\(k\)在序列\(P\)中出现\((k+1)\)......
  • [THUSC2015] 异或运算 题解
    学到新思路了:求解\(k\)大值时,可以将所有元素放一块一起跑。考虑到\(n,q\)奇小无匹,我们便可以制造一个\(O(qn\logV)\)的代码。那么对于我们不想在时间复杂度中出现的\(m\),我们直接把他扔进可持久化\(Trie\)中销赃。再根据刚才那个思路,将\([u,d]\)中所有点扔进可持......
  • ZJOI2016 旅行者 题解
    ZJOI2016旅行者题解题目大意:给定一个\(n\timesm\)的网格图,相邻的四连通的点之间有给定边权的双向边,有\(Q\)个离线询问,问两个点之间的最短路。\(n\timesm\le2\times10^4,Q\le10^5\)。发现了吗?和上次省选组的三角剖分那道题很像,这种平面图上的最短路很有可能是分治......
  • 省选模拟题解
    \(T1\)题解题意:有一张\(n\)个点的有标号无向图,分为了\(k\)个连通块,第\(i\)个连通块的大小是\(s_i\),每个连通块都是完全图(节点之间两两有边)。要加\(k-1\)条边使得图连通,计算所有连边方案的权值和。假设第\(i\)个连通块被多加了\(d_i\)条边,那么该连边方案的权值为\(......
  • 湖南科技大学2024年计算机程序设计新生赛题解
    @目录前言前置知识补充问题A:珂朵莉解题思路代码问题B:可莉的烦恼解题思路代码问题C:小A的画解题思路代码问题D:KMP自动机fail树dfs序建可持久化线段树解题思路代码问题E:谜题:结局解题思路代码问题F:奶龙列阵(easyversion)解题思路代码问题G:小A的密码解题思路代码问......
  • [BZOJ2741][FOTILE模拟赛] L 题解
    相当好的题目,虽然和我前几天出的题重了qwq。\(lmx\)是我们的红太阳,没有他我们就会死!!!暴力枚举一个端点,然后用可持久化\(01\Trie\)或者离线\(Trie\)(当然这题用不了,但不强制在线的话是可以的)得到答案。时间复杂度\(O(nm\logn)\),过不了,考虑优化。红太阳\(lmx\)曾经说过:当......
  • umount: /xxx: target is busy问题解决
    在卸载文件系统的时候,提示umount:/tqls_system:targetisbusy,表示挂载的文件系统正在被使用。要卸载文件系统,必须结束使用文件或者目录的进程`fuser`命令用于查看使用特定文件或者文件系统的进程ID主要参数如下:```-mNAME,--mountNAME NAMEspecifiesafileonamou......