AcWing 1214. 波动数列（复杂DP，重点理解）

techpupil

已于 2024-01-31 00:57:25 修改

阅读量463

点赞数 7

文章标签：算法动态规划 DP

于 2024-01-30 23:52:39 首次发布

本文链接：https://blog.csdn.net/nuc_ghp/article/details/135943329

版权

文章描述了一道关于求解给定长度n，和为s的整数数列，后一项比前一项增加a或减少b的数列种类的动态规划问题。通过状态表示和模运算技巧解决了这个问题。

摘要由CSDN通过智能技术生成

[题目概述]

观察这个数列：
1 3 0 2 -1 1 -2 …
这个数列中后一项总是比前一项增加2或者减少3，且每一项都为整数。
栋栋对这种数列很好奇，他想知道长度为 n ，和为 s ，而且后一项总是比前一项增加 a 或者减少 b 的整数数列可能有多少种呢？

输入格式

共一行，包含四个整数 n,s,a,b，含义如前面所述。

输出格式

共一行，包含一个整数，表示满足条件的方案数。
由于这个数很大，请输出方案数除以 100000007 的余数。

数据范围

$1 \leq n \leq 1000,$
$10^9 ≤ s ≤ 10^9$ ,
$1 ≤ a, b ≤ 10^6$

输入样例：

4 10 2 3

输出样例：

样例解释
两个满足条件的数列分别是2 4 1 3和7 4 1 -2。

分析题目
题目让我们求长度为n，总和为s，每个数总比前一个数+a或者-b的序列个数，看起来也是一道DP题目，但我们现在好像看不出来怎样进行状态表示，那我们先模拟一下

由于x的变数最大，所以我们用其他未知量来表示x
然后推出来一个规律，s与 $(n - 1) * d 1 + (n - 2) * d 2 + ...... d n - 1$ 余数相同。那么我们只需要枚举整个 $d_i$ 序列即可。

我们的状态表示就有了，那么状态计算又怎样解决？

这个是针对+a的情况，-b就是同理可得了
完整代码
求正余数这个函数需要说明一下
a%b -> -(b-1)~(b-1)
a%b+b -> 1~2b-1
(a%b+b)%b -> 1~b-1
此时的结果一定是个正数，就实现了目的

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
using namespace std;

const int N = 1010, MOD = 100000007;
int n, s, a, b, res;
int f[N][N];

// 求a%b的正余数
int mod(int a, int b) {
	return (a % b + b) % b;
}

int main () {
	cin >> n >> s >> a >> b;

	f[0][0] = 1;
	for (int i = 1; i < n; i ++) {
		for (int j = 0; j < n; j ++) {
			f[i][j] = (f[i - 1][mod(j - a * (n - i), n)] + f[i - 1][mod(j + b * (n - i), n)]) % MOD;
		}
	}
	// 这个为什么不是n而是n-1的问题我想了很长时间，最后也是看到一个人的回答才明白
	// 我们所要求的只有n-1个数，因为第一个数我们并不知道而是推出来的，我们从第二个数开始才进行枚举
	cout << f[n - 1][mod(s, n)] << endl;
	return 0;
}