牛客练习赛56C.小魂和他的数列（dp+树状数组）

题目大意：

给定一个n元素的数列，第i个数为Ai，求这个数列中一共有多少个长度为K的子序列是严格递增的。

思路：

线段树写多了，看见n这么大，感觉不能开k颗线段树呀，那怎么做呀？都忘了还有树状数组了23333

dp(i,j)表示以i元素为结尾的，长度为j的严格递增子序列的个数。

那么递推式如下：

所以就变成如何快速计算比i小且a[k]比a[i]小的dp(k,j-1)的和了。

发现k很小，那么就可以想到开k颗树状数组，每个树状数组存储上面所说的前缀和。

不过a[i]很大，还得离散化以下。

思维难度几乎就没有了。

复杂度O(nklogn)

#include <bits/stdc++.h>
#define int long long
#define debug cout<<"debug "<<endl;
using namespace std;

const int maxn=5e5+10;
int dp[maxn][11];
const int mod=998244353;
int a[maxn];
int sum[11][maxn];
int n,k;
vector<int>v;
int getid(int cur){
    return lower_bound(v.begin(),v.end(),cur)-v.begin()+1;
}
int lowbit(int x){
    return x&-x;
}
void add(int *sum,int x,int v){
    while(x<=n){
        sum[x]+=v;
        sum[x]%=mod;
        x+=lowbit(x);
    }
}
int query(int *sum,int x){
    int res=0;
    while(x){
        res+=sum[x];
        res%=mod;
        x-=lowbit(x);
    }
    return res;
}
int pos[maxn];
signed main(){
    scanf("%lld%lld",&n,&k);
    for(int i=1;i<=n;i++){
        scanf("%lld",&a[i]);
        v.push_back(a[i]);
    }
    sort(v.begin(),v.end());
    v.erase(unique(v.begin(),v.end()),v.end());
    for(int i=1;i<=n;i++){
        int id=getid(a[i]);
        pos[i]=id;
    }
    for(int i=1;i<=n;i++){
        int now=pos[i];
        add(sum[1],now,1);
        for(int j=2;j<=k;j++){
            if(i<j){
                continue;
            }
            int id=pos[i];
            int Sum=query(sum[j-1],id-1)%mod;
            dp[i][j]+=Sum;
            dp[i][j]%=mod;
            //printf("debug %lld %lld %lld\n",i,j,dp[i][j]);
            add(sum[j],id,dp[i][j]);
        }
    }
    int ans=0;
    for(int i=1;i<=n;i++){
        ans+=dp[i][k];
        ans%=mod;
    }
    printf("%lld\n",ans);
    return 0;

}