[牛客练习赛42E][LCT][树状数组]热爆了

本文链接：https://blog.csdn.net/Rose_max/article/details/88806497

Description

小 X 决定出一道送温暖题来和大家一起愉悦
他给了你一棵 n 个节点的树，每个点有个点权
ai
现在他给了你 Q 个询问，每次会给定 L,R ，然后定义满足
ai∈[L,R] 的点 i 为关键点
你需要回答出满足下列至少一个条件的点 x 的个数：1. x 是关键点2. 在树上删去 x 和所有与其相连的边后，存在两个关键点 a,b ，使得 a 和 b 不连通

Input

第一行两个整数 n,Q
第二行 n 个整数，表示
ai
接下来 n - 1 行，每行两个整数 u,v ，表示树上有一条连接点 u 和点 v 的边
接下来 Q 行，每行两个整数 L,R

Output

Q 行，每行一个整数，表示本次询问的答案

Sample Input

5 4
7 7 7 1 10
1 2
1 3
2 4
2 5
1 10
1 7
7 10
1 1

Sample Output

5
4
4
1

题解

之前看题解做法感觉很神仙这个题…
然后就把这个题扔给了同学玩qwq…
玩着玩着~~我胡汉三又回来啦~~
然后就玩了另一个做法
以1为根
首先可以离散化后使得所有节点权值各不相同
不难发现，有贡献的节点当且仅当其子树中拥有值域在 $[L, R]$ 中至少一个节点
注意到此处其斯坦纳树的根节点上一段会被统计到，最后去掉即可
其根节点一定是DFS序最小与最大的两个点的LCA，这个随便做完事
我们考虑离线从小往大扫节点
每条点到父亲的边，我们让他存在与其子树内当前扫过的权值最大的点所在的splay中
显而易见，我们需要维护这个的话，每次access一下那个点，在途中修改遇到的splay即可
多用一个树状数组维护每个splay的size，最后询问就可以直接搞了
把每个splay代表的点扔在这个splay的root上，修改就暴力跳到root找到他是哪个
随意清空一下就完事了
具体可以看代码
~~复杂度低又短还无脑比题解做法优越多了好吧qwq~~
然后这个题数据有点锅
注意一下 $L, R$ 可能存在大于 $10^9$ 的情况…我被这个锅了半个晚上mmp

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<queue>
#include<vector>
#include<ctime>
#include<map>
#include<bitset>
#include<set>
#define LL long long
#define mp(x,y) make_pair(x,y)
#define pll pair<long long,long long>
#define pii pair<int,int>
using namespace std;
inline int read()
{
	int f=1,x=0;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
int stack[20];
inline void write(int x)
{
	if(x<0){putchar('-');x=-x;}
    if(!x){putchar('0');return;}
    int top=0;
    while(x)stack[++top]=x%10,x/=10;
    while(top)putchar(stack[top--]+'0');
}
inline void pr1(int x){write(x);putchar(' ');}
inline void pr2(int x){write(x);putchar('\n');}
const int MAXN=100005;
const int MAXM=500005;
const int INF=(1<<31-1);

struct edge{int x,y,next;}a[2*MAXN];int len,last[MAXN];
void ins(int x,int y){len++;a[len].x=x;a[len].y=y;a[len].next=last[x];last[x]=len;}

int bi[MAXN],n,m,id[MAXN],bin[25],Log[MAXN];
int lowbit(int x){return x&-x;}
void modify(int x,int c){for(;x<=n;x+=lowbit(x))bi[x]+=c;}
int qry(int x){int ret=0;for(;x>=1;x-=lowbit(x))ret+=bi[x];return ret;}

struct linkcuttree
{
	int son[2],f,tot;
	linkcuttree(){son[0]=son[1]=f=tot=0;}
}tr[MAXN];
void updata(int now){tr[now].tot=tr[tr[now].son[0]].tot+tr[tr[now].son[1]].tot+1;}
void rotate(int x,int w)
{
	int f=tr[x].f,ff=tr[f].f;
	int R,r;
	R=f;r=tr[x].son[w];
	tr[R].son[1-w]=r;
	if(r)tr[r].f=R;
	R=ff;r=x;
	if(tr[R].son[0]==f)tr[R].son[0]=r;
	else if(tr[R].son[1]==f)tr[R].son[1]=r;
	tr[r].f=R;
	R=x;r=f;
	tr[R].son[w]=r;
	tr[r].f=R;
	updata(f);updata(x);
}
bool isroot(int x){return (tr[tr[x].f].son[0]!=x&&tr[tr[x].f].son[1]!=x);}
void splay(int x,int rt)
{
	while(!isroot(x))
	{
		int f=tr[x].f,ff=tr[f].f;
		if(ff==rt||isroot(f))
		{
			if(tr[f].son[0]==x)rotate(x,1);
			else rotate(x,0);
		}
		else
		{
			if(tr[ff].son[0]==f&&tr[f].son[0]==x)rotate(f,1),rotate(x,1);
			else if(tr[ff].son[0]==f&&tr[f].son[1]==x)rotate(x,0),rotate(x,1);
			else if(tr[ff].son[1]==f&&tr[f].son[1]==x)rotate(f,0),rotate(x,0);
			else rotate(x,1),rotate(x,0);
		}
	}
}
int getroot(int x)
{
	while(tr[x].son[0])x=tr[x].son[0];
	return x;
}
void access(int x,int now)
{
	int y=0;
	while(x!=0)
	{
		splay(x,0);
		if(tr[x].son[1])
		{
			int u=getroot(x),num0=tr[x].tot,num1=tr[tr[x].son[1]].tot;
			modify(id[u],-num0+num1);int lin=id[u];id[u]=INF;
			u=getroot(tr[x].son[1]);id[u]=lin;
		}
		else
		{
			int u=getroot(x),num0=tr[x].tot;
			modify(id[u],-num0);id[u]=INF;
		}
		tr[x].son[1]=y;updata(x);
		if(y!=0)tr[y].f=x;
		y=x;x=tr[x].f;
	}
	int u=getroot(y);id[u]=now;
	modify(id[u],tr[y].tot);
}
vector<pii> vec[MAXN];
int cal[MAXN],pos[MAXN],A[MAXN],B[MAXN];
struct LS{int y,p;}ls[MAXN];
bool cmp(LS n1,LS n2){return n1.y<n2.y;}
int answer[MAXM];

int in[MAXN],dfn,dep[MAXN];
int fa[25][MAXN];
int lca(int x,int y)
{
	if(dep[x]<dep[y])swap(x,y);
	for(int i=17;i>=0;i--)if(bin[i]<=dep[x]&&dep[fa[i][x]]>=dep[y])x=fa[i][x];
	if(x==y)return x;
	for(int i=17;i>=0;i--)if(bin[i]<=dep[x]&&fa[i][x]!=fa[i][y])x=fa[i][x],y=fa[i][y];
	return fa[0][x];
}
void DFS(int x)
{
	for(int i=1;bin[i]<=dep[x];i++)fa[i][x]=fa[i-1][fa[i-1][x]];
	tr[x].f=fa[0][x];in[x]=++dfn;
	for(int k=last[x];k;k=a[k].next)if(a[k].y!=fa[0][x])
	{
		fa[0][a[k].y]=x;
		dep[a[k].y]=dep[x]+1,DFS(a[k].y);
	}
}
int mn[25][MAXN],mx[25][MAXN];
inline int _min(int x,int y){return in[ls[x].p]<in[ls[y].p]?x:y;}
inline int _max(int x,int y){return in[ls[x].p]>in[ls[y].p]?x:y;}
void init()
{
	Log[1]=0;for(int i=2;i<MAXN;i++)Log[i]=Log[i>>1]+1;
	for(int i=1;i<=n;i++)mn[0][i]=mx[0][i]=i;
	for(int i=1;bin[i]<=n;i++)
		for(int x=1;x+bin[i]-1<=n;x++)
		{
			mn[i][x]=_min(mn[i-1][x],mn[i-1][x+bin[i-1]]);
			mx[i][x]=_max(mx[i-1][x],mx[i-1][x+bin[i-1]]);
		}
}
int get(int l,int r,int o)
{
	int K=Log[r-l+1];
	if(o)return _max(mx[K][l],mx[K][r-bin[K]+1]);
	else return _min(mn[K][l],mn[K][r-bin[K]+1]);
}
int main()
{
	bin[0]=1;
	for(int i=1;i<=20;i++)bin[i]=bin[i-1]<<1;
	memset(id,63,sizeof(id));
	n=read();m=read();
	for(int i=1;i<=n;i++)cal[i]=ls[i].y=read(),ls[i].p=i;
	sort(ls+1,ls+1+n,cmp);	
	for(int i=1;i<n;i++)
	{
		int x=read(),y=read();
		ins(x,y);ins(y,x);
	}
	DFS(1);init();
	int ln=n;cal[++ln]=-INF;cal[++ln]=INF;
	sort(cal+1,cal+1+ln);
	for(int i=1;i<=m;i++)
	{
		int L=read(),R=read();
		if(L>R)continue;
		int u=lower_bound(cal+1,cal+1+ln,L)-(cal+1);
		int v=upper_bound(cal+1,cal+1+ln,R)-(cal+2);
		if(R>cal[ln-1])v=n;
		if(u<=v)
		{
			vec[v].push_back(mp(u,i));
			int u1=get(u,v,0),u2=get(u,v,1);
			int LA=lca(ls[u1].p,ls[u2].p);
			answer[i]-=dep[LA];
		}
	}
	for(int i=1;i<=n;i++)
	{
		access(ls[i].p,i);
		for(int j=0;j<vec[i].size();j++)
		{
			int l=vec[i][j].first,op=vec[i][j].second;
			answer[op]+=qry(i)-qry(l-1);
		}
	}
	for(int i=1;i<=m;i++)pr2(answer[i]);
	return 0;
}