【树状数组优化dp】【求M元上升子序列个数】UVA12983 The Battle of Chibi

题面

https://www.luogu.com.cn/problem/UVA12983

题意

求在一个序列中m元上升子序列的个数（不用连续）

思路

1. 显然我们可以定义一个状态dp[i][j]表示以长度为i并且以j结尾的子序列个数的个数
2. 转移方程为
   $$dp[i][j]=\sum _{k<j,a[k]<a[j]}dp[i-1][k]$$
3. 但是如果暴力地来转移，显然复杂度是O($n^3$)的
4. 现在就来考虑如何优化：
5. 我们需要满足两个条件，一个是a[k]<a[j]，一个是k<j,怎么样同时满足呢？
6. 我们假设现在把问题从二维简化为一维，也就是现在问题简化成了求
   $$dp[j]=\sum _{k<j,a[k]<a[j]}dp[k]$$
   这个问题，我们很容易想到树状数组求逆序对的问题
   （例题：P1908 逆序对）
   这个问题我们解决的时候for一遍数组，对于每个dp[i]，我们求树状数组中已经放入的小于a[i]的答案和，在一个询问后再把dp[i]，放在a[i]这个位置上，这样下一次查询的时候如果查询的那个数比a[i]大，dp[i]的答案就不会进入那次查询
7. 显然，我们这个题也可以用这个方法解决，dp[i][j]只需要查询线段树中，小于a[j]的数,同时应当把dp[i-1][j]更新到a[j]的位置
   (? 因为dp[i][j]的答案只对第i+1层有影响，在dp[i][j+1]查询时候dp[i][j]还不能放到树状数组中去),
   每次清空一下树状数组即可
   (? 因为i的答案是由i-1转移过来的，对其他层的答案没有影响，所以需要清空掉)

---

线段树按理也能过，但是卡常了，而且线段树在清空的时候的复杂度是O($nlog(n)$)的，而树状数组的清空只需要memset©即可，O($n$)的复杂度，所以这个题用树状数组解决更优

代码

#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
#include<vector>
#include<algorithm>
#define ls k<<1
#define rs k<<1|1
using namespace std;
typedef long long ll;
const int maxn=1100;
const int mod=1e9+7;
int n,m,a[maxn],b[maxn],lisan,t;
ll dp[maxn][maxn];
ll c[maxn];
int lowbit(int x)
{
    return x&(-x);
}
ll ask(int x)
{
    ll ans=0;
    while(x)
    {
        ans=(ans+c[x])%mod;
        x-=lowbit(x);
    }
    return ans;
}
void add(int x,ll val)
{
    while(x<=lisan)
    {
        c[x]=(c[x]+val)%mod;
        x+=lowbit(x);
    }
}
int main()
{
    scanf("%d",&t);
    for(int cas=1;cas<=t;cas++)
    {
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            b[i]=a[i];
        }
        sort(b+1,b+1+n);
        lisan=unique(b+1,b+1+n)-(b+1);
        for(int i=1;i<=n;i++)
        {
            a[i]=lower_bound(b+1,b+1+lisan,a[i])-b;
            dp[1][i]=1;
        }
        for(int i=2;i<=m;i++)
        {
            memset(c,0,sizeof(c));
            for(int j=1;j<=n;j++)
            {
                dp[i][j]=ask(a[j]-1);
                add(a[j],dp[i-1][j]);
            }

        }
        ll ans=0;
        for(int i=1;i<=n;i++)
            ans=(ans+dp[m][i])%mod;
        printf("Case #%d: %lld\n",cas,ans);
    }
    return 0;
}