题目描述
一天,小魂正和一个数列玩得不亦乐乎。
小魂的数列一共有n个元素,第i个数为Ai。
他发现,这个数列的一些子序列中的元素是严格递增的。
他想知道,这个数列一共有多少个长度为K的子序列是严格递增的。
请你帮帮他,答案对998244353取模。
对于100%的数据,1≤ n ≤ 500,000,2≤ K ≤ 10,1≤ Ai ≤ 109。
输入描述:
第一行包含两个整数n,K,表示数列元素的个数和子序列的长度。
第二行包含n个整数,表示小魂的数列。
输出描述:
一行一个整数,表示长度为K的严格递增子序列的个数对998244353取模的值。
示例1
题解:
用第i个树状数组维护长度为i的子序列个数
每次递推更新,最后输出第k个树状数组里有多少个数字即可
注意数字太大,而且我们只需要知道每个数的大小关系,所以先离散化一下
AC代码:
#pragma GCC optimize(2)
#include<bits/stdc++.h>
#include<ext/rope>
using namespace std;
using namespace __gnu_cxx;
#define LL long long
#define pii pair<int,int>
#define mp(a,b) make_pair(a,b)
const int MAXN = 5e5+10;
const int MOD = 1e9+7;
const int INF = 0x3f3f3f3f;
int n,k;
struct Tree{
int bit[MAXN];
inline int lowbit(int x){ return x&-x; }
inline void add(int x,int val){ for(int i=x;i<=n;i+=lowbit(i)) bit[i]=(bit[i]+val)%MOD; }
inline LL query(int x,LL res=0){ for(int i=x;i;i-=lowbit(i)) res=(res+bit[i])%MOD; return res; }
}t[12];
int a[MAXN]; vector<int> v;
signed main(){
#ifndef ONLINE_JUDGE
freopen("C:\\Users\\Administrator\\Desktop\\in.txt","r",stdin);
#endif // ONLINE_JUDGE
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++) scanf("%d",&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++) a[i]=lower_bound(v.begin(),v.end(),a[i])-v.begin()+1;
for(int i=1;i<=n;i++){
t[1].add(a[i],1);//长度为1的,以a[i]结尾的子序列+1
for(int j=2;j<=k;j++) t[j].add(a[i],t[j-1].query(a[i]-1));//然后递推
}
printf("%lld\n",t[k].query(n));
return 0;
}