倍增LCA+树上差分样例过不了,求助
查看原帖
倍增LCA+树上差分样例过不了,求助
1101266
Fe_CuSO4__FeSO4_Cu楼主2025/1/26 09:23
#include <bits/stdc++.h>
using namespace std;
#define N 1000020
#define ll long long
#define bol bitset
#define imax INT_MAX
#define imin INT_MIN
#define inf 0x3f3f3f3f
#define B 100000010
#define min(a,b) (a<b ? a:b)
#define max(a,b) (a>b ? a:b)
#define lowbit(x) (x&(-x))
vector<int>e[N];
int fa[N][20],dep[N],n,m;
ll sum[N],ans;
void dfs(int u)
{
	for(int i=0;i<e[u].size();i++)
	{
		int k=e[u][i];
		if(k==fa[u][0]) continue;
		dep[k]=dep[u]+1;
		fa[k][0]=u;
		dfs(k);
	}
}
void init()
{
	for(int i=1;i<=18;i++)
	{
		for(int j=1;j<=n;j++)
		{
			fa[j][i]=fa[fa[j][i-1]][i-1];
		}
	}
}
int LCA(int x,int y)
{
	if(dep[x]<dep[y]) swap(x,y);
	for(int i=18;i>=0;i--)
	{
		if(dep[y]<=dep[fa[x][i]]) x=fa[x][i];
	}
	if(x==y) return x; 
	for(int i=18;i>=0;i--)
	{
		if(fa[x][i]!=fa[y][i])
		{
			x=fa[x][i],y=fa[y][i];
		}
	}
	return (fa[x][0]==0 ? 1 : fa[x][0]);
}
void get(int u)
{
	for(int i=0;i<e[u].size();i++)
	{
		int k=e[u][i];
		if(k==fa[u][0]) continue;
		dfs(k);
		sum[u]+=sum[k];
	}
}
int a[N];
int main()
{
	
	cin>>n;
	for(int i=1;i<=n;i++) cin>>a[i];
	for(int i=1;i<n;i++)
	{
		int u,v;
		cin>>u>>v;
		e[u].push_back(v);
		e[v].push_back(u);
	}
	dfs(1);
	init();
	a[0]=1;
	for(int i=1;i<n;i++)
	{
		sum[a[i]]++;
		sum[a[i+1]]++;
		int f=LCA(a[i+1],a[i]);
		sum[f]--;
		sum[fa[f][0]]--;
	}
	sum[a[n]]++;
	get(1);
	for(int i=1;i<=n;i++)
	{
		cout<<sum[i]<<"\n";
	}
	return 0;
}

输出有负数。无须发代码,指出错误即可

2025/1/26 09:23
加载中...