题目描述
一天,小魂正和一个数列玩得不亦乐乎。
小魂的数列一共有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;
}