QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#670260#9492. 树上简单求和Naganohara_Yoimiya17 2066ms235940kbC++206.3kb2024-10-23 21:04:062024-10-23 21:04:06

Judging History

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

  • [2024-10-23 21:04:06]
  • 评测
  • 测评结果:17
  • 用时:2066ms
  • 内存:235940kb
  • [2024-10-23 21:04:06]
  • 提交

answer

#include<bits/stdc++.h>

#define ll long long
#define mk make_pair
#define fi first
#define se second

#define ull unsigned long long

using namespace std;

#define gc getchar
inline int read(){
	int x=0;char c=getchar();
	for(;(c<'0'||c>'9');c=getchar());
	for(;(c>='0'&&c<='9');c=getchar())x=x*10+(c&15);
	return x;
}
inline ull readu(){
	ull x=0;char c=getchar();
	for(;(c<'0'||c>'9');c=getchar());
	for(;(c>='0'&&c<='9');c=getchar())x=x*10+(c&15);
	return x;
}

const int mod=998244353;
int ksm(int x,ll y,int p=mod){
	int ans=1;y%=(p-1);
	for(int i=y;i;i>>=1,x=1ll*x*x%p)if(i&1)ans=1ll*ans*x%p;
	return ans%p;
}
int inv(int x,int p=mod){return ksm(x,p-2,p)%p;}
mt19937 rnd(time(0));
int randint(int l,int r){return rnd()%(r-l+1)+l;}
void add(int &x,int v){x+=v;if(x>=mod)x-=mod;}
void Mod(int &x){if(x>=mod)x-=mod;}
int cmod(int x){if(x>=mod)x-=mod;return x;}

template<typename T>void cmax(T &x,T v){x=max(x,v);}
template<typename T>void cmin(T &x,T v){x=min(x,v);}

const int N=2e5+5;
const int B=205;
// const int NB=(N/B+5)*2
// const int NB=N;

int n,q;

struct HLD_Tree{
	vector<int>G[N];
	int dep[N],dfn[N],top[N],hson[N],sz[N],fa[N],dfc;
	void dfs1(int u){
		dep[u]=dep[fa[u]]+1,sz[u]=1;
		for(int v:G[u])if(v!=fa[u]){
			fa[v]=u,dfs1(v),sz[u]+=sz[v];
			if(sz[v]>sz[hson[u]])hson[u]=v;
		}
	}
	void dfs2(int u,int tp){
		dfn[u]=++dfc,top[u]=tp;
		if(hson[u])dfs2(hson[u],tp);
		for(int v:G[u])if(v!=fa[u]&&v!=hson[u])dfs2(v,v);
	}
	void build(){dfs1(1),dfs2(1,1);}
	int LCA(int u,int v){
		while(top[u]!=top[v]){
			if(dep[top[u]]<dep[top[v]])swap(u,v);
			u=fa[top[u]];
		}
		if(dep[u]>dep[v])swap(u,v);
		return u;
	}
}T1,T2;

int dep[N],mxdis[N],dfn[N],dfc;
vector<int>G[N],T[N];

void dfs_dep(int u,int fa){
	// cout<<"dfs_dep "<<u<<" "<<fa<<endl;
	mxdis[u]=0,dfn[u]=++dfc;
	for(int v:G[u])if(v!=fa)dep[v]=dep[u]+1,dfs_dep(v,u),cmax(mxdis[u],mxdis[v]+1);
}
bool In[N];
vector<int>nodes;

vector<vector<int> >cc;
int Id[N],ccnt;
void dfs_cc(int u,int fa,int id){
	cc[id][u]+=cc[id][fa];
	for(int v:T[u])if(v!=fa)dfs_cc(v,u,id);
}

