LOJ #2831. 「JOISC 2018 Day 1」道路建设 线段树+Link-cut-tree

用 LCT 维护颜色相同连通块，然后在线段树上查一下逆序对个数就可以了.

code:

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <string>
#define N 100005
#define ll long long
using namespace std;
namespace IO
{
    void setIO(string s)
    {
        string in=s+".in";
        string out=s+".out";
        freopen(in.c_str(),"r",stdin);
        // freopen(out.c_str(),"w",stdout);
    }
};
ll ans;
int n,val[N],A[N];
namespace seg
{
    int lazy[N<<2],sum[N<<2];
    void mark(int now) { sum[now]=0,lazy[now]=1; }
    void pushdown(int now)
    {
        if(lazy[now])
            mark(now<<1),mark(now<<1|1),lazy[now]=0;
    }
    void update(int l,int r,int now,int p,int v)
    {
        if(l==r) { sum[now]+=v;   return; }
        pushdown(now);
        int mid=(l+r)>>1;
        if(p<=mid)
            update(l,mid,now<<1,p,v);
        else
            update(mid+1,r,now<<1|1,p,v);
        sum[now]=sum[now<<1]+sum[now<<1|1];
    }
    int query(int l,int r,int now,int L,int R)
    {
        if(L>R)
            return 0;
        if(l>=L&&r<=R)
            return sum[now];
        pushdown(now);
        int re=0,mid=(l+r)>>1;
        if(L<=mid)
            re+=query(l,mid,now<<1,L,R);
        if(R>mid)
            re+=query(mid+1,r,now<<1|1,L,R);
        return re;
    }
};
namespace LCT
{
    #define lson s[x].ch[0]
    #define rson s[x].ch[1]
    struct node {
        int ch[2],rev,f,tag,size;
    }s[N];
    int sta[N];
    inline int get(int x) { return s[s[x].f].ch[1]==x; }
    inline int isr(int x) { return s[s[x].f].ch[0]!=x&&s[s[x].f].ch[1]!=x; }
    inline void pushup(int x)     { s[x].size=s[lson].size+s[rson].size+1; }
    inline void mark(int x,int v) { s[x].tag=v; }
    inline void pushdown(int x)
    {
        if(lson) mark(lson,s[x].tag);
        if(rson) mark(rson,s[x].tag);
    }
    inline void rotate(int x)
    {
        int old=s[x].f,fold=s[old].f,which=get(x);
        if(!isr(old))
            s[fold].ch[s[fold].ch[1]==old]=x;
        s[old].ch[which]=s[x].ch[which^1];
        if(s[old].ch[which])
            s[s[old].ch[which]].f=old;
        s[x].ch[which^1]=old,s[old].f=x,s[x].f=fold;
        pushup(old),pushup(x);
    }
    inline void splay(int x)
    {
        int v=0,u=x,fa;
        for(sta[++v]=u;!isr(u);u=s[u].f)
            sta[++v]=s[u].f;
        for(;v;--v)
            pushdown(sta[v]);
        for(u=s[u].f;(fa=s[x].f)!=u;rotate(x))
            if(s[fa].f!=u)
                rotate(get(fa)==get(x)?fa:x);
    }
    inline void Access(int x,int v)
    {
        for(int y=0;x;y=x,x=s[x].f)
        {
            splay(x),rson=0,pushup(x);
            ans+=(ll)s[x].size*seg::query(1,n,1,1,s[x].tag-1);
            seg::update(1,n,1,s[x].tag,s[x].size);
            rson=y,s[x].tag=v,pushup(x);
        }
    }
    inline void link(int x,int y) { s[y].f=x; }
    #undef lson
    #undef rson
};
int main()
{
    // IO::setIO("input");
    int i,j;
    scanf("%d",&n);
    for(i=1;i<=n;++i)
        scanf("%d",&val[i]),A[i]=val[i];
    sort(A+1,A+1+n);
    for(i=1;i<=n;++i)
        val[i]=lower_bound(A+1,A+1+n,val[i])-A,LCT::s[i].tag=val[i];
    for(i=1;i<n;++i)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        ans=0;
        seg::mark(1);
        LCT::Access(x,val[y]);
        LCT::link(x,y);
        printf("%lld\n",ans);
    }
    return 0;
}