文章目錄

• 解析
• • update:
• 代碼

所謂樹上莫隊，就是在樹上的莫隊

（逃）
傳送門

解析

似乎就是樹上的這道題
考慮如何轉化為序列問題呢?
考慮dfs序
但是又一個問題。。。
似乎這條鏈的dfs序不連續啊
樹剖一下就好啦
考慮更陽間的方法
求出這棵樹的歐拉序，在這個歐拉序上詢問
那么我們發現，這樣的話，其實會多算的部分就都會多算2遍
比如樣例：

以1為根，歐拉序為：
13442231
那么我們考慮（4-3）
對應的序列就是：

44223

不在路徑上的2恰好算了2次
所以我們可以利用異或的性質
還有一些特判的問題：

lca不在端點時，需要額外計算lca

左端點不時lca時，需要額外計算左端點
畫畫圖就很清楚了

update:

上面那個特判是洛谷野生題解的，有點過于陰間了…
ybt的實現優美的多（至少易于記憶）
設一個點進入和離開dfs的時間分別為 $in(x),out(x)$。
令 $in(x)<in(y)$，分情況討論：

$x$ 是 $y$ 的祖先，那么答案就是 $[in(x),in(y)]$。

$x$ 不是 $y$ 的祖先，那么在 $[out(x),in(y)]$ 的基礎上，還要加上 $lca(x,y)$。

代碼

#include<bits/stdc++.h> using namespace std; #define ll long long const int N=3e5+100; const int M=1050; const int mod=998244353; const double eps=1e-6; ll read(){ll x=0,f=1;char c=getchar();while(!isdigit(c)){if(c=='-') f=-1;c=getchar();}while(isdigit(c)){x=x*10+c-'0';c=getchar();}return x*f; } int n,m; struct node{int to,nxt; }p[N<<1]; int fi[N],cnt; void addline(int x,int y){p[++cnt]=(node){y,fi[x]};fi[x]=cnt; } int pl[N][22],dep[N],siz[N]; ll v[N],w[N]; int pos[N],dfn[N],ed[N],Tim; void dfs(int x,int f){pl[x][0]=f;pos[x]=++Tim;dfn[Tim]=x;siz[x]=1;for(int k=1;k<=18;k++) pl[x][k]=pl[pl[x][k-1]][k-1];for(int i=fi[x];~i;i=p[i].nxt){int to=p[i].to;if(to==f) continue;dep[to]=dep[x]+1;dfs(to,x);siz[x]+=siz[to];}dfn[++Tim]=x;ed[x]=Tim; } bool vis[N]; int a[N],tim,tot,from[N],to[N],id[N],bel[N]; int Lca(int x,int y){if(dep[x]<dep[y]) swap(x,y);for(int k=18;k>=0;k--){if(!pl[x][k]||dep[pl[x][k]]<dep[y]) continue;x=pl[x][k];} // printf("LCA:x=%d y=%d ",x,y);if(x==y){ // printf("ares=%d\n",x);return x;} // printf(" mid:x=%d y=%d ",x,y);for(int k=18;k>=0;k--){if(pl[x][k]==pl[y][k]) continue;x=pl[x][k];y=pl[y][k];} // printf("x=%d res=%d\n",x,pl[x][0]);return pl[x][0]; } struct query{int l,r,t,id,lca;bool operator < (const query o)const{if(bel[l]!=bel[o.l]) return bel[l]<bel[o.l];else if(bel[r]!=bel[o.r]) return bel[r]<bel[o.r];else return t<o.t;} }ask[N]; int l,r,t,bac[N]; ll now; ll ans[N]; inline void work(int x){x=dfn[x];(vis[x]^=1)?now+=v[a[x]]*w[++bac[a[x]]]:now-=v[a[x]]*w[bac[a[x]]--]; } int q; int ww; int main(){memset(fi,-1,sizeof(fi));cnt=-1;n=read();m=read();q=read();for(int i=1;i<=m;i++) v[i]=read();for(int i=1;i<=n;i++) w[i]=read();for(int i=1;i<n;i++){int x=read(),y=read();addline(x,y);addline(y,x);}for(int i=1;i<=n;i++) a[i]=read();dep[1]=1;dfs(1,0); // for(int i=1;i<=Tim;i++) printf("%d ",dfn[i]); // printf("\n");ww=floor(pow(Tim,2.0/3));for(int i=1;i<=Tim;i++){bel[i]=(i-1)/ww+1;}for(int i=1;i<=q;i++){int op=read(),x=read(),y=read();if(op==0){++tim;int o=x;from[tim]=a[o];id[tim]=o;a[o]=y;to[tim]=a[o];}else{if(pos[x]>pos[y]) swap(x,y);ask[++tot]=(query){pos[x],pos[y],tim,tot,pos[Lca(x,y)]};} // printf("i=%d op=%d x=%d y=%d\n",i,op,x,y);}sort(ask+1,ask+1+tot);l=1;t=tim;for(int i=1;i<=tot;i++){int nl=ask[i].l,nr=ask[i].r,nt=ask[i].t,lca=ask[i].lca;while(l<nl) work(l++);while(l>nl) work(--l);while(r<nr) work(++r);while(r>nr) work(r--);while(t<nt){t++;int o=id[t],f=0;if((pos[o]<l&&l<=ed[o]&&ed[o]<=r)) f=1;else if(l<=pos[o]&&pos[o]<=r&&r<ed[o]) f=2;if(f==1) work(ed[o]);else if(f==2) work(pos[o]);a[o]=to[t];if(f==1) work(ed[o]);else if(f==2) work(pos[o]);}while(t>nt){int o=id[t],f=0;if((pos[o]<l&&l<=ed[o]&&ed[o]<=r)) f=1;else if(l<=pos[o]&&pos[o]<=r&&r<ed[o]) f=2;if(f==1) work(ed[o]);else if(f==2) work(pos[o]);a[o]=from[t];if(f==1) work(ed[o]);else if(f==2) work(pos[o]);t--;}if(nl!=lca){work(nl);if(nr!=lca) work(lca);} // printf("id=%d (%d %d %d %d) res=%lld\n",ask[i].id,nl,nr,nt,lca,now);ans[ask[i].id]=now;if(nl!=lca){work(nl);if(nr!=lca) work(lca);}}for(int i=1;i<=tot;i++) printf("%lld\n",ans[i]);return 0; } /**/ 創作挑戰賽新人創作獎勵來咯，堅持創作打卡瓜分現金大獎

總結

以上是生活随笔為你收集整理的YBTOJ洛谷P4074：糖果公园（树上莫队）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。