BZOJ2001 HNOI2010城市建设（线段树分治+LCT）

一个很显然的思路是把边按时间段拆开线段树分治一下，用lct维护MST。理论上复杂度是O((M+Q)logNlogQ)，实际常数爆炸T成狗。正解写不动了。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<vector>
#include<stack>
using namespace std;
int read()
{
    int x=,f=;char c=getchar();
    while (c<''||c>'') {if (c=='-') f=-;c=getchar();}
    while (c>=''&&c<='') x=(x<<)+(x<<)+(c^),c=getchar();
    return x*f;
}
#define N 20010
#define M 50010
#define lson tree[k].ch[0]
#define rson tree[k].ch[1]
#define lself tree[tree[k].fa].ch[0]
#define rself tree[tree[k].fa].ch[1]
int n,m,q,f[M],L[M<<],R[M<<],cnt;
long long tot=,ans[M];
struct edge{int x,y,z;
}e[M],a[N+M<<];
struct data{int i,x;};
vector<data> Q[M];
vector<int> tree[M<<];
stack<int> undo[M<<];
namespace lct
{
    struct data{int ch[],fa,rev,x;}tree[N+M<<];
    void up(int k)
    {
        tree[k].x=k;
        if (a[tree[lson].x].z>a[tree[k].x].z) tree[k].x=tree[lson].x;
        if (a[tree[rson].x].z>a[tree[k].x].z) tree[k].x=tree[rson].x;
    }
    void rev(int k){if (k) swap(lson,rson),tree[k].rev^=;}
    void down(int k){if (tree[k].rev) rev(lson),rev(rson),tree[k].rev=;}
    bool isroot(int k){return lself!=k&&rself!=k;}
    int whichson(int k){return rself==k;}
    void push(int k){if (!isroot(k)) push(tree[k].fa);down(k);}
    void move(int k)
    {
        int fa=tree[k].fa,gf=tree[fa].fa,p=whichson(k);
        if (!isroot(fa)) tree[gf].ch[whichson(fa)]=k;tree[k].fa=gf;
        tree[fa].ch[p]=tree[k].ch[!p],tree[tree[k].ch[!p]].fa=fa;
        tree[fa].fa=k,tree[k].ch[!p]=fa;
        up(fa),up(k);
    }
    void splay(int k)
    {
        push(k);
        while (!isroot(k))
        {
            int fa=tree[k].fa;
            if (!isroot(fa))
                if (whichson(fa)^whichson(k)) move(k);
                else move(fa);
            move(k);
        }
    }
    void access(int k){for (int t=;k;t=k,k=tree[k].fa) splay(k),tree[k].ch[]=t,up(k);}
    void makeroot(int k){access(k),splay(k),rev(k);}
    int findroot(int k){access(k),splay(k);for (;lson;k=lson) down(k);splay(k);return k;}
    void link(int x,int y){makeroot(x);tree[x].fa=y;}
    void cut(int x,int y){makeroot(x),access(y),splay(y);tree[y].ch[]=tree[x].fa=;up(y);}
    int query(int x,int y){makeroot(x),access(y),splay(y);return tree[y].x;}
    void newpoint(int k){tot+=a[k].z;link(k,a[k].x),link(k,a[k].y);}
    void delpoint(int k){tot-=a[k].z;cut(k,a[k].x),cut(k,a[k].y);}
}
void build(int k,int l,int r)
{
    L[k]=l,R[k]=r;
    if (l==r) return;
    int mid=l+r>>;
    build(k<<,l,mid);
    build(k<<|,mid+,r);
}
void add(int k,int l,int r,int e)
{
    if (L[k]==l&&R[k]==r) {tree[k].push_back(e);return;}
    int mid=L[k]+R[k]>>;
    if (r<=mid) add(k<<,l,r,e);
    else if (l>mid) add(k<<|,l,r,e);
    else add(k<<,l,mid,e),add(k<<|,mid+,r,e);
}
void solve(int k)
{
    int s=tree[k].size();
    for (int i=;i<s;i++)
    if (lct::findroot(a[tree[k][i]].x)!=lct::findroot(a[tree[k][i]].y))
    lct::newpoint(tree[k][i]),undo[k].push(tree[k][i]);
    else
    {
        int t=lct::query(a[tree[k][i]].x,a[tree[k][i]].y);
        if (a[t].z>a[tree[k][i]].z)
        {
            lct::delpoint(t),lct::newpoint(tree[k][i]);
            undo[k].push(-t),undo[k].push(tree[k][i]);
        }
    }
    if (L[k]==R[k]) ans[L[k]]=tot;
    else solve(k<<),solve(k<<|);
    while (!undo[k].empty())
    {
        if (undo[k].top()<) lct::newpoint(-undo[k].top());
        else lct::delpoint(undo[k].top());
        undo[k].pop();
    }
}
int main()
{
    freopen("bzoj2001.in","r",stdin);
    freopen("bzoj2001.out","w",stdout);
    n=read(),m=read(),q=read();
    for (int i=;i<=m;i++)
    e[i].x=read(),e[i].y=read(),e[i].z=read();
    for (int i=;i<=q;i++)
    {
        int x=read(),y=read();
        Q[x].push_back((data){i,y});
    }
    build(,,q);
    cnt=n;
    for (int i=;i<=m;i++)
    {
        int s=Q[i].size();
        if (s==) a[++cnt]=e[i],add(,,q,cnt);
        else
        {
            if (Q[i][].i>) a[++cnt]=e[i],add(,,Q[i][].i-,cnt);
            for (int j=;j<s-;j++)
            a[++cnt]=(edge){e[i].x,e[i].y,Q[i][j].x},
            add(,Q[i][j].i,Q[i][j+].i-,cnt);
            a[++cnt]=(edge){e[i].x,e[i].y,Q[i][s-].x},
            add(,Q[i][s-].i,q,cnt);
        }
    }
    solve();
    for (int i=;i<=q;i++) printf("%lld\n",ans[i]);
    fclose(stdin);fclose(stdout);
    return ;
}