小魂和他的数列

题目描述
一天,小魂正和一个数列玩得不亦乐乎。
小魂的数列一共有n个元素,第i个数为Ai。
他发现,这个数列的一些子序列中的元素是严格递增的。
他想知道,这个数列一共有多少个长度为K的子序列是严格递增的。
请你帮帮他,答案对998244353取模。
对于100%的数据,1≤ n ≤ 500,000,2≤ K ≤ 10,1≤ Ai ≤ 109。
输入描述:
第一行包含两个整数n,K,表示数列元素的个数和子序列的长度。
第二行包含n个整数,表示小魂的数列。
输出描述:
一行一个整数,表示长度为K的严格递增子序列的个数对998244353取模的值。
示例1
输入
复制
5 3
2 3 3 5 1
输出
复制
2
说明
两个子序列分别是2 3 3 5 1和2 3 3 5 1。


没什么思维,用BIT维护过LIS的应该都会。看到k这么小,很明显,直接递推咯。

从前往后,维护k个BIT,每一个维护的就是当前长度为K的个数,然后每个都可以从k-1递推得到。

每次统计一下答案即可。

别忘了BIT里面也需要取模。


AC代码:

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int p=998244353;
const int N=5e5+10;
int d[N][15],n,k,a[N],m,res;
vector<int> v;
inline void add(int x,int v,int k){for(;x<=m;x+=(x&(-x))) d[x][k]=(d[x][k]+v)%p;}
inline int ask(int x,int k){int s=0; for(;x;x-=(x&(-x))) s=(s+d[x][k])%p; return s;}
signed main(){
    ios::sync_with_stdio(false); cin.tie(nullptr); cout.tie(nullptr);
    cin>>n>>k;
    for(int i=1;i<=n;i++)   cin>>a[i],v.push_back(a[i]);
    sort(v.begin(),v.end()); v.erase(unique(v.begin(),v.end()),v.end());
    m=v.size();
    for(int i=1;i<=n;i++)   a[i]=lower_bound(v.begin(),v.end(),a[i])-v.begin()+1;
    for(int i=1;i<=n;i++){
        add(a[i],1,1);	int s=ask(a[i],k);
        for(int j=2;j<=k;j++)   add(a[i],ask(a[i]-1,j-1),j);
        res=(res+ask(a[i],k)-s+p)%p;
    }
    cout<<res<<'\n';
    return 0;
}
发布了416 篇原创文章 · 获赞 228 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/weixin_43826249/article/details/103739788
今日推荐