inversions URAL - 1523(dp + 线段树)

题意:

给你一个序列由1~n组成,找出长度为k的严格递减的子序列一共有多少个

思路:
可以定义状态dp[i][j],来表示以第i项结尾,长度为j的递减序列有多少个。那么转移方程就很容易写出来,即:
d p [ i ] [ j ] = ∑ d p [ k ] [ j − 1 ] &ThickSpace;&ThickSpace;&ThickSpace; ( k &lt; i &ThickSpace; &amp; &amp; &ThickSpace; a k &gt; a i ) dp[i][j] = \sum dp[k][j - 1] \;\;\; (k &lt; i \;\&amp;\&amp;\; a_k &gt; a_i) dp[i][j]=dp[k][j1](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;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值