DP + 树状数组例题

例题一：HDU 3030 Increasing Speed Limits

题意：

求一组数中有多少严格上升子序列。

题解：

由于要求严格上升，因此相同大小的数只能出现一次，我们开始先对原始数组进行排序去重操作。得到的新数组就是离散化的数组，每个数对应的下标表明它是第几大的元素。

这里的状态表示就是$dp[i]$表示以第$i$大的元素结尾的严格上升子序列的个数。

显然，对于当前的状态，可以转移到所有满足的$d[j]$，这里的$d[j]$要求$a[j]<a[i]\&\&j<i$。这样最终的答案就是${\sum }_{i=1}^{n}dp[i]$。

对于原数组中的每个数，通过二分找到他在离散数组中的位置，然后，使用树状数组快速求出在该位置之前有多少可加入的子序列个数。这里$dp[i]$之所以等于前缀和+1，是因为，单独的一个元素也算一个上升序列。然后求和之后，在该位置更新树状数组即可。

实现细节见代码：

#include <bits/stdc++.h>
using namespace std;
#define int long long
typedef long long ll;
const int MAXN = 5e5 + 10;
const int mod = 1e9 + 7;
int a[MAXN], b[MAXN], now[MAXN], sum[MAXN], dp[MAXN];
int lowbit(int x) {
	return x & -x;
}
void add(int x, int k) {
	for (int i = x; i < 500005; i += lowbit(i)) {
		sum[i] = (sum[i] + k) % mod;
	}
}
int getsum(int x) {
	int ans = 0;
	for (int i = x; i; i -= lowbit(i)) {
		ans = (ans + sum[i]) % mod;
	}
	return ans;
}
signed main() {
	ios::sync_with_stdio(false);
	cin.tie(0), cout.tie(0);
	int t, kase = 0;
	cin >> t;
	while (t--) {
		memset(sum, 0, sizeof sum);
		int n, m, x, y, z;
		cin >> n >> m >> x >> y >> z;
		for (int i = 0; i < m; i++) {
			cin >> a[i];
		}
		for (int i = 0; i < n; i++) {
			b[i + 1] = a[i % m]; // 记录原始数组
			now[i + 1] = b[i + 1]; // 离散数组
			a[i % m] = (x * a[i % m] + y * (i + 1)) % z;
		}
		int ans = 0;
		sort(now + 1, now + 1 + n);
		int len = unique(now + 1, now + 1 + n) - now - 1; // 进行离散化操作，并求出离散之后数组的长度
		for (int i = 0; i < n; i++) {
			int pos = lower_bound(now + 1, now + 1 + len, b[i + 1]) - now; // 找到原始数组元素离散之后的位置
			dp[i] = getsum(pos - 1) + 1; // 由于单独的一个元素也算一个序列，因此加1
			add(pos, dp[i]); // 在该位置更新
			ans = (ans + dp[i]) % mod;
		}
		cout << "Case #" << ++kase << ": " << ans << endl;
	}
	return 0;
}

例题二：HDU 3450 Counting Sequences

题意：

求满足相邻两个数之差不超过$d$的子序列个数。

题解：

这道题首先可以想到一个$dp$做法，也就是，对于每一个$i$，$d[i]$表示以$i$结尾的子序列的个数。$d[i]$可以转移到满足$j\ge 1\&\&j<i\&\&abs(a[j]-a[i])\le d$的$d[j]$状态。

这是个$O(n^2)$的做法，显然要超时。

下面我们就要进行优化。

我们可以发现，对每个$d[i]$，其累加的部分是$[a[i]-d,a[i]+d]$范围内且在$i$之前出现过的所有的$d[j]$，这符合树状数组的特点：求连续和，单点修改。

因此，我们可以在每次更新$d[i]$之后，在当前位置上加上$d[i]+1$，注意，$d[i]$本事是不能加一的，因为该题一个元素不能看作一个合法的子序列。我们在后面更新时+1，是因为，后面的元素可以和当前元素组成一个合法子序列。

参考链接

实现细节见代码：

#include <bits/stdc++.h>
using namespace std;
#define endl '\n'
#define int long long
typedef long long ll;
const int MAXN = 1e6 + 10;
const int mod = 9901;
int n, d, sum[MAXN], b[MAXN], di[MAXN];
struct Node {
	int val;
}a[MAXN];
int lowbit(int x) {
	return x & -x;
}
void add(int x, int k) {
	for (int i = x; i <= n; i += lowbit(i)) {
		sum[i] = (sum[i] + k) % mod;
	}
}
int getsum(int x) {
	int ans = 0;
	for (int i = x; i; i -= lowbit(i)) {
		ans = (ans + sum[i]) % mod;
	}
	return ans;
}
signed main() {
	ios::sync_with_stdio(false);
	cin.tie(0), cout.tie(0);
	while (cin >> n >> d) {
		memset(sum, 0, sizeof sum);
		for (int i = 1; i <= n; i++) {
			cin >> a[i].val;
			b[i] = a[i].val;
		}
		sort(b + 1, b + 1 + n);
		int len = unique(b + 1, b + 1 + n) - b - 1; // 离散化
		for (int i = 1; i <= n; i++) {
			int l = lower_bound(b + 1, b + 1 + len, a[i].val - d) - b; // 求区间范围
			int r = lower_bound(b + 1, b + 1 + len, a[i].val + d) - b;
			int now = lower_bound(b + 1, b + 1 + len, a[i].val) - b;
			if (r > n || b[r] > a[i].val + d) r--;
			di[i] = (getsum(r) - getsum(l - 1) + mod) % mod;
			add(now, di[i] + 1);
		}
		int ans = 0;
		for (int i = 1; i <= n; i++) {
			ans = (ans + di[i]) % mod;
		}
		cout << ans << endl;
	}
	return 0;
}

例题三：HDU 2227 Find the nondecreasing subsequences

题意：

求给定数组中，有多少个不下降子序列。

题解：

思路和上面几乎一样，这题可以不离散处理。

实现细节见代码：

#include <bits/stdc++.h>
using namespace std;
#define int long long
typedef long long ll;
const int MAXN = 1e6 + 10;
const int mod = 1e9 + 7;
int n, c[MAXN];
struct Node {
	int val, id;
}a[MAXN];
int lowbit(int x) {
	return x & -x;
}
void add(int x, int k) {
	for (int i = x; i <= n; i += lowbit(i)) {
		c[i] += k;
		c[i] %= mod;
	}
}
int getsum(int x) {
	int ans = 0;
	for (int i = x; i; i -= lowbit(i)) {
		ans += c[i];
		ans %= mod;
	}
	return ans;
}
bool cmp(Node x, Node y) {
	if (x.val == y.val) {
		return x.id < y.id;
	}
	return x.val < y.val;
}
signed main() {
	ios::sync_with_stdio(false);
	cin.tie(0), cout.tie(0);
	while (cin >> n) {
		memset(c, 0, sizeof c);
		int ans = 0;
		for (int i = 1; i <= n; i++) {
			cin >> a[i].val;
			a[i].id = i;
		}
		sort(a + 1, a + 1 + n, cmp);
		int sum = 0;
		for (int i = 1; i <= n; i++) {
			sum = getsum(a[i].id) + 1; // dp的值
			sum %= mod;
			ans = (ans + sum) % mod;
			add(a[i].id, sum);
		}
		cout << ans << endl;
	}
	return 0;
}