QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#90277#5830. 树FxorG10 68ms6164kbC++146.5kb2023-03-22 16:35:522023-03-22 16:36:12

Judging History

你现在查看的是最新测评结果

  • [2023-08-10 23:21:45]
  • System Update: QOJ starts to keep a history of the judgings of all the submissions.
  • [2023-03-22 16:36:12]
  • 评测
  • 测评结果：10
  • 用时：68ms
  • 内存：6164kb
查看
  • [2023-03-22 16:35:52]
  • 提交

answer

#include <bits/stdc++.h>
#define ll long long
#define int ll
#define pb push_back
using namespace std;
const int N=(int)(1e4+1);
vector<int>g[N];
ll d[20][N],ans[N];
int t[2][N][20],num[20][N],v[N],n,q;

struct node {
	int x,dep,id,add;
	node(int X,int Dep,int Add,int Id) {
		x=X; dep=Dep; add=Add; id=Id;
	}
};
vector<node>vec[20];
vector<int>D[N];
int dep[N],id[N],idtot,sz[N],m;
void dfs0(int x,int ff) {
	dep[x]=dep[ff]+1; id[x]=++idtot; sz[x]=1;
	D[dep[x]].pb(id[x]);
	for(int y:g[x]) {
		if(y==ff) continue ;
		dfs0(y,x); sz[x]+=sz[y];
	}
}

namespace DS1 {
	ll sum[N*20],sum2[N*20];
	int rt[N],ls[N*20],rs[N*20],tot;
	void clr() {
		for(int i=1;i<=n;i++) rt[i]=0;
		for(int i=1;i<=tot;i++) ls[i]=rs[i]=sum[i]=0;
		tot=0;
	}
	void push_up(int cur) {
		sum[cur]=0;
		if(ls[cur]) sum[cur]=sum[ls[cur]];
		if(rs[cur]) sum[cur]+=sum[rs[cur]];
		sum2[cur]=0;
		if(ls[cur]) sum2[cur]=sum2[ls[cur]];
		if(rs[cur]) sum2[cur]+=sum2[rs[cur]];
	}
	void upt(int &cur,int l,int r,int pos,int v) {
		if(!cur) cur=++tot;
		if(l==r) {
			sum[cur]=v; return ;
		}
		int mid=(l+r)>>1;
		if(pos<=mid) upt(ls[cur],l,mid,pos,v);
		else upt(rs[cur],mid+1,r,pos,v);
		push_up(cur);
	}
	void upt2(int &cur,int l,int r,int pos,int v) {
		if(!cur) cur=++tot;
		if(l==r) {
			sum2[cur]=v; return ;
		}
		int mid=(l+r)>>1;
		if(pos<=mid) upt2(ls[cur],l,mid,pos,v);
		else upt2(rs[cur],mid+1,r,pos,v);
		push_up(cur);
	}
	ll qry(int cur,int l,int r,int cl,int cr) {
		if(!cur) return 0;
		if(cl<=l&&r<=cr) return sum[cur];
		int mid=(l+r)>>1;
		if(cr<=mid) return qry(ls[cur],l,mid,cl,cr);
		if(cl>mid) return qry(rs[cur],mid+1,r,cl,cr);
		return qry(ls[cur],l,mid,cl,cr)+qry(rs[cur],mid+1,r,cl,cr);
	}
	ll qry2(int cur,int l,int r,int cl,int cr) {
		if(!cur) return 0;
		if(cl<=l&&r<=cr) return sum2[cur];
		int mid=(l+r)>>1;
		if(cr<=mid) return qry2(ls[cur],l,mid,cl,cr);
		if(cl>mid) return qry2(rs[cur],mid+1,r,cl,cr);
		return qry2(ls[cur],l,mid,cl,cr)+qry2(rs[cur],mid+1,r,cl,cr);
	}
}