void build(){
	dfs_dep(1,0);
	int mxdep=0;
	for(int i=1;i<=n;i++)cmax(mxdep,dep[i]);
	// cerr<<"mxdep = "<<mxdep<<endl;
	// cout<<"dep = ";for(int i=1;i<=n;i++)cout<<dep[i]<<" ";puts("");
	// cout<<"mxdis = ";for(int i=1;i<=n;i++)cout<<mxdis[i]<<" ";puts("");
	for(int i=1;i<=n;i++)if(dep[i]%B==0&&mxdis[i]>=B)nodes.emplace_back(i);
	// cout<<"now nodes = ";for(int x:nodes)cout<<x<<" ";puts("");

	sort(nodes.begin(),nodes.end(),[&](int x,int y){return dfn[x]<dfn[y];});
	vector<int>new_nodes=nodes;
	for(int i=0;i+1<nodes.size();i++)new_nodes.emplace_back(T1.LCA(nodes[i],nodes[i+1]));
	auto rec=[&](vector<int>&A){sort(A.begin(),A.end()),A.resize(unique(A.begin(),A.end())-A.begin());};
	new_nodes.emplace_back(1);rec(new_nodes);nodes=new_nodes;
	for(int x:nodes)In[x]=true;

	// assert(nodes.size()<=NB);
	ccnt=nodes.size();
	cc.resize(ccnt,vector<int>(n+1,0));

	// cerr<<"ccnt = "<<ccnt<<endl;
	// cerr<<"nodes = ";for(int x:nodes)cerr<<x<<" ";cerr<<endl;

	for(int i=1;i<=n;i++)Id[i]=-1;
	for(int i=0;i<nodes.size();i++){
		// cout<<""
		int x=nodes[i],y=x;Id[x]=i;
		while(y)cc[i][y]=1,y=T1.fa[y];
		dfs_cc(1,0,i);
		// cout<<"cc "<<i<<" = ";for(int j=1;j<=n;j++)cout<<cc[i][j]<<" \n"[j==n];
	}
}
ull tag[N];

namespace ds{

const int C=605;
const int NC=N/C+5;

int bl[N],L[NC],R[NC];
ull sum[NC],val[N];

void build(){
	memset(L,63,sizeof(L));
	for(int i=1;i<=n;i++)bl[i]=(i-1)/C+1;
	for(int i=1;i<=n;i++)cmax(R[bl[i]],i),cmin(L[bl[i]],i);
}

void Add(int x,ull v){if(1<=x&&x<=n)sum[bl[x]]+=v,val[x]+=v;}
ull Sum(int l,int r){
	ull ans=0;
	if(l>r)return ans;
	if(bl[l]==bl[r]){for(int i=l;i<=r;i++)ans+=val[i];return ans;}
	for(int i=l;i<=R[bl[l]];i++)ans+=val[i];
	for(int i=L[bl[r]];i<=r;i++)ans+=val[i];
	for(int i=bl[l]+1;i<=bl[r]-1;i++)ans+=sum[i];
	return ans;
}

}

ull a[N];
void Add_1(int x,ull v){
	// cout<<"Add "<<x<<" "<<v<<endl;
	// cout<<"Add_1 "<<x<<" "<<v<<" -> dfn = "<<T2.dfn[x]<<" sz = "<<T2.sz[x]<<endl;
	ds::Add(T2.dfn[x],v),ds::Add(T2.dfn[x]+T2.sz[x],-v);
	// F.add(T2.dfn[x],v),F.add(T2.dfn[x]+T2.sz[x],-v);
}
ull Qsum_1(int x,int y){
	int z=T2.LCA(x,y);
	// cout<<"Qsum "<<x<<" "<<y<<" = "<<ds::Sum(T2.dfn[z],T2.dfn[x])+ds::Sum(T2.dfn[z],T2.dfn[y])<<endl;
	// int dx=T2.dfn[x],dy=T2.dfn[y],dz=T2.dfn[z],df=T2.dfn[T2.fa[z]];
	return ds::Sum(T2.dfn[z]+1,T2.dfn[x])+ds::Sum(T2.dfn[T2.fa[z]]+1,T2.dfn[y]);
}

