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

题目大意:

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

思路:

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

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

那么递推式如下:

dp[i][j]=\sum _{k=1}^{i-1}(a[k]<a[i])*dp[k][j-1]

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

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

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

思维难度几乎就没有了。

复杂度O(nklogn)

#include 
#define int long long
#define debug cout<<"debug "<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

 

你可能感兴趣的:(线段树,树状数组,动态规划)