题意:
给你一个序列由1~n组成,找出长度为k的严格递减的子序列一共有多少个
思路:
可以定义状态dp[i][j],来表示以第i项结尾,长度为j的递减序列有多少个。那么转移方程就很容易写出来,即:
d
p
[
i
]
[
j
]
=
∑
d
p
[
k
]
[
j
−
1
]
      
(
k
<
i
  
&
&
  
a
k
>
a
i
)
dp[i][j] = \sum dp[k][j - 1] \;\;\; (k < i \;\&\&\; a_k > a_i)
dp[i][j]=∑dp[k][j−1](k<i&&ak>ai)
但是如果直接遍历前面所有的项找k,然后求和的话,时间复杂度是明显不允许的。但它是所有在i项之前,大于
a
i
a_i
ai的项求和,我们可以想办法按照一定的顺序排一下序,然后利用线段树或者树状数组来求和,这样求和的复杂度就降到了log(n),总的时间复杂度为n * k * log(n),完全可行。
但我们应该怎么建树呢?既然我们要的是所有大于
a
i
a_i
ai的dp值的和,那么我们就可以按照所给序列的值排序后建树,也就是1~n的顺序来建树,每个叶子节点的值就存对应i项结尾的dp值,这样求和就没有问题了。
#include <iostream>
#include <string>
#include <cstring>
#include <cstdio>
using namespace std;
const int maxn = 20200;
const int mod = 1e9;
long long int a[maxn], dp[maxn][12], tree[maxn * 4];
int n, k, M;
long long int Query(int s, int t) //这里我用了zkw线段树,可能比树状数组慢点,但比普通线段树应该还是快不少的
{
long long int ans = 0;
for(s += M - 1, t += M + 1; s ^ t ^ 1; s >>= 1, t >>= 1)
{
if(~ s & 1)
ans = (ans + tree[s ^ 1]) % mod;
if(t & 1)
ans = (ans + tree[t ^ 1]) % mod;
}
return ans % mod;
}
int main()
{
//freopen("in.txt", "r", stdin);
cin >> n >> k;
memset(tree, 0, sizeof tree);
for(int i = 1; i <= n; ++ i)
{
scanf("%lld", &a[i]);
dp[i][1] = 1;
}
for(M = 1; M <= n + 1; M <<= 1);
long long int ans = 0;
for(int j = 2; j <= k; ++ j)
{
memset(tree, 0, sizeof tree);
for(int i = 1; i <= n; ++ i)
{
dp[i][j] = Query(a[i], n);
int cnt = ((M + a[i]) << 1);
while(cnt >>= 1)
tree[cnt] += dp[i][j - 1];
}
}
for(int i = 1; i <= n; ++ i)
ans = (ans + dp[i][k]) % mod;
cout << ans << endl;
return 0;
}