洛谷 P1637 三元上升子序列（权值线段树+离散化）

 

 

和求逆序对的题目差不多，还是按照桶排的思想，将线段树上的每一个点映射到对应位置，然后动态加点，统计每个数 a[j] 之前有多少个 a[i] 满足题目要求，将线段树清空后，再从后遍历一遍数组，寻找 满足 a[k] 的个数，根据乘法原理，两边求得的数相乘即可得到答案，当然对数据需要离散化才可以，用树状数组的话码量会小一点

const int N=3e4+5;

    int i,j,k;
    int n,m,t;
    ll a[N];
    vector <ll> v;
    struct Node
    {
        int l,r;
        ll sum;
    }node[N<<2];
    ll ans[N];

void build(int l,int r,int id)
{
    node[id].l=l,node[id].r=r;
    node[id].sum=0;
    if(l==r) return ;
    else{
        int mid=l+r>>1;
        build(l,mid,id<<1);
        build(mid+1,r,id<<1|1);
    }
}

int get_id(ll x)
{
    return lower_bound(v.begin(),v.end(),x)-v.begin()+1;
}

void push_up(int id)
{
    node[id].sum=node[id<<1].sum+node[id<<1|1].sum;
}

void update(int id,int val)
{
    int L=node[id].l,R=node[id].r;
    if(L>=val && val>=R) node[id].sum++;
    else{
        int mid=L+R>>1;
        if(mid>=val) update(id<<1,val);
        if(val>=mid+1) update(id<<1|1,val);
        push_up(id);
    }
}

ll query(int id,int val,int tag)
{
    int L=node[id].l,R=node[id].r;
    if(L>=val && val>=R) return node[id].sum;
    else{
        int mid=L+R>>1;
        ll ans=0;
        if(tag){
            if(mid>=val) ans+=query(id<<1,val,tag);
            if(val>=mid+1) ans+=query(id<<1|1,val,tag)+node[id<<1].sum;
        } else{
            if(mid>=val) ans+=query(id<<1,val,tag)+node[id<<1|1].sum;
            if(val>=mid+1) ans+=query(id<<1|1,val,tag);
        }
        return ans;
    }
}

int main()
{
    //IOS;
    while(~sd(n)){
        for(int i=1;i<=n;i++) sll(a[i]),v.pb(a[i]);
        sort(v.begin(),v.end());
        v.erase(unique(v.begin(),v.end()),v.end());
        int num=v.size();
        build(1,num+1,1);
        for(int i=1;i<=n;i++){
            int id=get_id(a[i]);
            ans[i]=query(1,id-1,1);
            update(1,id);
        }
        ll ANS=0;
        for(int i=0;i<(N<<2);i++) node[i].sum=0;
        for(int i=n;i;i--){
            int id=get_id(a[i]);
            ANS+=(ans[i]*=query(1,id+1,0));
            //dbg(ans[i]);
            update(1,id);
        }
        pll(ANS);
    }
    //PAUSE;
    return 0;
}