P1637 三元上升子序列 （权值线段树优化dp）

题目链接

首先我们很容易想到这题可以用dp来做，
d1[i]表示前i-1个元素中，小于a[i]的元素个数
d2[i]表示前i个元素中三元上升子序列的对数
转移过程非常朴素，见代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn = 1e5+7;
const int INF = 2e9+1;
int d1[maxn],d2[maxn];
int n,a[maxn];
ll ans=0;
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",a+i);
    for(int i=2;i<=n;i++)
    {
        for(int j=1;j<i;j++)
        {
            if(a[j]<a[i]) d1[i]++;
        }
    }
    for(int i=2;i<=n;i++)
    {
        for(int j=1;j<i;j++)
        {
            if(a[j]<a[i]) d2[i]+=d1[j];
        }
        ans+=d2[i];
    }
    printf("%lld\n",ans);
    return 0;
}

但是这样肯定会tle的 ，
我们想到第二层循环的本质是在[0,a[i]-1]这个区间内统计元素个数，可以用桶来做，
但是这里数据量大，一般的桶空间是开不下的。
所以我们可以用权值线段树来维护。
细节见注释

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn = 1e5+7;
const ll inf = 34359738370;
ll tree[maxn*100];
//桶前后维护了2次信息，第一次就是维护区间内数出现的次数
//第二次维护的信息是 叶子节点[x,x]表示d1[x]的值  区间[l,r]表示Σd1[k](l<=k<=r)
int root=1,cnt=1,lc[maxn*100],rc[maxn*100];
int d1[maxn],d2[maxn];
int n,a[maxn];
ll ans=0;
inline void pushup(int rt)
{
    tree[rt]=tree[lc[rt]]+tree[rc[rt]];
    return ;
}
inline void updata(int &rt,ll l,ll r,ll x,int v)
{
    if(!rt) rt=++cnt;
    if(l == r)
    {
        tree[rt]+=v;
        if(tree[rt]<0) tree[rt]=0;//第二次统计前的清空操作
        return ;
    }
    ll mid=(l+r)>>1;
    if(x<=mid) updata(lc[rt],l,mid,x,v);
    else updata(rc[rt],mid+1,r,x,v);
    pushup(rt);
}
inline int query(int rt,ll l,ll r,ll vl,ll vr)
{
    if(!rt || r<vl || l>vr) return 0;
    if(vl<=l && r<=vr) return tree[rt];
    ll mid=(l+r)>>1;
    return query(lc[rt],l,mid,vl,vr)+query(rc[rt],mid+1,r,vl,vr);
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",a+i);
    updata(root,0,inf,a[1],1);//先把第一个元素插入
    for(int i=2;i<=n;i++)
    {
        d1[i]=query(root,0,inf,0,a[i]-1);//nlogn求出d1数组的值
        updata(root,0,inf,a[i],1);//询问完再把当前元素插入线段树
    }
    //开始第二次维护前，需要把线段树清空，这里我们用updata函数 ，memset会tle
    for(int i=1;i<=n;i++) 
    {
        updata(root,0,inf,a[i],-maxn);//全部清空
    }
    for(int i=2;i<=n;i++)
    {
        ans+=query(root,0,inf,0,a[i]-1);
        updata(root,0,inf,a[i],d1[i]);//注意第二次维护的信息是d1[i]
    }
    printf("%lld\n",ans);
    return 0;
}