【ZJOI2015】幻想乡战略游戏【点分树】【带权重心】
題意:nnn個點帶邊權的樹,動態修改點權viv_ivi?,最小化 欽定一個點xxx 后 ∑idist(x,i)?vi\sum\limits_{i} dist(x,i)*v_ii∑?dist(x,i)?vi?的值。
n,q≤105n,q \leq10^5n,q≤105,度數不超過202020
限制度數的樹上的一些詭異的操作,時限很長,多半是點分樹。
也叫動態點分治,但實際上并不是動態的點分治并且有一定誤導性,所以以后都叫點分樹。
點分樹是對一個樹點分治后的結構建出的樹,即在點分治時將下一層的重心的父親設為當前的分治中心。
它具有以下常用的性質:
本質上是對樹建出的線段樹。
在本題中詳細講解。
首先本題實際上求的是帶權重心
有個結論:
設當前點是uuu,如果vvv比uuu更優
那么有
len(u,v)?(n?sumv?sumv)<0len(u,v)*(n-sum_v-sum_v)<0len(u,v)?(n?sumv??sumv?)<0
其中sumsumsum表示子樹點權和
即
2sumv>n2sum_v>n2sumv?>n
然后繼續往下走
不難看出對于一個uuu,這樣的vvv最多只有111個,所以答案一定在滿足條件的vvv的子樹內。如果沒有這樣的vvv說明uuu是帶權重心。
這樣是O(n)O(n)O(n)的,考慮搬到點分樹上
從點分樹的根開始往下走
設當前在uuu,我們找到點分樹上uuu的一個兒子vvv
注意之前的結論只能往原樹上相鄰的點走,所以你不能直接用這個結論判斷vvv
但是如果我們設uuu往 vvv在點分樹上的子樹 的這個方向走一步 到達的點是www
即:
(紅色為點分樹)
因為www在原樹上的子樹 等于 vvv在點分樹上的子樹
我們想判斷答案是否在 vvv在點分樹上的子樹內,可以轉換為是否在 www在原樹上的子樹內
然而如果你判2?sumv>sumrt2*sum_v>sum_{rt}2?sumv?>sumrt?,會發現你還是WA了
原因是你欽定uuu為根之后,這棵樹的形態已經確定了
你在點分樹上一直往下走,實際上原樹上仍然在亂跳
人話:只有第一層的www(原樹)和vvv(點分樹)的子樹一樣,后面的點分樹上的子樹在原樹上甚至可能不是子樹。
但是上面已經證明過最多只有一個vvv
我們可以直接算出uuu在原樹上的每個兒子的答案和根結點比較,如果有一個www比根結點優,因為只有一個,說明答案在 www在原樹上的子樹(或vvv在點分樹上的子樹)內。
然后想象把這條邊斷開,化歸到從vvv開始的子問題。
也就是說uuu和vvv并沒有實質關聯,只是從重心開始方便處理而已。
現在考慮如何計算一個點的答案
維護ansuans_uansu?表示以faufa_ufau?為根時,uuu在點分樹上的子樹中的點到faufa_ufau?的帯權距離(距離*點權)之和,fafafa為在點分樹上的父結點。
詢問點xxx的答案時,先加入點分樹上子結點的所有ansansans,然后在點分樹上往上跳,把兄弟結點的子樹中的所有點權挪到父親上,再一起挪到xxx。因為樹高O(log?n)O(\log n)O(logn),可以保證復雜度。詳見代碼。
修改的時候暴力跳父親修改sumsumsum和ansansans就可以了。
#include <iostream> #include <cstdio> #include <cstring> #include <cctype> #include <vector> #include <cassert> #define MAXN 100005 #define MAXM 200005 using namespace std; typedef long long ll; inline int read() {int ans=0,f=1;char c=getchar();while (!isdigit(c)) (c=='-')&&(f=-1),c=getchar();while (isdigit(c)) ans=(ans<<3)+(ans<<1)+(c^48),c=getchar();return f*ans; } struct edge{int u,v,w;}e[MAXM]; int head[MAXN],nxt[MAXM],cnt; void addnode(int u,int v,int w) {e[++cnt]=(edge){u,v,w};nxt[cnt]=head[u];head[u]=cnt; } int dis[MAXN],pos[MAXN],dfn[MAXM],up[MAXN],tim; void dfs(int u) {dfn[pos[u]=++tim]=u;for (int i=head[u];i;i=nxt[i])if (!pos[e[i].v]){dis[e[i].v]=dis[u]+e[i].w;up[e[i].v]=u;dfs(e[i].v);dfn[++tim]=u;} } int LOG[MAXM],st[20][MAXM]; inline int Min(const int& x,const int& y){return pos[x]<pos[y]? x:y;} inline void init() {LOG[0]=-1;for (int i=1;i<MAXM;i++) LOG[i]=LOG[i>>1]+1;for (int i=1;i<=tim;i++) st[0][i]=dfn[i];for (int i=1;i<20;i++)for (int j=1;j+(1<<(i-1))<=tim;j++)st[i][j]=Min(st[i-1][j],st[i-1][j+(1<<(i-1))]); } inline int lca(const int& x,const int& y) {int l=pos[x],r=pos[y];if (l>r) swap(l,r);int t=LOG[r-l+1];return Min(st[t][l],st[t][r-(1<<t)+1]); } inline int dist(const int& x,const int& y){return dis[x]+dis[y]-2*dis[lca(x,y)];} int rt; int siz[MAXN],maxp[MAXN]={0x7fffffff}; bool cut[MAXN]; void findrt(int u,int f,int sum) {siz[u]=1,maxp[u]=0;for (int i=head[u];i;i=nxt[i])if (e[i].v!=f&&!cut[e[i].v]){findrt(e[i].v,u,sum);siz[u]+=siz[e[i].v],maxp[u]=max(maxp[u],siz[e[i].v]);}if (sum-siz[u]>maxp[u]) maxp[u]=sum-siz[u];if (maxp[u]<maxp[rt]) rt=u; } int getsiz(int u,int f) {int ans=1;for (int i=head[u];i;i=nxt[i])if (e[i].v!=f&&!cut[e[i].v])ans+=getsiz(e[i].v,u);return ans; } int d[MAXN],sum[MAXN]; ll ans[MAXN]; int fa[MAXN]; vector<int> son[MAXN],top[MAXN]; void build() {int u=rt;cut[u]=true;for (int i=head[u];i;i=nxt[i])if (!cut[e[i].v]){rt=0;findrt(e[i].v,0,getsiz(e[i].v,0));son[u].push_back(rt),top[u].push_back(e[i].v),fa[rt]=u;build();} } bool vis[MAXN]; void DFS(int u) {assert(!vis[u]);vis[u]=true;for (int i=0;i<(int)son[u].size();i++)DFS(son[u][i]); } inline void modify(int x,int v) {int u=x;for (;fa[x];x=fa[x]) sum[x]+=v,ans[x]+=(ll)dist(fa[x],u)*v;sum[x]+=v; } inline ll calc(int x) {ll res=0;for (int i=0;i<(int)son[x].size();i++)res+=ans[son[x][i]];for (int u=fa[x],v=x;u;v=u,u=fa[u]){int tot=d[u];for (int i=0;i<(int)son[u].size();i++)if (son[u][i]!=v)res+=ans[son[u][i]],tot+=sum[son[u][i]];res+=(ll)tot*dist(u,x); }return res; } inline ll query(int x) {ll v=calc(x);for (int i=0;i<(int)son[x].size();i++)if (calc(top[x][i])<v)return query(son[x][i]);return v; } int main() {int n,q;n=read(),q=read();for (int i=1;i<n;i++){int u,v,w;u=read(),v=read(),w=read();addnode(u,v,w),addnode(v,u,w);}dfs(1);init();int Rt;findrt(1,0,n),Rt=rt,build();while (q--){int x,v;x=read(),v=read();d[x]+=v,modify(x,v);printf("%lld\n",query(Rt));}return 0; } 創作挑戰賽新人創作獎勵來咯,堅持創作打卡瓜分現金大獎總結
以上是生活随笔為你收集整理的【ZJOI2015】幻想乡战略游戏【点分树】【带权重心】的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: K-D Tree学习笔记
- 下一篇: 报告:到 2026 年中国将占据全球太阳