Codeforce 597C(dp+树状数组优化)

链接：点击打开链接

题意：求给出的n个数中，上升子序列长度是k+1的种数

代码：

#include <stack>
#include <cstdio>
#include <cstring>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
long long dp[20][100005];
long long a[100005],bit[100005];
long long sum(long long i){
    long long s=0;
    while(i>0){
        s+=bit[i];
        i-=i&-i;
    }
    return s;
}
void add(long long i,long long x){
    while(i<=100000){
        bit[i]+=x;
        i+=i&-i;
    }
}
int main(){
    long long n,k,i,j,ans;                      //dp[i][j]表示长度是i，最后一位是a[j]的种数
    while(scanf("%I64d%I64d",&n,&k)!=EOF){      //dp[i][j]+=dp[i-1][1,2,....,a[i-1]];
        memset(dp,0,sizeof(dp));                //其实就是用前缀和进行转移，所以用树状数组维护
        for(i=1;i<=n;i++){
            scanf("%I64d",&a[i]);
            dp[1][a[i]]=1;
        }
        for(i=2;i<=k+1;i++){
            memset(bit,0,sizeof(bit));
            for(j=2;j<=n;j++){                  //每次放长度是i-1的
                add(a[j-1],dp[i-1][a[j-1]]);
                dp[i][a[j]]=sum(a[j]-1);        //最后求a[j]之前的个数
            }
        }
        ans=0;
        for(i=1;i<=n;i++)
        ans+=dp[k+1][a[i]];
        printf("%I64d\n",ans);
    }
    return 0;
}