算法初探系列12——递推

概述

大家好，我是颜值与才华并存的蒟蒻君~这节课，蒟蒻君将和大家一起学习最简单的动态规划——线性DP初步。

代入

问题1：爬楼梯

共有n级台阶，蒟蒻君站在第0级的位置，每次能上1或2级台阶。请问，蒟蒻君走到第n级台阶有多少种走法？
答案：89

问题2：平面分割

在一个平面上画n条直线，最多能将平面分割成多少部分？
答案：56

一、一维递推

概念

大家通过仔细观察后会发现，以上两道题都可以用找规律的方法解决。
我们定义dp_i为n = i时的答案，则：

• 对于问题1，dp_i = dp_i-1 + dp_i-2
• 对于问题2，dp_i = dp_i-1 + i + 1;

总结一下，这类题目都有一个共同点：
后面问题的解都可以由前面问题的解推出来，我们管这个方法叫递推。

题目1：斐波那契

题目

最开始有一只小兔兔，一个月后小兔兔会变成大兔兔，大兔兔会生小兔兔。

时间	小兔兔数量	大兔兔数量	兔兔总数
第一个月	1	0	1
第二个月	0	1	1
第三个月	1	1	2
第四个月	1	2	3
第五个月	2	3	5

…
现在农场主想知道第n个月有多少只兔兔（兔兔已修炼成仙，不会死）。

分析

相信聪明又可爱的你一定找到规律了，我们来分析一下…

• 我们定义dp_i为第i个月的兔兔总数，a_i为第i个月的小兔兔总数，b_i为第i个月的大兔兔总数，则：dp_i = a_i + b_i
• 第i个月的成年兔兔由第i-1个月的成年兔兔和第i-1个月的小兔兔组成，即：b_i = a_i-1 + b_i-1 = dp_i-1
• 第i个月的小兔兔都是第i-1个月的大兔兔生下来的，即：a_i = b_i-1 = dp_i-2
• dp_i = a_i + b~i = dp_i-2 + dp_i-1

实现

dp数组写法

#include <bits/stdc++.h>
using namespace std;
long long dp[100005];
int main() {
	int n;
	cin >> n; 
	// 前两次要手动算 
	dp[1] = dp[2] = 3;
	for (int i = 3; i <= n; ++i) {
		dp[i] = dp[i - 1] + dp[i - 2];
	}
	cout << dp[n] << '\n';
	return 0;
}

时间复杂度： O(n)。
空间复杂度： O(n)。
还可以优化吗（本次不讲矩阵快速幂）？

滚动数组优化

大家可以发现，dp_i的取值之和dp_i-1和dp_i-2有关，即这个时候dp_i-3到dp₁这些数就莫有用了。
这里我们用三个变量模拟dp_i-2，dp_i-1和dp_i。

#include <bits/stdc++.h>
using namespace std;
int main() {
	int n;
	cin >> n; 
	// 特判 
	if (n < 3) {
		cout << 1 << '\n';
		return 0;
	} 
	// 此时a为dp[1]，b为dp[2]，c为dp[3] 
	long long a = 1, b = 1, c;
	for (int i = 3; i <= n; ++i) {
		c = a + b;
		a = b;
		b = c;
	}
	cout << c << '\n';
	return 0;
}

题目2：错排

题目

蒟蒻君拿到了n封信，要送给n个人，每人1封。但是，粗心的蒟蒻君把每封信都给错人了，请问蒟蒻君有多少种送信的方法。

分析

• 设dp_i为n = i时的答案。
• 在放第n封信时，不能放在正确的位置（1种方法），也就是说还剩下n - 1个位置，即有dp_i-1 * (n - 1)种方法。
• 这时候貌似可能应该是正确答案了吧。。NO！！
• 前n - 1封信都已经错排了，所以dp_n-1中所有信的位置都是错误的。
• 但是！！如果有一封信放对了，第n封信和这封信换个地方，也是一种方案呀！这时候就相当于不管这两封信，其他的n - 2封信错排啦~
• 因此，dp_i = (dp_i-1 + dp_i-2) * (n - 1)

实现

dp数组写法

#include <bits/stdc++.h>
using namespace std;
long long dp[100005]; 
int main() {
	int n;
	cin >> n;
	// 前两封的要手动算
	dp[1] = 0;
	dp[2] = 1;
	for (int i = 3; i <= n; ++i) {
		dp[i] = (dp[i - 1] + dp[i - 2]) * (n - 1);
	} 
	cout << dp[n] << '\n';
	return 0;
}

滚动数组优化

原理和上一个问题一样。

#include <bits/stdc++.h>
using namespace std;
long long dp[100005]; 
int main() {
	int n;
	cin >> n;
	// 前两封的要手动算
	int a = 0, b = 1, c;
	if (n == 1) {
		cout << 0 << '\n';
		return 0;
	}
	if (n == 2) {
		cout << 1 << '\n';
		return 0;
	}
	for (int i = 3; i <= n; ++i) {
		c = (a + b) * (i - 1);
		a = b;
		b = c;
	} 
	cout << c << '\n';
	return 0;
}

二、二维递推

相信聪明绝顶的你已经很熟悉一维递推了，接下来我们要加大难度了哦~

题目1：杨辉三角

题目

杨辉三角是二项式系数在三角形中的排列，每个数 = 左上方的数 + 右上方的数。

求杨辉三角第n行第m个数。
我们先将杨辉三角变成二维数组的格式。

分析

• 定义dp_ij为数组的第i行第j个数。
• 观察图片，相信大家可以想到递推式：dp_ij = dp_i-1j + dp_i-1j-1

实现

#include <bits/stdc++.h>
using namespace std;
long long dp[10005][10005];
int main() {
	int n;
	cin >> n;
	dp[0][0] = 1;
	for (int i = 1; i <= n; ++i)  {
		for (int j = 1; j <= i; ++j) {
			dp[i][j] = dp[i - 1][j] + dp[i - 1][j - 1];
			cout << dp[i][j] << ' ';
		}
		cout << '\n';
	}
	return 0;
}

题目2：[NOIP2002]普及组过河卒

传送

这节课我们学习了递推 （为下节课DP做铺垫），下节课我们来学习DP。