void Add(int x,int y,ull k){
	// cout<<"Add "<<x<<" "<<y<<" "<<k<<endl;
	int z=T1.LCA(x,y);
	while(x!=z&&Id[x]==-1)Add_1(x,k),x=T1.fa[x];
	while(y!=z&&Id[y]==-1)Add_1(y,k),y=T1.fa[y];
	if(x==y)return Add_1(x,k);

	// cerr<<"now x,y = "<<x<<" "<<y<<" Id = "<<Id[x]<<" "<<Id[y]<<endl;

	if(Id[x]!=-1&&Id[y]!=-1){
		assert(Id[z]!=-1);
		tag[Id[x]]+=k,tag[Id[y]]+=k,tag[Id[z]]-=k+k,Add_1(z,k);
	}
	else{
		if(Id[y]!=-1)swap(x,y);
		assert(Id[x]!=-1&&Id[y]==-1&&y==z);
		// cerr<<" -> x,y = "<<x<<" "<<y<<" Id = "<<Id[x]<<" "<<Id[y]<<endl;
		y=T1.fa[y];
		while(Id[y]==-1)Add_1(y,-k),y=T1.fa[y];
		// cerr<<" -> y = "<<y<<endl;
		tag[Id[x]]+=k,tag[Id[y]]-=k;
	}
}

ull Qsum(int x,int y){
	ull ans=0;
	int z=T2.LCA(x,y),fz=T2.fa[z];
	for(int i=0;i<ccnt;i++){
		int cnt=cc[i][x]+cc[i][y]-cc[i][z]-cc[i][fz];
		// cout<<"i = "<<i<<" tag = "<<tag[i]<<" cnt = "<<cnt<<endl;
		ans+=((ull)(cnt))*tag[i];
	}
	// cout<<"now ans = "<<ans<<endl;
	ans+=Qsum_1(x,y);
	return ans;
}

