bzoj5210最大连通子块和 （动态dp+卡常好题）

卡了一晚上，经历了被卡空间，被卡T，被卡数组等一堆惨惨的事情之后，终于在各位大爹的帮助下过了这个题qwqqq

（全网都没有用矩阵转移的动态dp，让我很慌张）

首先，我们先考虑一个比较基础的\(dp\)

我们令\(dp1[i]\)表示必须选\(i\)的最大连通块的权值是多少。

然后令\(ans1[i]\)表示\(i\)的子树中的\(dp1\)的最大值。

那么该怎么计算这两个数组呢。

首先$$dp1[i] = dp1[i]+max(0,dp1[p])$$ 其中\(p\)是\(i\)的儿子。

\[ans1[i]=max(ans1[i],ans1[p]),ans1[i]=max(ans1[i],dp1[i])
\]

那么朴素的\(dp\)大概就是这样。

应该如何去优化呢。

由于是树上问题，不难想到树链剖分然后划分出轻重儿子。

我们令\(g\)表示考虑重儿子的\(dp\)数组，\(ans\)表示考虑重儿子的\(max\)数组。

然后\(f,ans1\)分别表示不考虑重儿子的\(dp和max\)数组。

若\(p\)是\(x\)的重儿子的话

不难发现，\(g[x]=max(f[x],g[p]+f[x]),ans[x]=max(ans1[x],max(ans[p],g[p]+f[x]))\)

我们这里默认\(ans1[x]\)一开始和\(f[x]\)取过\(max\).

那么我们就可以直接构造如下矩阵

f[x] f[x] -inf
-inf 0 -inf
f[x] ans1[x] 0

来进行转移（注意特判链尾）

但是有一个需要注意的地方，就是我们的\(ans1[x]\)由于涉及到修改，所以需要一个可删堆（或者\(multiset\)）

来进行维护，感觉\(modify\)的时候还是有很多细节。qwq懒得写太多了，直接看代码吧。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#include<queue>
#include<cmath>
#include<map>
#include<set>
#define pb push_back
#define mk make_pair
#define ll long long
#define lson ch[x][0]
#define rson ch[x][1]
#define rint register int
using namespace std;
//char buf[100010],*buff = buf + 100000;

//#define getchar() (buff == buf + 100000 ? (fread(buf,1,100000,stdin),buff = buf) : 0,*buff++)

