正题
题目链接:https://www.luogu.com.cn/problem/P4338
题目大意
给出nnn个点的一棵树,和每个点进行accessaccessaccess的次数aia_iai,要求安排一个顺序使得虚实边转换最多。
mmm次修改一个点让aia_iai加上www后求答案
n,m∈[1,4∗105],ai,w∈[1,107]n,m\in [1,4*10^5],a_i,w\in[1,10^7]n,m∈[1,4∗105],ai,w∈[1,107]
解题思路
好像本来就很麻烦还带修改,那先不考虑修改
考虑统计每个节点下的边的虚实切换最大化,可以发现一个节点的子树中无论怎样安排accessaccessaccess顺序也不会影响外面的答案,因为对于外面的来说都相当于accessaccessaccess了这个节点。
所以这个满足子最优?(好像是这么叫的),那每一个节点的分块考虑就好了。现在对于这个节点下的边,如果两次accessaccessaccess的是在不同的儿子的子树中就会产生一点贡献。
转换一下现在的问题就是有若干种个数不同的颜色排成一排,要求相邻的异色最多。这个可以贪心解决,正常来说只要每次拿与上个不同的最多的来排就能到达sum−1sum-1sum−1的答案上线,但是需要特判一下如果最多的颜色个数mxmxmx有2×mx>sum2\times mx>sum2×mx>sum那么此时这样排到最后还有一种颜色剩下,答案就是2×(sum−mx)2\times (sum-mx)2×(sum−mx)。
所以一个节点的答案就是min{sum−1,2×(sum−mx)}min\{sum-1,2\times (sum-mx)\}min{sum−1,2×(sum−mx)}
但是带修改怎么搞,考虑到每次修改一定是加一个正权。
我们显然有一个式子
2×mx>sum⇒2×(mx+c)>sum+c2\times mx> sum\Rightarrow 2\times (mx+c)> sum+c2×mx>sum⇒2×(mx+c)>sum+c
这个式子表明如果一个节点选择了2×(sum−mx)2\times (sum-mx)2×(sum−mx)作为权值,那么它以后也都是这个权值。
sxs_xsx表示xxx子树中的权值和,对一个节点xxx定义r=max{sy(fay=x)}r=max\{s_y(fa_y=x)\}r=max{sy(fay=x)}和sum=∑fay=xsysum=\sum_{fa_y=x}s_ysum=∑fay=xsy
如果2×mx>sum2\times mx>sum2×mx>sum那么向(x,y)(x,y)(x,y)连一条实边,其他儿子连虚边。否则全连虚边。
那么每次修改的过程中我们只需要遍历到根节点的虚边看是否需要切换即可,这个可以用LCTLCTLCT来维护。
而且因为每条虚边代表着2×sy≤sx2\times s_y\leq s_x2×sy≤sx,所以路径上虚边的个数不会超过log∑ai\log \sum a_ilog∑ai级别。
时间复杂度O(nlog∑ai)O(n\log \sum a_i)O(nlog∑ai)(SplaySplaySplay那个logn\log nlogn因为小于log∑ai\log \sum a_ilog∑ai就舍去)
code
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const ll N=4e5+10;
struct node{ll to,next;
}a[N<<1];
ll n,m,s[N],ls[N],ans,tot;
struct LCT{ll fa[N],s[N],w[N],v[N],t[N][2];bool Nroot(ll x){return fa[x]&&(t[fa[x]][0]==x||t[fa[x]][1]==x);}bool Direct(ll x){return t[fa[x]][1]==x;}void PushUp(ll x){s[x]=s[t[x][0]]+s[t[x][1]]+w[x]+v[x];return;}void Rotate(ll x){ll y=fa[x],z=fa[y];ll xs=Direct(x),ys=Direct(y);ll w=t[x][xs^1];t[x][xs^1]=y;t[y][xs]=w;if(Nroot(y))t[z][ys]=x;fa[x]=z;fa[y]=x;if(w)fa[w]=y;PushUp(y);PushUp(x);return;}void Splay(ll x){while(Nroot(x)){ll y=fa[x];if(!Nroot(y))Rotate(x);else if(Direct(x)==Direct(y))Rotate(y),Rotate(x);else Rotate(x),Rotate(x);}return;}ll ct(ll x,ll r,ll h){if(t[x][1])return (r-h)*2;return min(r-1,(r-v[x])*2);}void Access(ll x,ll c){Splay(x);ll r=s[x]-s[t[x][0]],h=s[t[x][1]];ans-=ct(x,r,h);v[x]+=c;r+=c;PushUp(x);if(h*2<r+1)w[x]+=h,t[x][1]=0;ans+=ct(x,r,h);PushUp(x);ll y;for(y=x,x=fa[x];x;y=x,x=fa[x]){Splay(x);ll r=s[x]-s[t[x][0]],h=s[t[x][1]];ans-=ct(x,r,h);w[x]+=c;r+=c;if(h*2<r+1)w[x]+=h,t[x][1]=h=0;if(s[y]*2>r)w[x]-=s[y],t[x][1]=y,h=s[y];ans+=ct(x,r,h);PushUp(x);}return;}
}T;
void addl(ll x,ll y){a[++tot].to=y;a[tot].next=ls[x];ls[x]=tot;return;
}
void dp(ll x,ll fa){ll son=0,mx=T.v[x]=s[x];T.fa[x]=fa;for(ll i=ls[x];i;i=a[i].next){ll y=a[i].to;if(y==fa)continue;dp(y,x);s[x]+=s[y];if(s[y]>mx)son=y,mx=s[y];}ans+=min(s[x]-1,(s[x]-mx)*2);if(mx*2>s[x])T.t[x][1]=son;T.w[x]=s[x]-T.v[x]-s[T.t[x][1]];T.s[x]=s[x];return;
}
signed main()
{scanf("%lld%lld",&n,&m);for(ll i=1;i<=n;i++)scanf("%lld",&s[i]);for(ll i=1;i<n;i++){ll x,y;scanf("%lld%lld",&x,&y);addl(x,y);addl(y,x);}dp(1,0);printf("%lld\n",ans);for(ll i=1;i<=m;i++){ll x,w;scanf("%lld%lld",&x,&w);T.Access(x,w);printf("%lld\n",ans);}return 0;
}