bzoj 3995: [SDOI2015]道路修建 线段树

题意

给出一个2行n列的图，只有相邻的点有连边，要求资瓷两个操作：
Q l r查询第l列到第r列的最小生成树
C x1 y1 x2 y2 w修改边权
n,m<=60000,w<=10000

分析

第一想法是lct来xjb维护一下，发现不会做。。。想了一下，想到了一个用set维护待修改莫队的方法。。。然后又想了一下，想到了一个n*m的做法。。。
毕竟太弱。。。
直接线段树维护即可。
考虑合并区间[l,mid],[mid,r]
线段树每个节点维护该区间的mst最左边的竖线位置，最右边的竖线位置，横线的最大值，左端点到最左边竖线中横线最大值，右端点到最右边竖线中横线的最大值，还有答案，合并的时候瞎讨论一下去掉那条边即可即可。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define N 60005
#define MAX 60005*10-5
using namespace std;

int n,m,c[N],v[N][2];
struct tree{int l,r,lmx,rmx,mx,ans;}t[N*10];

void updata(int d,int l,int r)
{
    t[d].mx=max(t[l].mx,t[r].mx);
    t[d].ans=t[l].ans+t[r].ans;
    t[d].l=t[l].l;t[d].lmx=t[l].lmx;
    t[d].r=t[r].r;t[d].rmx=t[r].rmx;
    int mx=max(t[l].rmx,t[r].lmx);
    if (t[l].r==t[r].l) t[d].ans-=c[t[l].r];
    else if (mx>=max(c[t[l].r],c[t[r].l])) t[d].ans-=mx;
    else if (c[t[l].r]>c[t[r].l])
    {
        t[d].ans-=c[t[l].r];
        if (t[l].l==t[l].r) t[d].l=t[r].l,t[d].lmx=max(t[l].mx,t[r].lmx);
    }
    else
    {
        t[d].ans-=c[t[r].l];
        if (t[r].l==t[r].r) t[d].r=t[l].r,t[d].rmx=max(t[r].mx,t[l].rmx);
    }
}

void build(int d,int l,int r)
{
    if (l+1==r)
    {
        t[d].mx=max(v[l][0],v[l][1]);
        if (t[d].mx>=max(c[l],c[r]))
        {
            t[d].l=l;t[d].r=r;t[d].lmx=t[d].rmx=0;t[d].ans=v[l][0]+v[l][1]+c[l]+c[r]-t[d].mx;
        }
        else if (c[l]>c[r])
        {
            t[d].l=t[d].r=r;t[d].lmx=t[d].mx;t[d].rmx=0;t[d].ans=v[l][0]+v[l][1]+c[r];
        }
        else
        {
            t[d].l=t[d].r=l;t[d].rmx=t[d].mx;t[d].lmx=0;t[d].ans=v[l][0]+v[l][1]+c[l];
        }
        return;
    }
    int mid=(l+r)/2;
    build(d*2,l,mid);build(d*2+1,mid,r);
    updata(d,d*2,d*2+1);
}

void ins(int d,int l,int r,int x,int y)
{
    if (x>y) return;
    if (l+1==r)
    {
        t[d].mx=max(v[l][0],v[l][1]);
        if (t[d].mx>=max(c[l],c[r]))
        {
            t[d].l=l;t[d].r=r;t[d].lmx=t[d].rmx=0;t[d].ans=v[l][0]+v[l][1]+c[l]+c[r]-t[d].mx;
        }
        else if (c[l]>c[r])
        {
            t[d].l=t[d].r=r;t[d].lmx=t[d].mx;t[d].rmx=0;t[d].ans=v[l][0]+v[l][1]+c[r];
        }
        else
        {
            t[d].l=t[d].r=l;t[d].rmx=t[d].mx;t[d].lmx=0;t[d].ans=v[l][0]+v[l][1]+c[l];
        }
        return;
    }
    int mid=(l+r)/2;
    ins(d*2,l,mid,x,min(y,mid));
    ins(d*2+1,mid,r,max(x,mid),y);
    updata(d,d*2,d*2+1);
}

tree query(int d,int l,int r,int x,int y)
{
    if (l==x&&r==y) return t[d];
    int mid=(l+r)/2;
    if (y<=mid) return query(d*2,l,mid,x,y);
    else if (x>=mid) return query(d*2+1,mid,r,x,y);
    else
    {
        tree u=query(d*2,l,mid,x,mid),v=query(d*2+1,mid,r,mid,y);
        t[MAX-2]=u;t[MAX-1]=v;
        updata(MAX,MAX-2,MAX-1);
        return t[MAX];
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<n;i++) scanf("%d",&v[i][0]);
    for (int i=1;i<n;i++) scanf("%d",&v[i][1]);
    for (int i=1;i<=n;i++) scanf("%d",&c[i]);
    build(1,1,n);
    for (int i=1;i<=m;i++)
    {
        char op[2];
        scanf("%s",op);
        if (op[0]=='Q')
        {
            int l,r;
            scanf("%d%d",&l,&r);
            if (l==r) printf("%d\n",c[l]);
            else
            {
                tree u=query(1,1,n,l,r);
                printf("%d\n",u.ans);
            }
        }
        else
        {
            int x1,y1,x2,y2,w;
            scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&w);
            if (x1>x2) swap(x1,x2);
            if (y1>y2) swap(y1,y2);
            if (x1==x2) v[y1][x1-1]=w;
            else c[y2]=w;
            ins(1,1,n,y1,y2);
        }
    }
    return 0;
}