inline int read()
{
  int x=0,f=1;char ch=getchar();
  while (!isdigit(ch)) {if (ch=='-') f=-1;ch=getchar();}
  while (isdigit(ch)) {x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
  return x*f;
}

const int maxn = 2e5+1e2;
const int maxm = 2*maxn;
const ll inf = -1e15;

struct Node{
    priority_queue<ll> q1,q2;
    inline void push(ll x) {q1.push(x);}
    inline void erase(ll x) {q2.push(x);}
    inline ll top(){while ((!q2.empty()) && (q1.top()==q2.top())) q1.pop(),q2.pop(); return q1.top();}
};

Node ans[maxn];

struct Ju{
   int x,y;
   ll a[3][3];
   Ju operator * (Ju b)
   {
       Ju ans;
       ans.x=x;
       ans.y=b.y;
       for (rint i=0;i<=ans.x;++i)
       {
           for (rint j=0;j<=ans.y;++j)
           {
		      ll uu = inf;
		      for (rint k=0;k<=y;++k)
                uu=max(uu,a[i][k]+b.a[k][j]);
              ans.a[i][j]=uu;
           }
       }
       return ans;
   }
};

int point[maxn],nxt[maxm],to[maxm];
int cnt,n,m;
int top[maxn],newnum[maxn],fa[maxn],son[maxn],size[maxn];
Ju f[4*maxn];
//Ju pre[maxn];
ll dp1[maxn],dp[maxn];
ll ans1[maxn];
int back[maxn];
int tot;
int tail[maxn];
ll val[maxn];
int leaf[maxn];
Ju caonima;

struct pp{
    ll cao,kao;
};

pp cnm[maxn];

inline void addedge(int x,int y)
{
    nxt[++cnt]=point[x];
    to[cnt]=y;
    point[x]=cnt;
}

inline void up(int root)
{
    f[root]=f[2*root+1]*f[2*root];
}

void build(int root,int l,int r)
{
    if (l==r)
    {
        int ymh = back[l];

        if (tail[top[ymh]]==ymh)
        {
            f[root].x=0;
            f[root].y=2;
            f[root].a[0][0]=dp[ymh];
            f[root].a[0][1]=max(dp[ymh],0ll);
        }
        else
        {   f[root].x=f[root].y=2;
            f[root].a[0][0]=dp[ymh];
            f[root].a[0][1]=dp[ymh];
            f[root].a[0][2]=inf;
            f[root].a[1][0]=inf;
            f[root].a[1][2]=inf;
            f[root].a[2][0]=dp[ymh];
            f[root].a[2][1]=ans[ymh].top();
        }
        leaf[ymh]=root;
        return;
    }
    int mid = l+r >> 1;
    build(root<<1,l,mid);
    build(root<<1|1,mid+1,r);
    up(root);
}

void update(int root,int l,int r,int x)
{
    if (l==r)
    {
        f[root]=caonima;
        return;
    }
    int mid = l+r >> 1;
    if (x<=mid) update(root<<1,l,mid,x);
    else update(root<<1|1,mid+1,r,x);
    up(root);
}

Ju query(int root,int l,int r,int x,int y)
{
    if(x<=l && r<=y)
    {
        return f[root];
    }
    int mid = l+r >> 1;
    if (y<=mid) return query(root<<1,l,mid,x,y);
    if (x>mid) return query(root<<1|1,mid+1,r,x,y);
    return query(root<<1|1,mid+1,r,x,y)*query(root<<1,l,mid,x,y);
}

void dfs1(int x,int faa)
{
    size[x]=1;
    int maxson=-1;
    for (int i=point[x];i;i=nxt[i])
    {
        int p = to[i];
        if (p==faa) continue;
        fa[p]=x;
        dfs1(p,x);
        size[x]+=size[p];
        if (size[p]>maxson)
        {
            maxson=size[p];
            son[x]=p;
        }
    }
}

void dfs2(int x,int chain)
{
    top[x]=chain;
    tail[chain]=x;
    newnum[x]=++tot;
    back[tot]=x;
    if (!son[x]) return;
    dfs2(son[x],chain);
    for (int i=point[x];i;i=nxt[i])
    {
        int p = to[i];
        if (!newnum[p]) dfs2(p,p);
    }
}

void solve(int x,int fa)
{
    dp[x]=dp1[x]=val[x];
    ans1[x]=val[x];
    for (rint i=point[x];i;i=nxt[i])
    {
        int p = to[i];
        if (p==fa) continue;
        solve(p,x);
        dp1[x]=max(dp1[x],dp1[x]+dp1[p]);
        ans1[x]=max(ans1[x],ans1[p]);
        if (p!=son[x]) dp[x]=max(dp[x],dp[x]+dp1[p]),ans[x].push(ans1[p]);
    }
    ans[x].push(dp[x]);
    ans[x].push(0);
    //cout<<x<<endl;
    ans1[x]=max(ans1[x],dp1[x]);
    ans1[x]=max(ans1[x],0ll);
    //cout<<x<<endl;
}

void modify(int x,ll y)
{
    Ju tmp = f[leaf[x]];
    if (tail[top[x]]==x)
    {
        tmp.a[0][0]+=y-val[x];
        tmp.a[0][1]=max(0ll,tmp.a[0][0]);
    }
    else
    {
        ans[x].erase(tmp.a[0][0]);
        tmp.a[0][0]+=y-val[x];
        tmp.a[0][1]=tmp.a[0][0];
        ans[x].push(tmp.a[0][0]);
        tmp.a[2][0]=tmp.a[0][0];
        tmp.a[2][1]=ans[x].top();
    }
    val[x]=y;
    caonima=tmp;
    update(1,1,n,newnum[x]);
    //cout<<x<<" "<<top[x]<<endl;
    for (rint now = top[x];now!=1;now=top[now])
    {
        //cout<<now<<endl;
        int faa = fa[now];
        //Ju ymh = query(1,1,n,newnum[faa],newnum[faa]);
        Ju ymh = f[leaf[faa]];
        Ju lyf;
        if (newnum[now]!=newnum[tail[top[now]]])
          lyf= query(1,1,n,newnum[now],newnum[tail[top[now]]]);
        else
          lyf = f[leaf[now]];
        ans[faa].erase(cnm[now].kao);
        ans[faa].erase(ymh.a[0][0]);
        ymh.a[0][0]+=max(lyf.a[0][0],0ll)-max(cnm[now].cao,0ll);
        ymh.a[0][1]=ymh.a[0][0];
        ymh.a[2][0]=ymh.a[0][0];
        ans[faa].push(ymh.a[0][0]);
        ans[faa].push(lyf.a[0][1]);
        ymh.a[2][1]=ans[faa].top();
        caonima=ymh;
        update(1,1,n,newnum[faa]);
        cnm[now].kao=lyf.a[0][1];
        cnm[now].cao=lyf.a[0][0];
        now=fa[now];
    }
}

int q;
char c;

signed main()
{
  //freopen("5210.in","r",stdin);
  //freopen("ymh.out","w",stdout);
  n=read(),q=read();
  //cout<<n<<" "<<q<<endl;
  for (rint i=1;i<=n;i++) val[i]=read();
  for (rint i=1;i<n;i++)
  {
      int x=read(),y=read();
      addedge(x,y);
      addedge(y,x);
  }
  dfs1(1,0);
  dfs2(1,1);
  solve(1,0);
  //cout<<1<<endl;
  build(1,1,n);
  //cout<<ans1[2]<<" "<<dp[2]<<" "<<dp1[2]<<endl;
  for (rint i=1;i<=n;++i)
  {
     Ju ymh = query(1,1,n,newnum[i],newnum[tail[top[i]]]);
     cnm[i].cao=ymh.a[0][0];
     cnm[i].kao=ymh.a[0][1];
  }
  for (rint i=1;i<=q;++i)
  {
      c=getchar();
      while (c<'A' || c>'Z') c=getchar();
      if (c=='M')
      {
         ll x=read(),y=read();
         modify(x,y);
      }
      else
      {
         int x=read();
         Ju ymh = query(1,1,n,newnum[x],newnum[tail[top[x]]]);
         cout<<max(ymh.a[0][1],max(ymh.a[0][0],0ll))<<"\n";
      }
  }
  return 0;
}