QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#733671 | #9492. 树上简单求和 | WrongAnswer_90 | 30 | 3217ms | 108996kb | C++23 | 5.3kb | 2024-11-10 20:25:23 | 2024-11-10 20:25:23 |
Judging History
answer
#include<bits/extc++.h>
#include<bits/stdc++.h>
//#pragma GCC optimize(2)
//This code contains niche emotional elements. It is recommended to watch it after the age of 18.
#ifdef __unix__
#define gc getchar_unlocked
#define pc putchar_unlocked
#else
#define gc _getchar_nolock
#define pc _putchar_nolock
#endif
#define debug(x) fprintf(stderr,"Line %d,%s=%lld\n",__LINE__,#x,(ll)(x))
#define debugline fprintf(stderr,"--------------------------\n")
#define all(x) x.begin(),x.end()
#define pii pair<int,int>
#define Inf (int)INFINITY
#define ull unsigned ll
#define inf 0x3f3f3f3f
#define pb push_back
#define ll long long
#define endl '\n'
#define y second
#define x first
using namespace std;
const int N=2e5+10,B=450,M=N;
//char in[1<<20],*P1=in,*P2=in;
//#define getchar() (P1==P2&&(P2=(P1=in)+fread(in,1,1<<20,stdin),P1==P2)?EOF:*P1++)
void read(){};
template<class T1,class...T2>
inline void read(T1&ret,T2&...rest){
ret=0;char c=gc();bool f=0;
for(;!isdigit(c);c=gc())f=c=='-';
for(;isdigit(c);c=gc())ret=ret*10+c-'0';
f&&(ret=-ret),read(rest...);
}
#define cin(...) read(__VA_ARGS__)
template<class T>
inline void print(T x){
static short s[35],t=0;
bool f=x<0;if(f)x=-x;
do s[t++]=x%10,x/=10;while(x);
for(f&&pc('-');t;)pc(s[--t]+'0');
}
template<int N>struct LCA{
basic_string<int>*nxt;
int k,dn,d[N],dfn[N],mn[__lg(N)+1][N];
int get(int u,int v){return dfn[u]<dfn[v]?u:v;}
void pre(int u,int fa=0){
d[u]=d[mn[0][dfn[u]=++dn]=fa]+1;
for(int v:nxt[u])if(v^fa)pre(v,u);
}
inline void init(int n,int rt,basic_string<int>*t){
nxt=t,pre(rt);
for(int j=1;j<=__lg(n);j++)
for(int i=1;i+(1<<j)-1<=n;i++)
mn[j][i]=get(mn[j-1][i],mn[j-1][i+(1<<(j-1))]);
}
inline int operator()(int u,int v){
if(u==v)return u;
if((u=dfn[u])>(v=dfn[v]))swap(u,v);
return k=__lg(v-u++),get(mn[k][u],mn[k][v-(1<<k)+1]);
}
inline int len(int u,int v){return d[u]+d[v]-2*d[(*this)(u,v)];}
};
basic_string<int>t0[N],t1[N];
int n,m,mt,p0[N],p1[N];
ull a[N],vl[N],res[N];
LCA<N>lca,lca1;
struct O{bool tg;int x,y;ull w;}o[N*5];
namespace FK{
int dn,*dfn=lca.dfn,dfr[N],ix[N],R[M];
ull s0[N],s1[M];
void dfs0(int u,int fa=0){
dfn[u]=++dn;
for(int v:t0[u])
if(v^fa)dfs0(v,u);
dfr[u]=dn,p0[u]=fa;
}
inline void init(){
dfs0(1);
for(int i=1;i<=n;i++)R[ix[i]=(i+B-1)/B]=i;
}
inline ull query(int u){
return s1[ix[dfr[u]]]+s0[dfr[u]]-s1[ix[dfn[u]-1]]-s0[dfn[u]-1];
}
inline void add(int u,ull v){
for(int i=u=dfn[u];i<=R[ix[u]];i++)s0[i]+=v;
for(int i=ix[u]+1;i<=ix[n];i++)s1[i]+=v;
}
inline void add(int u,int v,ll w){
int l=lca(u,v);
if(l!=1)add(p0[l],-w);
add(u,w),add(v,w),add(l,-w);
}
}
namespace TFK{
const int B=200;
int sz[N];
bool tg[N];
set<pii>st;
basic_string<int>sp;
void dfs(int u,int fa=0){
p1[u]=fa,sz[u]=1;
for(int v:t1[u])
if(v^fa)vl[v]+=vl[u],dfs(v,u),sz[u]+=sz[v];
}
void era(int u,int d=0){
if(d>B)return;
st.erase({sz[u],u});
for(int v:t1[u])
if(v^p1[u])era(v,d+1);
}
inline void init(){
dfs(1),sp+=1;
for(int u=2;u<=n;u++)st.insert({sz[u],u});
basic_string<int>&s=sp;
for(int u;!st.empty()&&st.rbegin()->x>B;)
era(u=st.rbegin()->y),s+=u;
auto cmp=[&](const int&x,const int&y)
->bool{return lca1.dfn[x]<lca1.dfn[y];};
sort(all(s),cmp);
for(int i=1,l=(int)(s.size());i<l;i++)s.pb(lca1(s[i-1],s[i]));
sort(all(s),cmp),s.erase(unique(all(s)),s.end());
sort(all(s)),s.erase(unique(all(s)),s.end());
for(int u:s)tg[u]=1;
}
}
namespace CT{
basic_string<int>df;
int fl,ct[N];
void dfs(int u){
df+=u;
for(int v:t0[u])
if(v^p0[u])dfs(v);
}
inline void init(int u){
fill(ct+1,ct+n+1,0);
for(int x=u;x;x=p1[x])ct[x]++;
if(!fl)dfs(1),fl=1;
for(int u:df)
for(int v:t0[u])
if(v^p0[u])ct[v]+=ct[u];
}
inline int count(int u,int v){
int l=lca(u,v),r=ct[u]+ct[v]-ct[l];
if(l!=1)r-=ct[p0[l]];
return r;
}
}
inline void Mmax(int &x,int y){x=max(x,y);}
int main(){
// freopen("1.in","r",stdin);
// freopen("1.ans","w",stdout);
cin(n,m);
for(int i=1;i<=n;i++)
cin(a[i]),vl[i]=a[i];
for(int i=1,u,v;i<n;i++)
cin(u,v),t0[u]+=v,t0[v]+=u;
for(int i=1,u,v;i<n;i++)
cin(u,v),t1[u]+=v,t1[v]+=u;
lca.init(n,1,t0),
lca1.init(n,1,t1),
FK::init(),TFK::init();
cerr<<TFK::sp.size()<<endl;
int mx=0;
for(int i=1,u,v,l,x;i<=m;i++){
ull w;cin(u,v,w),FK::add(u,v,w),l=lca1(u,v);
res[i]=vl[u]+vl[v]-vl[l];
o[++mt]={0,u,v,w};
int rl=0;
for(x=u;!TFK::tg[x];x=p1[x],++rl)
res[i]+=FK::query(x);
Mmax(mx,rl),rl=0;
o[++mt]={1,i,x,1};
for(x=v;!TFK::tg[x];x=p1[x],++rl)
res[i]+=FK::query(x);
Mmax(mx,rl),rl=0;
o[++mt]={1,i,x,1};
for(x=l;!TFK::tg[x];x=p1[x],++rl)
res[i]-=FK::query(x);
Mmax(mx,rl),rl=0;
o[++mt]={1,i,x,0};
if(l>1){
for(x=p1[l];!TFK::tg[x];x=p1[x],++rl)
res[i]-=FK::query(x);
Mmax(mx,rl),rl=0;
res[i]-=vl[p1[l]],o[++mt]={1,i,x,0};
}
}
if(n>=200000)
return print(mx),0;
// double ST=clock();
for(int u:TFK::sp){
CT::init(u);
ull sum=0;
for(int i=1;i<=mt;i++){
if(o[i].tg){
if(o[i].y==u)
res[o[i].x]+=(o[i].w?1:-1)*sum;
}
else
sum+=o[i].w*CT::count(o[i].x,o[i].y);
}
}
// cerr<<clock()-ST<<endl;
for(int i=1;i<=m;i++)
print(res[i]),pc(endl);
return 0;
}
详细
Subtask #1:
score: 5
Accepted
Test #1:
score: 5
Accepted
time: 7ms
memory: 49936kb
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: 3ms
memory: 48228kb
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: 14ms
memory: 50192kb
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: 4ms
memory: 51544kb
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: 48928kb
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: 37084kb
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: 328ms
memory: 71688kb
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: 839ms
memory: 75384kb
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: 661ms
memory: 74800kb
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: 565ms
memory: 72800kb
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: 995ms
memory: 71972kb
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: 411ms
memory: 72396kb
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: 400ms
memory: 72612kb
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: 13
Accepted
Dependency #2:
100%
Accepted
Test #14:
score: 13
Accepted
time: 2732ms
memory: 86888kb
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: 13
Accepted
time: 3217ms
memory: 84940kb
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:
ok 120000 lines
Test #16:
score: 13
Accepted
time: 656ms
memory: 82252kb
input:
120000 120000 12590795589408290093 11275909154009220197 10996087245163181988 287253987689120747 522028471439816976 5251098397309018220 9466903789635983479 12562504698210775380 14359608006525868271 12860739587215060171 6110493152041264342 8941951597283806533 7247902667356706183 13908919237133011991 1...
output:
12568574963997891813 9761481874717599084 9748193662109666281 17759229044598540251 9232146611495947992 16451851566322082145 10163587939420717881 11939061140179225100 13555756814075457024 15331444302135124578 12404089960631589833 10914316315261843243 3330610810849167077 4308673861372384247 22214038984...
result:
ok 120000 lines
Test #17:
score: 13
Accepted
time: 2148ms
memory: 82576kb
input:
120000 120000 16014732670888203993 17003360764706956461 4309217563402934767 12155136955819956058 9472943536135969437 8619903067520668384 8532935983471178458 840596314384730622 8808409849265669508 16268252642839060126 8703458878963184072 14133056030074803107 16125379684339605436 13795196609771489642 ...
output:
14157083647705042191 9626539498588941455 12672590587667741319 7637586779842324001 17945793757619266709 6486513804042528016 17128439522931715545 14506773931432125656 10149598925465005229 8493513931817247270 9843224754697262936 17563138715401598781 5251652421620155711 16347886781065131205 141775307219...
result:
ok 120000 lines
Test #18:
score: 13
Accepted
time: 1542ms
memory: 81068kb
input:
120000 120000 3799726989717681598 2540658873004706851 6999377479951345208 4574847538261052797 1388764904624490347 797597442974571877 12049698136986117426 1334073669070317189 11812534794423707083 12910806883079537898 8919907966949030320 3609251437048437267 6955259073591432492 1913761510032208708 1536...
output:
16410218753178084291 3452125570141203996 12637348475870430626 939828522750114530 6181904502892064992 7530673907543931274 11125835388154440123 1132493328298594689 6540415366631312545 10171022468601061874 10785984448364782773 15132462728937914062 13894055239781007518 14553135357399426810 1058534365058...
result:
ok 120000 lines
Test #19:
score: 13
Accepted
time: 1269ms
memory: 82156kb
input:
120000 120000 523123731336918243 4097858622836674571 1961367823237386001 2725250184116707092 15256083167104925470 8826573328357841773 14551942275839683916 4676612813711946771 16866097649532049020 16028370234860808559 16550040875881081238 13114506992715113111 18223636920754369434 4711187061384408363 ...
output:
13171866584326858355 7378923635054867195 4676544939953182559 8211313138413629930 3762100382261198154 17376905504590956432 11575432112253285846 6651270238220236939 7296214239679292747 3921715204635211841 13514446212830941534 14681462832343839777 9394758474079991863 2039117273115115478 162036285844685...
result:
ok 120000 lines
Test #20:
score: 13
Accepted
time: 1165ms
memory: 81908kb
input:
120000 120000 16563527794956746492 3507607345845239151 6704373347302736722 4366660651585002614 17477711326324414919 14064904073416411127 1886265819461820151 1530962154006256458 3386801749967932412 10176358119958383434 14136333004424650089 18047845530334381646 14290290829566494165 1979382774616512004...
output:
2687063429907612648 7246290134975359576 451523845887078136 6787443537167049842 11742387638116508426 9452080235999660003 1640133015384348445 9479626460563956802 8536081354965239985 14886879078945984440 5977814891978138604 16811290642393481835 2319291536056800936 4327858711873466013 508236457787094527...
result:
ok 120000 lines
Subtask #4:
score: 0
Wrong Answer
Test #21:
score: 0
Wrong Answer
time: 814ms
memory: 92888kb
input:
200000 200000 622783158027686223 2242697872372232537 8481648430436878777 10092474834140799044 15403999682625301609 12614289513474949582 9180944589267018841 7823784919308285798 8257785171198951273 5134508521895120821 8041682272181381093 3835432206618893170 2653803171409877650 5589823419153460372 1007...
output:
202
result:
wrong answer 1st lines differ - expected: '9042998055336671259', found: '202'
Subtask #5:
score: 0
Wrong Answer
Test #27:
score: 0
Wrong Answer
time: 1276ms
memory: 103328kb
input:
200000 200000 1958469220619413759 14991498002015735322 6054491201406941902 18206143187746582567 15082377615826460430 2936248617457291604 10073577150351675920 16534472678586906457 2207599132486246393 10301540360769075442 1492580560381080472 551692353431379140 13238280352539145808 8462626987240986565 ...
output:
376
result:
wrong answer 1st lines differ - expected: '11479812171669345085', found: '376'
Subtask #6:
score: 0
Wrong Answer
Test #34:
score: 0
Wrong Answer
time: 453ms
memory: 108996kb
input:
200000 200000 6794776813641982926 1561596256197101737 10910039723053043515 7892247858295192798 12233819960547881004 17695389034783066733 9173201689566865598 17626618141377486739 7358781671024283919 6787559733384974662 3884392438269280436 14872846228351316833 9037842441501571648 14299818404271084016 ...
output:
204
result:
wrong answer 1st lines differ - expected: '5519324519442957729', found: '204'
Subtask #7:
score: 0
Skipped
Dependency #1:
100%
Accepted
Dependency #2:
100%
Accepted
Dependency #3:
100%
Accepted
Dependency #4:
0%