2022牛客多校3 A-Ancestor

题意：给定两棵树，每个节点都有权重，有若干个关键节点，这些关键节点在两棵树上是相同的。对于所有关键节点的LCA，我们可以删除其中一个关键节点，使得A树中剩余关键节点的LCA的权值大于B树中剩余关键节点的LCA的权值的方案有几种。

前置知识：倍增法 nlogn 预处理 logn求l任意两点的 LCA
本题由于关键结点k的个数可能很大 若逐个求复杂度太高 可以考虑先求出k个关键结点的前缀和，后缀和 然后k次枚举关键结点,每次logn时间求 front[i-1]和back[i+1]的LCA（因为a，b，c三点的LCA就是a和b的LCA再和c求LCA ，故k个结点中删除第i个结点的LCA就是前i-1个结点的LCA和后i+1个结点的LCA）
代码实现：

#include <bits/stdc++.h>
using namespace std;
const int N = 1e5+50;

vector<int> treeA[N];
vector<int> treeB[N];

int depthA[N],depthB[N];
int stA[N][25],stB[N][25];
int wa[N],wb[N];
int n,k;
int ftA[N],backA[N],fB[N],backB[N];

void dfsA(int u,int v)
{
	depthA[u] = depthA[v] + 1;
	for(int i=0; i<treeA[u].size(); ++i){
		dfsA(treeA[u][i],u);
	}
}

void dfsB(int u,int v)
{
	stB[u][0] = v;
	depthB[u] = depthB[v] + 1;
	for(int i=0; i<treeB[u].size(); ++i){
		dfsB(treeB[u][i],u);
	}
}

int lcaA(int x,int y)
{
	if(depthA[x]<depthA[y])swap(x,y);
	for(int i=log2(n); i>=0; --i){
		if(depthA[stA[x][i]]>=depthA[y])x = stA[x][i];
		if(depthA[x]==depthA[y])break;
	}
	if(x==y)return x;
	for(int i=log2(n); i>=0; --i){
		if(stA[x][i]!=stA[y][i]){
			x = stA[x][i];
			y = stA[y][i];
		}
	}
	return stA[x][0];
}

int lcaB(int x,int y)
{
	if(depthB[x]<depthB[y])swap(x,y);
	for(int i=log2(n); i>=0; --i){
		if(depthB[stB[x][i]]>=depthB[y])x = stB[x][i];
		if(depthB[x]==depthB[y])break;
	}
	if(x==y)return x;
	for(int i=log2(n); i>=0; --i){
		if(stB[x][i]!=stB[y][i]){
			x = stB[x][i];
			y = stB[y][i];
		}
	}
	return stB[x][0];
}


signed main()
{
	scanf("%d%d",&n,&k);
	vector<int> uniq(k+1);
	for(int i=1; i<=k; ++i)scanf("%d",&uniq[i]);
	for(int i=1; i<=n; ++i)scanf("%d",&wa[i]);
	for(int i=1; i<n; ++i){
		int x;
		scanf("%d",&x);
		treeA[x].push_back(i+1);
		stA[i+1][0] = x;
	}
	for(int i=1; i<=n; ++i)scanf("%d",&wb[i]);
	for(int i=1; i<n; ++i){
		int x;
		scanf("%d",&x);
		treeB[x].push_back(i+1);
		stB[i+1][0] = x;
	}
	dfsA(1,0);
	dfsB(1,0);
	for(int j=1; j<=log2(n); ++j){
		for(int i=1; i<=n; ++i){
			stA[i][j] = stA[stA[i][j-1]][j-1];
			stB[i][j] = stB[stB[i][j-1]][j-1];
		}
	}
	ftA[1] = uniq[1];
	fB[1] = uniq[1];
	backA[k] = uniq[k];
	backB[k] = uniq[k];
	for(int i=2; i<=k; ++i){
		ftA[i] = lcaA(ftA[i-1],uniq[i]);
		fB[i] = lcaB(fB[i-1],uniq[i]);
	}
	for(int i=k-1; i>=1; --i){
		backA[i] = lcaA(backA[i+1],uniq[i]);
		backB[i] = lcaB(backB[i+1],uniq[i]);
	}
	int res = 0;
	for(int i=2; i<=k-1; ++i){
		int x = lcaA(ftA[i-1],backA[i+1]);
		int y = lcaB(fB[i-1],backB[i+1]);
		if(wa[x]>wb[y])res++;
	}
	if(wa[backA[2]]>wb[backB[2]])res++;
	if(wa[ftA[k-1]]>wb[fB[k-1]] )res++;
	printf("%d\n", res);
	return 0;
}

还有一种办法是利用结论: k个点的LCA只与DFS序最大的和DFS序最小的这两个点有关,题目需要删除一个点，因此只需先DFS跑一遍打上标记，然后分类讨论是否删除即可