namespace DS2 {
	int sum[N*20];
	int rt[N],ls[N*20],rs[N*20],tot;
	void clr() {
		for(int i=1;i<=n;i++) rt[i]=0;
		for(int i=1;i<=tot;i++) ls[i]=rs[i]=sum[i]=0;
		tot=0;
	}
	void push_up(int cur) {
		sum[cur]=0;
		if(ls[cur]) sum[cur]=sum[ls[cur]];
		if(rs[cur]) sum[cur]+=sum[rs[cur]];
	}
	void upt(int &cur,int l,int r,int pos,int v) {
		if(!cur) cur=++tot;
		if(l==r) {
			sum[cur]=v; return ;
		}
		int mid=(l+r)>>1;
		if(pos<=mid) upt(ls[cur],l,mid,pos,v);
		else upt(rs[cur],mid+1,r,pos,v);
		push_up(cur);
	}
	int qry(int cur,int l,int r,int cl,int cr) {
		if(!cur) return 0;
		if(cl<=l&&r<=cr) return sum[cur];
		int mid=(l+r)>>1;
		if(cr<=mid) return qry(ls[cur],l,mid,cl,cr);
		if(cl>mid) return qry(rs[cur],mid+1,r,cl,cr);
		return qry(ls[cur],l,mid,cl,cr)+qry(rs[cur],mid+1,r,cl,cr);
	}
}

