洛谷P3345 [ZJOI2015]幻想乡战略游戏 [动态点分治]

传送门


调了两个小时,终于过了……

凭啥人家代码80行我180行啊!!!

谁叫你大括号换行

谁叫你写缺省源


思路

显然,补给点所在的位置就是这棵树的带权重心。

考虑size已知时如何找重心:一开始设答案在\(x\),若存在\(x\)的一个子节点\(v\),使\(size_v>sum-size_v\),即\(2size_v>sum\),就往\(v\)走,重复该过程,直到走不动为止。

考虑用点分树优化这一过程:由于点分树层数不超过\(\log n\),这题又保证\(deg\leq 20\),可以暴力往下跳。

然而由于在点分树上跳,与原树有一些区别,需要搞一些奇怪的东西。

假设点分树上\(x\)点有\(v\)点为儿子且满足上述要求,在原树上有\(x\rightarrow w\)这条边,且\(w\)在\(v\)的点分树的子树里,那么就要把除了\(v\)这个连通块以外的\(size\)加到\(w\)上。

即:\(size_w+=size_x-size_v\),其中\(size\)表示点分树上的子树的\(\sum d\)。(不明白为什么要叫作\(d\),叫\(val\)不好么?)

求出重心后还要把刚才的修改操作撤销。

得到重心之后就很好做了,沿点分树往上跳就好了。

需要维护以下几样东西:

点分树上有节点\(x\),它在点分树上的父亲是\(fa\),它在点分树上的子树为\(S\),则有:

\[\begin{align*}
&sumv_x=\sum_{u\in S} val_u\\
&sumV_x=\sum_{u\in S} val_u dis(u,x)\\
&sumFV_x=\sum_{u\in S} val_u dis(u,fa)
\end{align*}
\]

记录这几样东西就可以统计答案了。

听着还好,写的时候思路很容易乱,务必要理清。

不然您就会像我一样记录了一堆东西全都没用

假设\(q,n\)同阶,则最后复杂度\(O(20n\log n)\)?反正能过就行,而且还挺快。


代码

#include<bits/stdc++.h>
namespace my_std{
using namespace std;
#define pii pair<int,int>
#define fir first
#define sec second
#define MP make_pair
#define rep(i,x,y) for (int i=(x);i<=(y);i++)
#define drep(i,x,y) for (int i=(x);i>=(y);i--)
#define go(x) for (int i=head[x];i;i=edge[i].nxt)
#define sz 101010
typedef long long ll;
template<typename T>
inline void read(T& t)
{
t=0;char f=0,ch=getchar();
double d=0.1;
while(ch>'9'||ch<'0') f|=(ch=='-'),ch=getchar();
while(ch<='9'&&ch>='0') t=t*10+ch-48,ch=getchar();
if(ch=='.')
{
ch=getchar();
while(ch<='9'&&ch>='0') t+=d*(ch^48),d*=0.1,ch=getchar();
}
t=(f?-t:t);
}
template<typename T,typename... Args>
inline void read(T& t,Args&... args){read(t); read(args...);}
void file()
{
#ifndef ONLINE_JUDGE
freopen("a.txt","r",stdin);
#endif
}
inline bool chkmin(int &x,int y){return x>y?x=y,1:0;}
inline bool chkmax(int &x,int y){return x<y?x=y,1:0;}
// inline ll mul(ll a,ll b){ll d=(ll)(a*(double)b/mod+0.5);ll ret=a*b-d*mod;if (ret<0) ret+=mod;return ret;}
}
using namespace my_std; #define v edge[i].t
namespace T
{
struct hh{int t;ll w;int nxt;}edge[sz<<1];
int head[sz],ecnt;
void make_edge(int f,int t,ll w)
{
edge[++ecnt]=(hh){t,w,head[f]};
head[f]=ecnt;
edge[++ecnt]=(hh){f,w,head[t]};
head[t]=ecnt;
}
ll val[sz];
ll Dep[sz];
void dfs(int x,int fa){ go(x) if (v!=fa) { Dep[v]=Dep[x]+edge[i].w; dfs(v,x); } }
bool vis[sz];
int size[sz],sum,rt,rtS;
void calcS(int x,int fa)
{
size[x]=1;
go(x) if (v!=fa&&!vis[v]) { calcS(v,x); size[x]+=size[v]; }
}
void findrt(int x,int fa)
{
int S=sum-size[x];
go(x) if (v!=fa&&!vis[v]) { findrt(v,x); chkmax(S,size[v]); }
if (chkmin(rtS,S)) rt=x;
}
int findrt(int x){ calcS(x,0); sum=size[x]; rtS=1e9; findrt(x,0); return rt; }
}
#define goT(x) for (int i=T::head[x];i;i=T::edge[i].nxt)
#define Tv T::edge[i].t
namespace S
{
int root;
struct hh{int t,nxt;}edge[sz<<1];
int head[sz],ecnt;
void make_edge(int f,int t){edge[++ecnt]=(hh){t,head[f]};head[f]=ecnt;}
ll sumv[sz],sumV[sz],sumFV[sz];
int ff[sz];
struct hhh{int u,uu;ll dis;};
vector<hhh>in[sz];
void calcDis(int x,int fa,int u,int uu,ll dis)
{
in[x].push_back((hhh){u,uu,dis});
goT(x) if (Tv!=fa&&!T::vis[Tv]) calcDis(Tv,x,u,uu,dis+T::edge[i].w);
}
void build(int x)
{
T::vis[x]=1;
goT(x) if (!T::vis[Tv])
{
int rt=T::findrt(Tv);
make_edge(x,rt);
calcDis(Tv,0,x,rt,T::edge[i].w);
ff[rt]=Tv;
build(rt);
}
}
void build(){root=T::findrt(1);build(root);}
void change(int x,ll s) // val_x+=s
{
T::val[x]+=s;sumv[x]+=s;
drep(t,(int)in[x].size()-1,0)
{
int u=in[x][t].u,uu=in[x][t].uu;ll D=in[x][t].dis*s;
sumv[u]+=s;sumV[u]+=D;sumFV[uu]+=D;
}
}
ll getans(int x)
{
ll ret=sumV[x];
drep(t,(int)in[x].size()-1,0)
{
int u=in[x][t].u,uu=in[x][t].uu;ll dis=in[x][t].dis;
go(u) if (v!=uu) ret+=sumFV[v]+sumv[v]*dis;
ret+=T::val[u]*dis;
}
return ret;
}
int getroot(int x)
{
go(x) if (sumv[v]*2>sumv[x])
{
int delta=sumv[x]-sumv[v];
change(ff[v],delta);
int ret=getroot(v);
change(ff[v],-delta);
return ret;
}
return x;
}
ll getans(){return getans(getroot(root));}
} int n,Q; int main()
{
file();
read(n,Q);
int x,y,z;
rep(i,1,n-1) read(x,y,z),T::make_edge(x,y,z);
S::build();
while (Q--)
{
read(x,y);
S::change(x,y);
printf("%lld\n",S::getans());
}
}

(压了一下行,只剩\(151\)行了)

(反正我写这么丑也没人看QwQ)

上一篇:uva 12549


下一篇:POJ 2289 Jamie's Contact Groups / UVA 1345 Jamie's Contact Groups / ZOJ 2399 Jamie's Contact Groups / HDU 1699 Jamie's Contact Groups / SCU 1996 Jamie's Contact Groups (二分,二分图匹配)