QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#687655 | #9492. 树上简单求和 | xlwang | 19 | 4344ms | 130120kb | C++14 | 6.3kb | 2024-10-29 20:14:54 | 2024-10-29 20:14:54 |
Judging History
answer
#include<bits/stdc++.h>
#define ll long long
#define ull unsigned long long
#define fr(i,j,k) for(register int i=j;i<=k;++i)
#define rf(i,j,k) for(register int i=j;i>=k;--i)
#define foredge(i,j) for(register int i=head[j];i;i=e[i].nxt)
#define pb push_back
#define Times printf("Time:%.3lf\n",clock()/CLOCKS_PER_SEC)
#define pii pair<int,int>
#define mk make_pair
using namespace std;
inline int read(){
int x=0;
bool f=0;
char c=getchar();
while(!isdigit(c)) f|=(c=='-'),c=getchar();
while(isdigit(c)) x=(x<<3)+(x<<1)+(c^48),c=getchar();
return f?-x:x;
}
inline void write(int x){
if(x<0){putchar('-');x=-x;}
if(x>9)write(x/10);
putchar(x%10+'0');
}
inline void writeln(int x){write(x); puts("");}
inline void writepl(int x){write(x); putchar(' ');}
mt19937 rnd(chrono::steady_clock::now().time_since_epoch().count());
inline int randfind(int l,int r){return rnd()%(r-l+1)+l;}
//inline void init(){
// int t=read();
// while(t--) work();
//}
const int Maxn=2e5+10,B=500;
int n,q;
ull a[Maxn];
int toid[Maxn];
struct sgt{
ull s[Maxn<<2],tag[Maxn<<2];
inline int ls(int x){return x<<1;}
inline int rs(int x){return x<<1|1;}
inline void pushup(int x){s[x]=s[ls(x)]+s[rs(x)];}
inline void change(int x,int l,int r,ull k){s[x]+=(r-l+1)*k;tag[x]+=k;}
inline void pushdown(int x,int l,int r){
if(tag[x]){
int mid;mid=(l+r)>>1;
change(ls(x),l,mid,tag[x]);
change(rs(x),mid+1,r,tag[x]);
tag[x]=0;
}
}
inline void update(int ql,int qr,int l,int r,int x,ull k){
// cout<<ql<<' '<<qr<<' '<<l<<' '<<r<<' '<<x<<' '<<k<<endl;
if(ql<=l && r<=qr) {change(x,l,r,k);return;}
pushdown(x,l,r);int mid;mid=(l+r)>>1;
if(ql<=mid) update(ql,qr,l,mid,ls(x),k);
if(mid<qr) update(ql,qr,mid+1,r,rs(x),k);
pushup(x);
}
inline ull query(int ql,int qr,int l,int r,int x){
if(ql<=l && r<=qr) return s[x];
int mid;mid=(l+r)>>1;pushdown(x,l,r);ull ans=0;
if(ql<=mid) ans+=query(ql,qr,l,mid,ls(x));
if(mid<qr) ans+=query(ql,qr,mid+1,r,rs(x));
return ans;
}
}SGT;
struct BLOCK{
ull S[Maxn],s[Maxn];
int id[Maxn];
int L[Maxn],R[Maxn];
inline void into(){
fr(i,1,n){
id[i]=(i-1)/B+1;
if(!L[id[i]]) L[id[i]]=i;
R[id[i]]=i;
}
// fr(i,1,n){
// s[i]=a[toid[i]];
// if(i!=L[id[i]]) s[i]+=s[i-1];
// }
// fr(i,1,n) if(id[i]!=id[n]) S[id[i]+1]+=a[toid[i]];
// fr(i,1,id[n]) S[i]+=S[i-1];
}
inline void update(int x,ull k){
// cout<<x<<' '<<k<<endl;
fr(i,x,R[id[x]]) s[i]+=k;
fr(i,id[x]+1,id[n]) S[i]+=k;
}
inline ull getans(int x){return s[x]+S[id[x]];}
inline ull query(int l,int r){if(l>r) return 0;return getans(r)-getans(l-1);}
}BL;
struct Tree{
vector<int> vc[Maxn];
int f[20][Maxn<<1],dep[Maxn];
int dfn[Maxn],idx;
int num,id[Maxn];
int lg[Maxn<<1];
int father[Maxn];
int siz[Maxn];
inline void dfs(int x,int fa){
dep[x]=dep[fa]+1;f[0][++idx]=x;dfn[x]=idx;father[x]=fa;id[x]=++num;siz[x]=1;
for(auto y:vc[x]) if(y!=fa){
dfs(y,x);
f[0][++idx]=x;siz[x]+=siz[y];
}
}
inline int getmin(int x,int y){if(dep[x]<dep[y]) return x;return y;}
inline int getlca(int x,int y){
x=dfn[x],y=dfn[y];if(x>y) swap(x,y);
int ln=lg[y-x+1];
// cout<<ln<<' '<<x<<' '<<y<<' '<<f[ln][x]<<' '<<f[ln][y-(1<<ln)+1]<<endl;
return getmin(f[ln][x],f[ln][y-(1<<ln)+1]);
}
inline void Read(){
fr(i,1,n-1){
int x,y;
cin>>x>>y;
vc[x].pb(y);vc[y].pb(x);
}
dfs(1,0);
lg[0]=-1;fr(i,1,idx) lg[i]=lg[i/2]+1;
fr(j,1,lg[idx]) fr(i,1,idx){
if(i+(1<<j)-1>idx) break;
f[j][i]=getmin(f[j-1][i],f[j-1][i+(1<<(j-1))]);
// cout<<j<<' '<<i<<' '<<f[j][i]<<endl;
}
}
}T1,T2;
int vis[Maxn];
vector<int> ID;
struct node{int x,y;ull z;}que[Maxn];
ull ans[Maxn];
inline void init(){
cin>>n>>q;
fr(i,1,n) cin>>a[i];
T1.Read();T2.Read();
// fr(i,1,n) vis[i]=1;
fr(i,1,n) SGT.update(T2.id[i],T2.id[i]+T2.siz[i]-1,1,n,1,a[i]);
// fr(i,1,n) ID.pb(i);
// shuffle(ID.begin(),ID.end(),rnd());
// fr(i,0,min(n-1,B)) vis[ID[i]]=1;vis[1]=1;
fr(i,1,q) cin>>que[i].x>>que[i].y>>que[i].z;
BL.into();
}
vector<pii> vc[Maxn];
inline void update(int x,int y,ull z){
int llca=T1.getlca(x,y);
// cout<<llca<<endl;
BL.update(T1.id[x],z);BL.update(T1.id[y],z);
BL.update(T1.id[llca],-2*z);
// cout<<"**"<<endl;
SGT.update(T2.id[llca],T2.id[llca]+T2.siz[llca]-1,1,n,1,z);
a[llca]+=z;
}
inline ull query(int x){return BL.query(T1.id[x],T1.id[x]+T1.siz[x]-1);}
inline ull getval(int x,int y){
int llca=T2.getlca(x,y);
ull ans=0;
ans=SGT.query(T2.id[x],T2.id[x],1,n,1);
ans+=SGT.query(T2.id[y],T2.id[y],1,n,1);
ans-=2*SGT.query(T2.id[llca],T2.id[llca],1,n,1);
ans+=a[llca];
return ans;
}
inline void query(int x,int y,int id){
// cout<<"query:"<<x<<' '<<y<<' '<<id<<endl;
ans[id]=getval(x,y);
// cout<<"ans:"<<ans[id]<<endl;
while(1){
// cout<<x<<' '<<y<<endl;
if(vis[x] && vis[y]) break;
if(x==y){
ans[id]+=query(x),x=T2.father[x];
return;
}
if(vis[y]) ans[id]+=query(x),x=T2.father[x];
else if(vis[x]) ans[id]+=query(y),y=T2.father[y];
else {
if(T2.dep[x]<T2.dep[y]) swap(x,y);
// cout<<x<<endl;
ans[id]+=query(x),x=T2.father[x];
// cout<<"ans:"<<ans[id]<<endl;
}
}
int llca=T2.getlca(x,y);
// vc[x].pb(mk(id,1));vc[y].pb(mk(id,1));vc[llca].pb(mk(id,-2));ans[id]+=query(llca);
}
inline void work(){
// cout<<"dfn:\n";
// fr(i,1,n) cout<<i<<' '<<T1.id[i]<<endl;
fr(i,1,q) update(que[i].x,que[i].y,que[i].z),query(que[i].x,que[i].y,i);
fr(i,1,q) cout<<ans[i]<<'\n';
}
signed main(){
// freopen("input.in","r",stdin);
// freopen("output.out","w",stdout);
ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
init();work();
// printf("\nTIME:%.3lf",(double)clock()/CLOCKS_PER_SEC);
return 0;
}
详细
Subtask #1:
score: 5
Accepted
Test #1:
score: 5
Accepted
time: 3ms
memory: 27520kb
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: 13ms
memory: 19092kb
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: 12ms
memory: 28316kb
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: 0ms
memory: 30008kb
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: 12ms
memory: 20624kb
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: 24588kb
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: 360ms
memory: 53216kb
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: 4344ms
memory: 103644kb
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: 3189ms
memory: 106976kb
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: 0
Time Limit Exceeded
input:
70000 70000 2918414982140182939 1004760492603077644 7526656799259998488 6665485253854847449 7752199419154649757 12763267823077347079 11745132191692540338 6726116817426709990 15550876907005962464 9760509858122842638 684733892856965421 10077915441058780247 8380400329996723109 16920573433866702239 3069...
output:
result:
Subtask #3:
score: 0
Skipped
Dependency #2:
0%
Subtask #4:
score: 14
Accepted
Test #21:
score: 14
Accepted
time: 1509ms
memory: 125796kb
input:
200000 200000 622783158027686223 2242697872372232537 8481648430436878777 10092474834140799044 15403999682625301609 12614289513474949582 9180944589267018841 7823784919308285798 8257785171198951273 5134508521895120821 8041682272181381093 3835432206618893170 2653803171409877650 5589823419153460372 1007...
output:
9042998055336671259 11611293489264521142 5835924579879681322 9187084356907537870 17810346410706951073 565636160725988981 837626748701483168 16059573289829807099 7246210357888652619 7725251776483176497 17088098442183693937 9042305714006927228 10907378739216215456 6526772063609981609 51578202456469609...
result:
ok 200000 lines
Test #22:
score: 14
Accepted
time: 2055ms
memory: 126268kb
input:
200000 200000 13175752638648662841 17926176333479943540 18069418271192836667 7662981418770774166 17432280672869071045 9361466030141569604 17336291298429915451 758279154724011577 10229986883918215412 16695796270233481895 1104033984864960726 9768530369533627193 7121962912997584423 8574667967472399164 ...
output:
761007177180158471 99932139211644879 9085452500188024811 10579196290428182519 9823187704909577710 18023302821814112676 12490017484705421441 12628966555486388857 14265121989865566834 6520346880672680237 13101459183526308131 999924043939340162 18263995506773932901 5204528109864295202 12531805215875429...
result:
ok 200000 lines
Test #23:
score: 14
Accepted
time: 1600ms
memory: 130088kb
input:
200000 200000 7686280868723494190 956703982700755675 9999621735507690021 16173863373498393354 13710049849600478540 17103229081434028663 17565545023679367555 2828484246894512005 1583487132574088302 6282276626784421099 11842426946394217784 3255349046251970557 9837219010639574935 8803965402777990679 10...
output:
9027980728293426417 390552393210324231 11163738403290403569 7251051512011369232 11710945043516484177 8385783841330898676 10540689232459717148 13494924758898800208 10783463309429788767 15497109458285729613 3973164643641949159 16591368938886703497 17545967451093599325 7502098747509618204 7748818626114...
result:
ok 200000 lines
Test #24:
score: 14
Accepted
time: 1617ms
memory: 127328kb
input:
200000 200000 3398335727711776744 2517912491303558304 9944108242783740552 11465445588414101188 8918103911029611319 6248803476150904656 13839544089125989886 11613304797643373734 2743278001758631252 5880657146483100262 17520221750013284250 3574310479117269847 17332054826892442501 4186477155186295241 7...
output:
11201243883635739649 5642768912062346910 14237324928813743475 17949858083662777758 7007085524141292752 16431646654432642924 9544485471385114348 17223214017002242047 6358064993672703329 7126356965173878837 10226578739676773239 17581948280120185856 7547085902091221485 2256786006467014785 1348515941789...
result:
ok 200000 lines
Test #25:
score: 14
Accepted
time: 1662ms
memory: 130120kb
input:
200000 200000 16389428600328688123 13285293781493429938 16272776262151288852 2788638121841944928 840590085080737028 472104557233550161 2950757076856426026 884621482021485766 4656207248358869553 4325985129321868698 15439653714414044259 8869605634233383357 2875651646205284961 18315661660942366682 3209...
output:
11670281421082997569 13170194106693978241 4379481616026191349 1374955090149450188 16981223657037354332 15581757479756062245 964815911596550839 14653197660590615612 1244503873454847903 12992317503104122180 8922002840354854569 9883361056075385805 4661992164326801469 5993972796274466263 476463508437351...
result:
ok 200000 lines
Test #26:
score: 14
Accepted
time: 1704ms
memory: 123420kb
input:
200000 200000 8926134977558578929 14277420964906340273 14017501029945049702 16291239250458096854 5699993893720160591 1404316482439341580 6509187990544574711 3321495986616857673 9576515208059172862 16437943937474607467 3444518963957979419 17039197068804123693 9035882298315219046 10231648064038856650 ...
output:
11394865482866208122 4540012560447567167 14181315197904653108 1138165850159307501 4403319165822720694 3554076031362588972 4848001086504005989 17788785233422248859 5278865852900446472 9052657317349052491 7439239802335183804 280124506773607363 4951887064424754895 4442074242463250219 112728736814611771...
result:
ok 200000 lines
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%