signed main(void){

#ifndef ONLINE_JUDGE
	// freopen("in.txt","r",stdin);
	// freopen("b.in","r",stdin);
	// freopen("mist04.in","r",stdin);
	// freopen("mist04.ans","w",stdout);
#endif

	// double st=clock();
	n=read(),q=read();
	for(int i=1;i<=n;i++)a[i]=readu();
	for(int i=1;i<=n-1;i++){
		int u=read(),v=read();
		// cout<<u<<" "<<v<<endl;
		G[u].emplace_back(v),G[v].emplace_back(u);
		T1.G[u].emplace_back(v),T1.G[v].emplace_back(u);
	}
	for(int i=1;i<=n-1;i++){
		int u=read(),v=read();
		T[u].emplace_back(v),T[v].emplace_back(u);
		T2.G[u].emplace_back(v),T2.G[v].emplace_back(u);
	}

	ds::build();

	T1.build(),T2.build(),build();
	for(int i=1;i<=n;i++)Add_1(i,a[i]);

	for(int i=1;i<=q;i++){
		int x=read(),y=read();ull k=readu();
		Add(x,y,k);
		cout<<Qsum(x,y)<<'\n';
	}

	// cerr<<"time = "<<clock()-st<<"ms\n";

	return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Subtask #1:

score: 5
Accepted

Test #1:

score: 5
Accepted
time: 7ms
memory: 26804kb

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: 5ms
memory: 20804kb

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: 7ms
memory: 27256kb

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: 3ms
memory: 27176kb

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: 9ms
memory: 25336kb

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: 12
Accepted

Dependency #1:

100%
Accepted

Test #6:

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

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: 12
Accepted
time: 298ms
memory: 41896kb

input:

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

output:

5971729064136092190
6457394048987305727
13604212649915736394
8639973959364892219
437861319070967556
16133076880026962355
5384937395694479961
4591478439775690843
16071919565966962790
15485626634068969082
10235993901046758372
3449528613427081475
8064280362779764074
12784984512326434905
424951714880051...

result:

ok 70000 lines

Test #8:

score: 12
Accepted
time: 251ms
memory: 48396kb

input:

70000 70000
17769190865915081913 3772925482507158804 10559962993069063712 16307277356502651642 12014171661057147061 1923543107882042577 13408785599350410314 17786178374951015816 2038922879833426794 2540043772647346461 15419977514837351390 5175974305273838292 16815288359165841441 6295059675346852046 ...

output:

16215781699519408534
17067966839552063165
1639359200259068228
1157756671621253300
12850966537933214537
13917563606289473282
11146906493479190751
869141055866285398
529460535280965984
11437720548737856517
12321579881011015953
4005153170897692243
10217866116994297464
8892403813874757974
12520505236760...

result:

ok 70000 lines

Test #9:

score: 12
Accepted
time: 641ms
memory: 101468kb

input:

70000 70000
1322605819855709761 1534349070722535975 3956030287626175223 12996546673549161162 7258680666490714729 15591023033141410544 11626890152249303179 7745771567168540351 5535931029756133379 11840793767439557739 6286106656048048381 9490665709724541446 4561258384162386434 2460318488748442222 1303...

output:

7565012138645637258
1080785033897684285
4000254219257999844
8727142139647715419
1784876728989450460
2474052717732723820
5108017366064709316
5232698473118606856
7893212823648229982
6449010654774296779
16571818815110297674
603759348329356530
7364528294111530037
4667545362378304836
3039728935129459889
...

result:

ok 70000 lines

Test #10:

score: 12
Accepted
time: 785ms
memory: 67756kb

input:

70000 70000
2918414982140182939 1004760492603077644 7526656799259998488 6665485253854847449 7752199419154649757 12763267823077347079 11745132191692540338 6726116817426709990 15550876907005962464 9760509858122842638 684733892856965421 10077915441058780247 8380400329996723109 16920573433866702239 3069...

output:

8230389499860859172
16425656898047941538
107743004356580170
9778122844868660722
11068387722102791183
13252614309136720348
15937842372230698728
12777338070107774364
17974062134369145560
3740400391792770609
7367804332878038809
14236246024207211797
5659238205278608512
10550373456364765526
3478082332928...

result:

ok 70000 lines

Test #11:

score: 12
Accepted
time: 2066ms
memory: 143108kb

input:

70000 70000
14167059704556856337 16190708842842354431 16763990539754009056 7631426709261583690 16701377874952853623 13128000186728267818 13668914249103068169 11444044591715948726 461080622438520919 15327533341012334586 15905150558482528923 18113008235210277231 18273290154232335325 871461822812191943...

output:

5416890687002400795
15434184693210288436
14994504916760087024
2057026449542829151
14782289435774270062
5375237679514404106
6242405047854012647
13176621545709355733
14860610197328732602
2320525143444929350
4955538191022622551
16072981679771537209
16493487770453132249
7457010288198365370
1095949888193...

result:

ok 70000 lines

Test #12:

score: 12
Accepted
time: 1060ms
memory: 94344kb

input:

70000 70000
6512290618577097706 2307104154841663907 18099814251235047570 8297332016606109910 6979819983598849680 18022671181330012408 7003320957516774041 10765303713874539785 15263207007138552812 11713955610641877995 9084887894280210904 3653718255996209121 14197591595561260765 2937670413926210256 43...

output:

5372775214253596890
927985558228810546
3829815088328182672
17496384540548895622
2541458359607440535
9685902106698191409
13649653134779075960
2952563488513208867
3457470079648848247
11542323450217419837
6576344363223624061
12316990756988470568
17923006133291073450
13069551524451668138
648013229980407...

result:

ok 70000 lines

Test #13:

score: 12
Accepted
time: 1211ms
memory: 94700kb

input:

70000 70000
13665984219894847790 9458613748861462697 7467746948118990839 10855454155004540952 10025433108785732161 15816172836312183738 2834129139700401667 221649423184372325 8409217794427284711 16119623676185869010 12488380095384700010 3049877130176336551 5805665682633632307 13224802542929355280 18...

output:

7934210059911784858
14305091721658406168
5803801684631217062
8806866881905382618
14997911434771439753
7006465422324293550
15394754861139766679
8377831978907312075
12227086919743533414
8784212755151945751
17039860679476902214
6474495685436520748
11136139762939837997
16869294577244011226
1164744311638...

result:

ok 70000 lines

Subtask #3:

score: 0
Time Limit Exceeded

Dependency #2:

100%
Accepted

Test #14:

score: 13
Accepted
time: 1673ms
memory: 110188kb

input:

120000 120000
4056283459929576306 2264755903151268173 1157390036441353969 5734735320959854923 6025999163810189446 3972481234804681969 4746636248696530169 6716674455256322787 6407347371842702902 7463142557880503801 208361219405474896 512530621977574257 6488145455921761864 6595856237657889728 95997703...

output:

11686658894480913739
10283716998652647869
7469817527336516079
9879285786875030762
4982549977394044949
16564334076086174008
6433057925981833441
7460470339844352254
9945354316570680900
18377735143728853940
15872286040534058104
8507347369923543553
308747406436979607
18218197459972943141
177225056570719...

result:

ok 120000 lines

Test #15:

score: 0
Time Limit Exceeded

input:

120000 120000
5867632904403891095 7734115005912544376 13901514937742085110 17141033381317351710 17988246451665298411 17540600012243810041 15190868862458266725 15454563456231559301 8740470882859091311 17882360629171437337 12095218884748126199 17817518168343018595 7112391591378197276 65538601378146110...

output:

13420129597746517056
18337642296188485079
4380067206814687930
5525410739012465159
1979575298167242590
10855827974711636510
9874020479466690595
17720000790164899934
16724816545805525287
15594891417371432986
14655386085151616523
11435247647009150363
7635358811289719984
14609735500036643833
13339404812...

result:


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
Memory Limit Exceeded

Test #34:

score: 19
Accepted
time: 1824ms
memory: 235940kb

input:

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

output:

5519324519442957729
13462861144392030499
8898301730697138469
4148979398311169421
15090246851327208067
8628707816538336707
13867297907038176506
10296428352441857103
15654304415409320656
7404566644919251615
9870876264015800597
6356224262148620783
249874952637342736
9023132497407647441
1416175985367538...

result:

ok 200000 lines

Test #35:

score: 19
Accepted
time: 504ms
memory: 103636kb

input:

200000 200000
11863650499568632095 6646669915142734572 4053375998669045948 14662364203830894482 7319511971537928619 4131498054494361157 7103043576860354080 2730587527012777841 8626012955062792888 7098277606750148625 12990209623538415680 100871355087529915 12267084290544303817 7008468684400973426 856...

output:

11796827338356402281
853302424664654652
564419876379363056
11173501553976506440
2595891684253320024
5590778053561483334
16613470245879883493
5698255402711241072
8125718572356459513
11371252602486101178
7605800585391618993
17939653588292877927
15906636118593883045
14840423959426947118
122409373008752...

result:

ok 200000 lines

Test #36:

score: 0
Memory Limit Exceeded

input:

200000 200000
7819034666999754227 1261434745131385029 4729051582132640442 9603230030483067185 9188148629137054368 4647584887897271213 14878584596185020248 5036501082632549589 13434605396022727436 10747373329537943689 2859138487129973374 17399126170759425906 5170686633178482234 1548518177806164267 68...

output:

13979872615900358449
13089809959604496472
1561607543806976909
704865086165131395
4750114500765789463
8620101000132483891
1061990993707638791
8219417779659049453
16783172337878697026
731502258552473136
15516870835064396649
13412140913912394465
5029558179142432386
9655856473800271062
14498308915283050...

result:


Subtask #7:

score: 0
Skipped

Dependency #1:

100%
Accepted

Dependency #2:

100%
Accepted

Dependency #3:

0%