QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#714977#9492. 树上简单求和NKheyuxiang5 68ms29052kbC++143.2kb2024-11-06 09:36:212024-11-06 09:36:22

Judging History

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

  • [2024-11-06 09:36:22]
  • 评测
  • 测评结果:5
  • 用时:68ms
  • 内存:29052kb
  • [2024-11-06 09:36:21]
  • 提交

answer

#include<bits/stdc++.h>
#define N 400005
#define fi first
#define se second
#define mp make_pair
using namespace std;
int n,m;
unsigned long long a[N],val[N];
struct tree{
	int h[N],to[N],nxt[N],cnt;
	void jb(int u,int v){
		to[++cnt]=v;
		nxt[cnt]=h[u];
		h[u]=cnt;
	}
	int id[N],pl[N],pr[N],tot;
	int up[N][19],dep[N];
	void dfs(int u,int fa){
		val[u]=a[u]+val[fa];
		id[tot]=u;
		pl[u]=tot;
		++tot;
		dep[u]=dep[fa]+1;
		up[u][0]=fa;
		for(int i=1;i<=17;i++)
			up[u][i]=up[up[u][i-1]][i-1];
		for(int i=h[u];i!=0;i=nxt[i])
			if(to[i]!=fa) dfs(to[i],u);
		id[tot]=u;
		pr[u]=tot;
		tot++;
	}
	int lca(int x,int y){
		if(dep[x]>dep[y]) swap(x,y);
		for(int i=17;i>=0;i--)
			if(dep[up[y][i]]>=dep[x]) y=up[y][i];
		if(x==y) return x;
		for(int i=17;i>=0;i--)
			if(up[x][i]!=up[y][i])
				x=up[x][i],y=up[y][i];
		return up[x][0];
	}
	void work(){
		for(int i=1;i<n;i++){
			int x,y;
			scanf("%d%d",&x,&y);
			jb(x,y);
			jb(y,x);
		}
		dfs(1,0);
	}
}T1,T2;
const int t=1;
unsigned long long k[N],ans[N];
pair<int ,int > qt1[N][2],qt2[N][2];
int pre[N];
void init(int l,int r){
	for(int i=0;i<=2*n;i++) pre[i]=0;
	for(int i=l;i<=r;i++){
		int u=T2.id[i],f=(T2.pl[u]==i?1:-1);
		pre[T1.pl[u]+1]+=f;
		pre[T1.pr[u]+1]-=f;
	}
	for(int i=1;i<=2*n;i++) pre[i]+=pre[i-1];
}
unsigned long long s0[N],s1[N];
void add(pair<int ,int > p,unsigned long long ad){
	int idl=(p.fi-1)/t+1,idr=(p.se+1)/t-1;
	if(idl>idr){
		for(int i=p.fi;i<=p.se;i++) s0[i]+=ad;
	}
	else{
		for(int i=p.fi;i<idl*t;i++) s0[i]+=ad;
		for(int i=idr*t+t;i<=p.se;i++) s0[i]+=ad;
		for(int i=idl;i<=idr;i++) s1[i]+=ad;
	}
}
unsigned long long qry(pair<int ,int > p){
	int idl=(p.fi-1)/t+1,idr=(p.se+1)/t-1;
	if(p.se==2*n) idr=2*n/t-1;
	unsigned long long res=0;
	if(idl>idr){
		for(int i=p.fi;i<=p.se;i++){
			int u=T2.id[i];
			unsigned long long w=s0[T1.pl[u]]+s1[T1.pl[u]/t]-s0[T1.pr[u]]-s1[T1.pr[u]/t];
			if(T2.pl[u]==i) res+=w;
			else res-=w;
		}
	}
	else{
		for(int i=p.fi;i<idl*t;i++){
			int u=T2.id[i];
			unsigned long long w=s0[T1.pl[u]]+s1[T1.pl[u]/t]-s0[T1.pr[u]]-s1[T1.pr[u]/t];
			if(T2.pl[u]==i) res+=w;
			else res-=w;
		}
		for(int i=idr*t+t;i<=p.se;i++){
			int u=T2.id[i];
			unsigned long long w=s0[T1.pl[u]]+s1[T1.pl[u]/t]-s0[T1.pr[u]]-s1[T1.pr[u]/t];
			if(T2.pl[u]==i) res+=w;
			else res-=w;
		}
	}
	return res;
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) scanf("%llu",&a[i]);
	T1.work();
	T2.work();
	for(int i=1;i<=m;i++){
		int x,y;
		scanf("%d%d%llu",&x,&y,&k[i]);
		int lc1=T1.lca(x,y);
		int lc2=T2.lca(x,y);
		qt1[i][0]=mp(T1.pl[lc1]+1,T1.pl[x]);
		qt1[i][1]=mp(T1.pl[lc1],T1.pl[y]);
		qt2[i][0]=mp(T2.pl[lc2]+1,T2.pl[x]);
		qt2[i][1]=mp(T2.pl[lc2],T2.pl[y]);
		add(qt1[i][0],k[i]);
		add(qt1[i][1],k[i]);
		ans[i]=val[x]+val[y]-val[lc2]*2+a[lc2]+qry(qt2[i][0])+qry(qt2[i][1]);
	}
	for(int i=0;i*t<n*2;i++){
		int l=i*t,r=min(n*2-1,i*t+t-1);
		init(l,r);
		unsigned long long sum=0;
		for(int j=1;j<=m;j++){
			sum+=k[j]*(pre[qt1[j][0].se+1]-pre[qt1[j][0].fi]+pre[qt1[j][1].se+1]-pre[qt1[j][1].fi]);
			if(qt2[j][0].fi<=l&&r<=qt2[j][0].se) ans[j]+=sum;
			if(qt2[j][1].fi<=l&&r<=qt2[j][1].se) ans[j]+=sum;
		}
	}
	for(int i=1;i<=m;i++) printf("%llu\n",ans[i]);
}

