QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#662677 | #9492. 树上简单求和 | ANIG | 5 | 42ms | 105612kb | C++14 | 3.5kb | 2024-10-21 09:05:16 | 2024-10-21 09:05:16 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
#define int unsigned long long
const int N=2e5+5,T=200,B=500;
struct msg{
int bh,x,y,c,z;
};
vector<msg>q1,q2,g[N];
int n,m,f[N],w[N],rs[N],laz[N],sm[N],st[N],ed[N],bh[N],lx[N],dfn[N],eds[N],dy1[N],dy2[N],idx1,idx2,mk[N],w1[N],w2[N],dep1[N],dep2[N],f1[N][20],f2[N][20];
vector<int>p1[N],p2[N];
void dfs1(int x){
for(int i=1;i<=18;i++)f1[x][i]=f1[f1[x][i-1]][i-1];
mk[x]=1;
dy1[++idx1]=x;
for(auto c:p1[x]){
if(mk[c])continue;
f1[c][0]=x;
dep1[c]=dep1[x]+1;
dfs1(c);
}
mk[x]=0;
}
void dfs2(int x){
for(int i=1;i<=18;i++)f2[x][i]=f2[f2[x][i-1]][i-1];
mk[x]=1;
dy2[++idx2]=x;
dfn[x]=idx2;
for(auto c:p2[x]){
if(mk[c])continue;
f2[c][0]=x;
dep2[c]=dep2[x]+1;
dfs2(c);
}
eds[x]=idx2;
mk[x]=0;
}
int up1(int x,int k){
while(k){
x=f1[x][__lg(k&-k)];
k^=k&-k;
}
return x;
}
int up2(int x,int k){
while(k){
x=f2[x][__lg(k&-k)];
k^=k&-k;
}
return x;
}
int lca1(int a,int b){
if(dep1[a]>dep1[b])swap(a,b);
b=up1(b,dep1[b]-dep1[a]);
if(a==b)return a;
for(signed i=18;i>=0;i--){
if(f1[a][i]!=f1[b][i])a=f1[a][i],b=f1[b][i];
}
return f1[a][0];
}
int lca2(int a,int b){
if(dep2[a]>dep2[b])swap(a,b);
b=up2(b,dep2[b]-dep2[a]);
if(a==b)return a;
for(signed i=18;i>=0;i--){
if(f2[a][i]!=f2[b][i])a=f2[a][i],b=f2[b][i];
}
return f2[a][0];
}
void add(int bh,int x,int a,int b,int c,int z){
g[x].push_back({bh,a,b,c,z});
}
void add(int bh,int x,int y,int c1,int a,int b,int c2,int z){
add(bh,x,a,b,c2,z);
add(bh,y,a,b,c2,z);
add(bh,c1,a,b,c2,-z);
if(c1!=1)add(bh,f1[c1][0],a,b,c2,-z);
}
void solve(){
for(auto c:q2){
for(auto d:q1){
if(d.bh>c.bh)break;
add(c.bh,d.x,d.y,d.c,c.x,c.y,c.c,d.z);
}
}
memset(w1,0,sizeof(w1));
for(auto c:q1){
w1[c.x]+=c.z;
w1[c.y]+=c.z;
w1[c.c]-=c.z;
if(c.c!=1)w1[f1[c.c][0]]-=c.z;
}
for(int i=n;i>=1;i--){
int x=dy1[i];
for(auto c:p1[x]){
if(c==f1[x][0])continue;
w1[x]+=w1[c];
}
}
for(int i=1;i<=n;i++)w[i]+=w1[i],w2[i]=w[i];
for(int i=1;i<=n;i++){
int x=dy2[i];
for(auto c:p2[x]){
if(c==f2[x][0])continue;
w2[c]+=w2[x];
}
}
q1.clear();
q2.clear();
}
void ADD(int l,int r,int y){
for(int i=l;i<=r;i++)sm[i]+=y;
}
void add(int x,int y){
int l=dfn[x],r=eds[x];
if(bh[l]==bh[r]){
ADD(l,r,y);
return;
}
int ll=bh[l]+1,rr=bh[r]-1;
for(int i=ll;i<=rr;i++)laz[i]+=y;
ADD(l,st[ll]-1,y);ADD(ed[rr]+1,r,y);
}
int gets(int x){
if(!x)return 0;
return sm[dfn[x]]+laz[bh[dfn[x]]];
}
int gets(int a,int b,int c){
return gets(a)+gets(b)-gets(c)-gets(f2[c][0]);
}
void dfs(int x){
add(x,1);
mk[x]=1;
for(auto c:g[x])rs[c.bh]+=gets(c.x,c.y,c.c)*c.z;
for(auto c:p1[x]){
if(mk[c])continue;
dfs(c);
}
mk[x]=0;
add(x,-1);
}
signed main(){
ios::sync_with_stdio(0);
cin.tie(0);cout.tie(0);
cin>>n>>m;
for(int i=1;i<=n;i++)cin>>w[i],bh[i]=(i-1)/B+1;
for(int i=1;i<=bh[n];i++)st[i]=ed[i-1]+1,ed[i]=min(i*B,n);
for(int i=1;i<n;i++){
int x,y;
cin>>x>>y;
p1[x].push_back(y);
p1[y].push_back(x);
}
for(int i=1;i<n;i++){
int x,y;
cin>>x>>y;
p2[x].push_back(y);
p2[y].push_back(x);
}
dfs1(1);
dfs2(1);
solve();
for(int i=1;i<=m;i++){
int x,y,z;
cin>>x>>y>>z;
q1.push_back({i,x,y,lca1(x,y),z});
int c=lca2(x,y);
q2.push_back({i,x,y,c});
lx[i]=1;
rs[i]=w2[x]+w2[y]-w2[c];
if(c!=1)rs[i]-=w2[f2[c][0]];
if(i%T==0)solve();
}
solve();
dfs(1);
for(int i=1;i<=m;i++)if(lx[i])cout<<rs[i]<<"\n";
}
詳細信息
Subtask #1:
score: 5
Accepted
Test #1:
score: 5
Accepted
time: 39ms
memory: 105612kb
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: 39ms
memory: 100048kb
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: 42ms
memory: 101588kb
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: 32ms
memory: 104560kb
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: 35ms
memory: 99960kb
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
Memory Limit Exceeded
Dependency #1:
100%
Accepted
Test #6:
score: 12
Accepted
time: 4ms
memory: 36736kb
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
Memory Limit Exceeded
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:
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%