signed main() {
	cin.tie(0); ios::sync_with_stdio(false);
	cin>>n; m=0;
	for(int i=0;(1<<i)<=n;i++) m=i;
	for(int i=1;i<=n;i++) cin>>v[i]; 
	for(int i=2;i<=n;i++) {
		int x; cin>>x; g[x].pb(i);
	}
	dfs0(1,0);
	cin>>q;
	for(int i=1;i<=q;i++) {
		int x,K; cin>>x>>K; ++K; 
		int nw=0;
		for(int j=m;j>=0;j--) {
			if((K>>j)&1) {
				vec[j].pb(node(x,dep[x]+nw,nw,i));
				nw+=(1<<j);
			}
		}
	}
	
	for(int i=0;i<=m;i++) {
		if(!i) {
			for(int j=1;j<=n;j++) {
				d[0][j]=v[j]; num[0][j]=1;
				for(int k=0;k<=m;k++) {
					if((v[j]>>k)&1) ++t[0][j][k];
				}
			}
			DS1::clr(); DS2::clr();
			for(int j=1;j<=n;j++) DS1::upt(DS1::rt[dep[j]],1,n,id[j],d[i][j]),DS1::upt2(DS1::rt[dep[j]],1,n,id[j],num[i][j]);
			for(int j=1;j<=n;j++) DS2::upt(DS2::rt[dep[j]],1,n,id[j],t[i&1][j][i]);
			for(int j=1;j<=n;j++) {
				int qwq=DS1::qry2(DS1::rt[dep[j]+(1<<i)],1,n,id[j],id[j]+sz[j]-1)-2*DS2::qry(DS2::rt[dep[j]+(1<<i)],1,n,id[j],id[j]+sz[j]-1);
				d[i+1][j]=d[i][j]+DS1::qry(DS1::rt[dep[j]+(1<<i)],1,n,id[j],id[j]+sz[j]-1)+1ll*qwq*(1ll<<i);
				num[i+1][j]=num[i][j]+DS1::qry2(DS1::rt[dep[j]+(1<<i)],1,n,id[j],id[j]+sz[j]-1);
//				cout<<num[i+1][j]<<' ';
			}
//			cout<<'\n';
			for(int k=0;k<=m;k++) {
				DS2::clr();
				for(int j=1;j<=n;j++) DS2::upt(DS2::rt[dep[j]],1,n,id[j],t[i&1][j][k]);
				for(int j=1;j<=n;j++)
					t[(i&1)^1][j][k]+=t[i&1][j][k]+DS2::qry(DS2::rt[dep[j]+(1<<i)],1,n,id[j],id[j]+sz[j]-1);
			}
			for(auto j:vec[i]) {
				ans[j.id]+=DS1::qry(DS1::rt[j.dep],1,n,id[j.x],id[j.x]+sz[j.x]-1);
			}
			for(int k=0;k<=m;k++) {
				DS2::clr();
				for(int j=1;j<=n;j++) DS2::upt(DS2::rt[dep[j]],1,n,id[j],t[i&1][j][k]);
				for(auto j:vec[i]) {
					if((j.add>>k)&1) {
						int qwq=DS1::qry2(DS1::rt[j.dep],1,n,id[j.x],id[j.x]+sz[j.x]-1)-2*DS2::qry(DS2::rt[j.dep],1,n,id[j.x],id[j.x]+sz[j.x]-1);
//						cout<<i<<" "<<j.id<<" "<<j.x<<" "<<j.dep<<" "<<DS2::qry(DS2::rt[j.dep],1,n,id[j.x],id[j.x]+sz[j.x]-1)<<"mmm\n";
//						cout<<qwq<<'\n';
//						cout<<j.add<<" "<<(1<<i)<<'\n';
						ans[j.id]+=1ll*qwq*(1ll<<k);
					}
				}
			}
			continue ;
		}
		DS1::clr(); DS2::clr();
		for(int j=1;j<=n;j++) DS1::upt(DS1::rt[dep[j]],1,n,id[j],d[i][j]),DS1::upt2(DS1::rt[dep[j]],1,n,id[j],num[i][j]);
		for(int j=1;j<=n;j++) DS2::upt(DS2::rt[dep[j]],1,n,id[j],t[i&1][j][i]);
		for(int j=1;j<=n;j++) {
			int qwq=DS1::qry2(DS1::rt[dep[j]+(1<<i)],1,n,id[j],id[j]+sz[j]-1)-2*DS2::qry(DS2::rt[dep[j]+(1<<i)],1,n,id[j],id[j]+sz[j]-1);
//			cout<<DS3::qry(DS3::rt[dep[j]+(1<<i)],1,n,id[dep[j]+(1<<i)],id[j]+sz[j]-1)<<'\n';
			d[i+1][j]=d[i][j]+DS1::qry(DS1::rt[dep[j]+(1<<i)],1,n,id[j],id[j]+sz[j]-1)+1ll*qwq*(1ll<<i);
			num[i+1][j]=num[i][j]+DS1::qry2(DS1::rt[dep[j]+(1<<i)],1,n,id[j],id[j]+sz[j]-1);
		}
		for(int k=0;k<=m;k++) {
			DS2::clr();
			for(int j=1;j<=n;j++) DS2::upt(DS2::rt[dep[j]],1,n,id[j],t[i&1][j][k]);
			for(int j=1;j<=n;j++)
				t[(i&1)^1][j][k]=t[i&1][j][k]+DS2::qry(DS2::rt[dep[j]+(1<<i)],1,n,id[j],id[j]+sz[j]-1);
		}
		for(auto j:vec[i]) {
			ans[j.id]+=DS1::qry(DS1::rt[j.dep],1,n,id[j.x],id[j.x]+sz[j.x]-1);
//			cout<<j.dep<<" "<<j.x<<" "<<DS1::qry(DS1::rt[j.dep],1,n,id[j.x],id[j.x]+sz[j.x]-1)<<'\n';
		}
		for(int k=0;k<=m;k++) {
			DS2::clr();
			for(int j=1;j<=n;j++) DS2::upt(DS2::rt[dep[j]],1,n,id[j],t[i&1][j][k]);
			for(auto j:vec[i]) {
				if((j.add>>k)&1) {
					int qwq=DS1::qry2(DS1::rt[j.dep],1,n,id[j.x],id[j.x]+sz[j.x]-1)-2*DS2::qry(DS2::rt[j.dep],1,n,id[j.x],id[j.x]+sz[j.x]-1);
//					cout<<DS3::qry(DS3::rt[j.dep],1,n,id[j.x],id[j.x]+sz[j.x]-1)<<'\n';
//					cout<<"mmm\n";
					ans[j.id]+=1ll*qwq*(1ll<<k);
				}
			}
		}
	}
	for(int i=1;i<=q;i++) cout<<ans[i]<<'\n';
	return 0;
}
Source code, Language: C++14

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 10
Accepted
time: 68ms
memory: 6164kb