詳細信息

Subtask #1:

score: 5
Accepted

Test #1:

score: 5
Accepted
time: 61ms
memory: 27104kb

input:

3000 3000
7236742292501328495 17973811477309806363 16075782662531676171 17971236571771878676 11392080645527132110 3685563455925680459 9773593720088356683 8313828403245053795 7736401634567449043 1634817828009987181 6951124933529719486 12775126714635387213 15460977209223753216 397573676785925632 31372...

output:

12105153858659381124
18367442707572066757
11668241962484097878
11288238120352358472
1742468310074622166
9942835997686093671
3305677510569607477
17741602000425004088
14984128302052618266
1075081718074605786
6509217537832509095
16750513627843273113
8569443169249732820
14475184194298579044
156111071108...

result:

ok 3000 lines

Test #2:

score: 5
Accepted
time: 68ms
memory: 29052kb

input:

3000 3000
1612333876155866602 8538417838700679227 6080765231437578796 17905224638340228394 12270907925903144224 17944105326358594564 17302041033966840611 1006351124625222126 496336153231744288 9393087977687876980 9553975238547373621 9361882717200384390 15051881329169144319 9757999873162420435 882725...

output:

11133131376095771981
7909873024850695144
16250639243139481926
14562550655578101207
8274205996508264973
178549413271904466
2368406276743327913
7464009386554813982
9439464815411774627
1471756740732097060
15201641099137019227
6774030298556871576
18156105511913219667
1553508745644446823
4225137078364117...

result:

ok 3000 lines

Test #3:

score: 5
Accepted
time: 60ms
memory: 27016kb

input:

3000 3000
9709246061666095435 1861649101703072889 10620139893353930613 17635186539135419482 710209455559527146 6075101384669982511 1120305006358459674 9703156967435388252 1397046737759839382 5259056712870179169 8253156305433022999 710199693203327302 15130650033641744675 10720111924616886955 15543351...

output:

