比赛 期末考试4 评测结果 AAAAAAAAAAAAAMMMMMMMMTTTT
题目名称 树上查询 最终得分 52
用户昵称 2_16鸡扒拌面 运行时间 53.660 s
代码语言 C++ 内存使用 1039.06 MiB
提交时间 2026-02-12 11:32:56
显示代码纯文本
#include<bits/stdc++.h>
using namespace std;

typedef long long ll;
const int N=1e6+5,B=30;
int n,q,fa[N],a[N],dep[N],in[N],out[N],dfn;
vector<int> son[N],depv[N];
vector<ll> sumA[N];
vector<int> cntB[N][B+1];
int id[N];
ll pre[N][B+1][2];
bool chain;
void dfs(int x) {
	in[x]=++dfn;
	id[dfn]=x;
	depv[dep[x]].push_back(dfn);
	for(int i=0; i<son[x].size(); i++) {
		int y=son[x][i];
		dep[y]=dep[x]+1;
		dfs(y);
	}
	out[x]=dfn;
}
int main() {
	freopen("tree.in","r",stdin);
	freopen("tree.out","w",stdout);
	scanf("%d",&n);
	for(int i=1; i<=n; i++)scanf("%d",&a[i]);
	for(int i=2; i<=n; i++)scanf("%d",&fa[i]);
	chain=1;
	for(int i=2; i<=n; i++)if(fa[i]!=i-1) {
			chain=0;
			break;
		}
	if(chain) {
		vector<ll> sumA(n+1,0);
		vector<vector<ll>> cnt1(B+1,vector<ll>(n+1,0));
		vector<vector<ll>> bit1(B+1,vector<ll>(n+1,0));
		for(int i=1; i<=n; i++) {
			sumA[i]=sumA[i-1]+a[i];
			for(int j=0; j<=B; j++) {
				cnt1[j][i]=cnt1[j][i-1]+(((i-1)>>j)&1);
				bit1[j][i]=bit1[j][i-1]+((((i-1)>>j)&1)&((a[i]>>j)&1));
			}
		}
		scanf("%d",&q);
		while(q--) {
			int x,k;
			scanf("%d%d",&x,&k);
			int r=min(x+k,n);
			int len=r-x+1;
			ll ans=sumA[r]-sumA[x-1];
			for(int j=0; j<=B; j++) {
				ll cnt=cnt1[j][r]-cnt1[j][x-1];
				ll sumb=bit1[j][r]-bit1[j][x-1];
				ans+=(1ll<<j)*(cnt-2*sumb);
			}
			printf("%lld\n",ans);
		}
		return 0;
	}
	for(int i=2; i<=n; i++)son[fa[i]].push_back(i);
	dep[1]=0;
	dfs(1);
	for(int d=0; d<=n; d++) {
		if(depv[d].empty())continue;
		sort(depv[d].begin(),depv[d].end());
		int sz=depv[d].size();
		sumA[d].resize(sz+1,0);
		for(int b=0; b<=B; b++)cntB[d][b].resize(sz+1,0);
		for(int i=0; i<sz; i++) {
			int y=id[depv[d][i]];
			sumA[d][i+1]=sumA[d][i]+a[y];
			for(int b=0; b<=B; b++)cntB[d][b][i+1]=cntB[d][b][i]+((a[y]>>b)&1);
		}
	}
	scanf("%d",&q);
	while(q--) {
		int x,k;
		scanf("%d%d",&x,&k);
		ll ans=0;
		for(int d=0; d<=k; d++) {
			int h=dep[x]+d;
			if(h>n||sumA[h].empty())continue;
			int L=lower_bound(depv[h].begin(),depv[h].end(),in[x])-depv[h].begin();
			int R=upper_bound(depv[h].begin(),depv[h].end(),out[x])-depv[h].begin()-1;
			if(L>R)continue;
			int sz=R-L+1;
			ll s=sumA[h][R+1]-sumA[h][L];
			int cb[B+1];
			for(int b=0; b<=B; b++)cb[b]=cntB[h][b][R+1]-cntB[h][b][L];
			ll add=s;
			for(int b=0; b<=B; b++)if((d>>b)&1)add+=(1ll*sz-2*cb[b])*(1ll<<b);
			ans+=add;
		}
		printf("%lld\n",ans);
	}
	return 0;
}