input:

2000
946347867 341162491 202762650 295215762 254064439 66267204 693162580 357612557 492940637 939526638 59775103 374919339 144042807 861369313 651043728 999024805 439554916 167782038 597475252 56704409 69846137 22185655 79439847 769194737 145071391 226046618 915359433 392527325 84982946 54584098 827...

output:

31727996563
558747809
12352523
864526657694
91729843882
585197158
1628586765
922104954
8179034002
55032862871
1832814618
1719999173
57622639685
2418952903
118234180921
87152233480
403506301009
1458062157
139639462122
708763390
974312741
22160481283
541384937
113212700648
755479297
465075002
51201629...

result:

ok 1999 numbers

Test #2:

score: 0
Runtime Error

input:

99999
792064428 473106195 799314986 65440734 948726345 442414256 280245405 873012700 466192412 899092866 283555341 657824017 963883713 793944180 767444438 105576842 542107696 580140098 65321660 381184238 584604194 397414881 861590124 309323011 217641157 120832524 303744205 961590116 110259426 380351...

output:

result:

Test #3:

score: 0
Runtime Error

input:

1000000
947727010 179844140 923847675 171881267 5552129 974443359 989307850 869400987 126992154 527448411 141137718 136124474 917813105 392020809 79012342 473860575 969007624 833563354 90169336 878445705 84352622 403307122 733751738 670851448 942399068 731541999 101293644 545785337 964751520 9168003...

output:

result:

Test #4:

score: 0
Runtime Error

input:

1000000
264862971 751386013 921867736 711577153 262726588 565608444 975324815 440219681 107888226 928241413 729126923 283912914 86248857 896446999 12839598 651796991 139813366 105131395 341646170 839485925 939265720 844548518 102280410 457829889 8602879 737140565 17206920 974175632 535833885 8373832...

output:

result:

Test #5:

score: 0
Runtime Error

input:

1000000
978606419 773027389 111265179 979476504 280718851 476857501 751854950 579263969 848569548 781051974 31782627 533831186 812822170 111553645 297770650 331144396 676977983 2236128 258203325 75591120 676466973 60056446 494411414 286185093 92474576 173276071 535648669 87210101 355790411 880267291...

output:

result:

Test #6:

score: 0
Runtime Error

input:

1000000
952470566 585754087 120174600 401525004 458588768 5487567 31210348 446333263 231409083 521960132 457721893 866842852 925207283 16805978 4706826 99640835 619272676 136536623 459247161 308807462 633687300 717271369 23906473 865522890 173799280 424309108 719410673 118906385 110627845 730629403 ...

output:

result:

Test #7:

score: 0
Runtime Error

input:

1000000
732367509 105027907 958920212 886798715 102486738 813075884 301085392 242303497 979657287 944859684 307768 438158233 561755409 740706505 791145209 283862713 828081846 771569552 59044985 600549571 191330226 438693570 36976319 810654215 220068818 771875421 740642902 839964155 206129566 2065543...

output:

result:

Test #8:

score: 0
Runtime Error

input:

1000000
0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 ...

output:

result:

Test #9:

score: 0
Runtime Error

input:

1000000
465660691 982007525 816592310 377030959 572981469 679249520 86377999 709561525 940473306 35102782 886143915 792819787 903287397 264564177 857982095 91486434 217197704 123118964 383387342 820268798 497623987 255010796 607884194 848568529 38169627 197987657 421323589 664004905 485409127 696844...

output:

result:

Test #10:

score: 0
Runtime Error

input:

1000000
665830082 788228483 245541444 289601309 641764988 150723484 925214020 557415731 310210969 379707835 517820381 883917428 134445288 775557009 444476671 89856268 655841087 888410254 37788122 694551869 563331754 488108584 839551943 415095075 445425438 35452604 562044723 640544531 146258096 66852...

output:

result:
English 繁體中文 简体中文

Server Time: 2024-11-30 13:24:48 | OpenSource Project