7834604406305153073
5037061270969117785
16481572776620825702
15177894197606565804
3120320619896892806
18008650876379132344
7417108723176816402
13515164814425439399
3299769942258542105
15897528270699011770
11642805469843844638
16764682282380318054
4824039114054405772
4859834102876213962
1234210473247...

result:

ok 3000 lines

Test #4:

score: 5
Accepted
time: 60ms
memory: 24996kb

input:

3000 3000
16538965545220923528 18062192327708400751 10422465150728338588 3471522151129113073 1236650672072793692 1942240200040301168 13090729759591037952 15335798523677372669 9912100622761466753 11177948788405690381 3710859061697501523 4984944638666762977 17278589713462878008 6371292801024547050 868...

output:

8182453933067329108
13535217473847106938
17067385337010269798
3806121648880466130
11322569288575153037
11079197311131660121
9670138330007803226
6554062218199796758
965954569567598779
18055887214749050688
6142620503089407421
8690117812667761187
9547139298346295115
8890987597519353054
1755036654049586...

result:

ok 3000 lines

Test #5:

score: 5
Accepted
time: 61ms
memory: 28996kb

input:

3000 3000
17759588706587888497 10550000524636484378 11601004513528075994 7150322911283804521 4459707248078569712 10692395730842402625 8940418793863522991 12967068928670540447 9954278250450015940 13702413838608801301 10598390500439869870 15110245227553613794 490862872212325709 15164980555660957366 94...

output:

9743736929788175512
16812303667256960040
14694223512340829897
550204232580650311
1175342872438242313
17622261358285047637
7413682703975031220
12643066512274062227
1868985217436232595
5471830334855681322
8070132260376389587
3970361922096052085
218281824643752746
991917103472727104
2960248244218479023...

result:

ok 3000 lines

Subtask #2:

score: 0
Time Limit Exceeded

Dependency #1:

100%
Accepted

Test #6:

score: 12
Accepted
time: 0ms
memory: 28464kb

input:

5 7
0 3 2 6 4
1 2
2 4
1 5
5 3
3 4
4 2
2 5
5 1
5 3 0
3 2 5
4 4 4
4 4 3
5 2 0
3 4 3
5 5 6

output:

15
21
10
13
17
26
18

result:

ok 7 lines

Test #7:

score: 0
Time Limit Exceeded

input:

70000 70000
3805295436278888199 9842309351516174725 1566744796319231180 2206519284152256579 2715928675931950447 6346821976624501261 16020972671480798719 14702021753902144915 17127828773798978442 15779168055669690475 4964561323934614661 9395102787554964450 6377076753365184543 15167378195767668817 288...

output:


result:


Subtask #3:

score: 0
Skipped

Dependency #2:

0%

Subtask #4:

score: 0
Time Limit Exceeded

Test #21:

score: 0
Time Limit Exceeded

input:

200000 200000
622783158027686223 2242697872372232537 8481648430436878777 10092474834140799044 15403999682625301609 12614289513474949582 9180944589267018841 7823784919308285798 8257785171198951273 5134508521895120821 8041682272181381093 3835432206618893170 2653803171409877650 5589823419153460372 1007...

output:


result:


Subtask #5:

score: 0
Time Limit Exceeded

Test #27:

score: 0
Time Limit Exceeded

input:

200000 200000
1958469220619413759 14991498002015735322 6054491201406941902 18206143187746582567 15082377615826460430 2936248617457291604 10073577150351675920 16534472678586906457 2207599132486246393 10301540360769075442 1492580560381080472 551692353431379140 13238280352539145808 8462626987240986565 ...

output:


result:


Subtask #6:

score: 0
Time Limit Exceeded

Test #34:

score: 0
Time Limit Exceeded

input:

200000 200000
6794776813641982926 1561596256197101737 10910039723053043515 7892247858295192798 12233819960547881004 17695389034783066733 9173201689566865598 17626618141377486739 7358781671024283919 6787559733384974662 3884392438269280436 14872846228351316833 9037842441501571648 14299818404271084016 ...

output:


result:


Subtask #7:

score: 0
Skipped

Dependency #1:

100%
Accepted

Dependency #2:

0%