QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#409311#5830. 树zyz07100 ✓1691ms519248kbC++172.0kb2024-05-11 21:41:522024-05-11 21:41:53

Judging History

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

  • [2024-05-11 21:41:53]
  • 评测
  • 测评结果:100
  • 用时:1691ms
  • 内存:519248kb
  • [2024-05-11 21:41:52]
  • 提交

answer

#include <bits/stdc++.h>
using namespace std;
#define For(Ti,Ta,Tb) for(auto Ti=(Ta);Ti<=(Tb);++Ti)
#define Dec(Ti,Ta,Tb) for(auto Ti=(Ta);Ti>=(Tb);--Ti)
#define debug(...) fprintf(stderr,__VA_ARGS__)
#define range(Tx) begin(Tx),end(Tx)
using ll=long long;
const int N=1e6+5;
int n,q,a[N],fa[N],dep[N],pre[N],desc[N][20],g[N][20][2];
ll f[N][20];
vector<int> e[N],deps[N];
ll query(int u,int k){
	ll ans=0;
	Dec(j,19,0){
		if(k>>j&1){
			int v=desc[u][j];
			ans+=f[u][j]-(ll(g[v][j][1])<<j)+(ll(g[v][j][0])<<j);
			u=v;
		}
	}
	For(j,0,19){
		if(k>>j&1){
			ans+=(ll(g[u][j][1])<<j)-(ll(g[u][j][0])<<j);
		}
	}
	return ans;
}
int main(){
	cin.tie(nullptr)->sync_with_stdio(false);
	cin>>n;
	For(i,1,n){
		cin>>a[i];
	}
	For(i,2,n){
		cin>>fa[i];
		e[fa[i]].push_back(i);
		dep[i]=dep[fa[i]]+1;
	}
	{
		queue<int> q;
		q.push(1);
		while(q.size()){
			int u=q.front();
			q.pop();
			pre[u]=(deps[dep[u]].size()?deps[dep[u]].back():0);
			deps[dep[u]].push_back(u);
			for(int v:e[u]){
				q.push(v);
			}
		}
	}
	For(i,1,n){
		for(int u:deps[i]){
			desc[fa[u]][0]=u;
		}
	}
	Dec(i,n,0){
		for(int u:deps[i]){
			if(!desc[u][0]){
				desc[u][0]=desc[pre[u]][0];
			}
			For(j,1,19){
				desc[u][j]=desc[desc[u][j-1]][j-1];
			}
		}
	}
	Dec(i,n,0){
		for(int u:deps[i]){
			For(j,0,19){
				For(x,0,1){
					g[u][j][x]=((a[u]>>j&1)==x);
					for(int v:e[u]){
						g[u][j][x]+=g[v][j][x];
					}
				}
			}
		}
	}
	Dec(i,n,0){
		for(int u:deps[i]){
			For(j,0,19){
				For(x,0,1){
					g[u][j][x]+=g[pre[u]][j][x];
				}
			}
		}
	}
	Dec(i,n,0){
		for(int u:deps[i]){
			f[u][0]=f[pre[u]][0]+a[u];
			For(j,1,19){
				int v=desc[u][j-1],v_=desc[v][j-1];
				f[u][j]=f[u][j-1]+f[v][j-1];
				f[u][j]-=(1LL*g[v][j-1][1]-g[v_][j-1][1])<<(j-1);
				f[u][j]+=(1LL*g[v][j-1][0]-g[v_][j-1][0])<<(j-1);
			}
		}
	}
	cin>>q;
	For($,1,q){
		int u,k;
		cin>>u>>k;
		cout<<query(u,k+1)-query(pre[u],k+1)<<'\n';
	}
	return 0;
}

詳細信息

Test #1:

score: 10
Accepted
time: 4ms
memory: 61816kb

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: 10
Accepted
time: 76ms
memory: 103256kb

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:

2509771019
102808896745
528899856
125427731
1953894159
2464509509394
5038222699950
586671581
4009894465921
798898621
468901916
1620113250
25990778
1594379152
3312692683691
30663600697267
1721547881
1491500623
2600435021950
15307884508814
263989598
1350176621
1471646893657
727716410
2234098283207
184...

result:

ok 100000 numbers

Test #3:

score: 10
Accepted
time: 1691ms
memory: 519248kb

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:

322288180595345
64144900444023
131621992874362
185914449206495
4835864536999
52498320047300
91107191167946
133549265294890
287851078359140
123171067016186
128593134440538
8701280311335
14021811618693
65043919486113
310510908841731
15605852792829
75072265994522
73630260235120
66821009550818
217424650...

result:

ok 1000000 numbers

Test #4:

score: 10
Accepted
time: 1593ms
memory: 519164kb

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:

1437301063221
97809309869545
338628270305337
233680181746714
7398781587300
271306073103235
20877473367084
90775179215463
2535173725689
66604505519276
211534325021331
84757341592614
17886216279466
132051595671397
1860377236584
46253790496717
83950882453932
257274371637175
240943229927301
118446604805...

result:

ok 1000000 numbers

Test #5:

score: 10
Accepted
time: 869ms
memory: 485784kb

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:

2258826661
938786622
1803221827
20412967786
11943681449
2783437365
16665953904
2720451454
1859550937
476794111
986063591
1579646962
14494868585
38176990
781099105
527479272
2392930136
2519423502
1620138021
23553905552
5107065078
6903424087
1206547828
18830610755
49589751
499629803
4275328259
1379311...

result:

ok 1000000 numbers

Test #6:

score: 10
Accepted
time: 908ms
memory: 486316kb

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:

994051214
904057442
2995278795
3980490219156
35388089564067
976251948
369701890
43760580
1510389651
3077329408
66354186603909
716198149
190685967
105215742257612
13827392197052
557364187
1084839575
3104092
276690929
98459972717615
467852673
33049527
425554649
22790861096838
97736610924937
426018723
...

result:

ok 1000000 numbers

Test #7:

score: 10
Accepted
time: 891ms
memory: 486748kb

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:

999908753
401770479
206278425790162
194002588398182
57017982421941
107873913740677
1146711843
100798039882905
2910590132
4981801495
1079679207
2140524944
75514756
9410084351477
74281057669756
167205592
58363655788956
1085743066
51861395060073
416905260757083
99153021379917
990736484
4798942061
37301...

result:

ok 1000000 numbers

Test #8:

score: 10
Accepted
time: 1084ms
memory: 486448kb

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:

3826404725
4857376598
25168931
28
7223089410
64321117
0
334067473
60992
1
3
1
1
1
1
0
672988753
1
0
3075365115
8
0
0
3
0
320487913
1
0
0
0
0
423617904
687492110
0
301348
220115086
10
35419106
37297971
1
1
28561996
517977935
3
2643348428
0
1132906086
2
1
4
20029285275
6
6191566989
0
20341562923
93968...

result:

ok 1000000 numbers

Test #9:

score: 10
Accepted
time: 1049ms
memory: 486884kb

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:

96094116015
8243992183131
82893024196511
34208171180799
275400654
1384234654
64716568
11773600395352
10114549437012
263845831954
15632079623816
528398853
1456723425
1214744688
12973183499599
23475037
28419069552831
245789356
109738951
48543618600613
81139248787313
102619945
17650377918169
1556076353...

result:

ok 1000000 numbers

Test #10:

score: 10
Accepted
time: 1031ms
memory: 487400kb

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:

431856043
520373616
77562434895287
7077779085197
63182570356266
13801713959444
762015332
4985151936205
1184478776
594507511
2675566549865
25669163666486
306250439
62500137
75019025742612
786078000
31380423590936
4162775188
4984310933262
877965179
18603382047372
1013123764
8945453173145
5683574410067...

result:

ok 1000000 numbers

Extra Test:

score: 0
Extra Test Passed