递归解决汉诺塔问题
题目链接:汉诺塔问题
在经典汉诺塔问题中,有 3 根柱子及 N 个不同大小的穿孔圆盘,盘子可以滑入任意一根柱子。一开始,所有盘子自上而下按升序依次套在第一根柱子上(即每一个盘子只能放在更大的盘子上面)。移动圆盘时受到以下限制:
(1) 每次只能移动一个盘子;
(2) 盘子只能从柱子顶端滑出移到下一根柱子;
(3) 盘子只能叠在比它大的盘子上。请编写程序,用栈将所有盘子从第一根柱子移到最后一根柱子。
你需要原地修改栈。
示例1:
输入:A = [2, 1, 0], B = [], C = []
输出:C = [2, 1, 0]
示例2:
输入:A = [1, 0], B = [], C = []
输出:C = [1, 0]
1.分析题目
如图所示A、B、C为三根柱子,N表示盘子的数量,我们的目标是将A和B中所有的盘子都转移到C上,一次只能转移一个盘子,盘子只能叠在比它大的盘子上面。
当N=1时,A上面只有一个盘子,直接将A上的盘子转移到C上即可。
当N=2时,A上面有两个盘子,第一步:先将柱子A的最顶上盘子移动到中间柱子B上;第二步:将柱子A上最后一个盘子移动到柱子C上;最后一步:将中间柱子B的盘子移动到柱子C。
当N=3时,A上面有三个盘子,这里整体的思想很重要。第一步:先将柱子A的顶上的两个盘子视为一个整体,经过N=2所示的方法移动到中间柱子B上;第二步:将柱子A上最后一个盘子移动到柱子C上;最后一步:将中间柱子B上的两个盘子移动到柱子C。我们无需在意这个整体挪动到C的中间过程是什么,因为在N=2时我们已经实现了如何移动,只需要调用即可,我们把N=2时的问题叫做N=3的子问题。
同理,当N=4时,将柱子A上三个盘子视为一个整体,这个整体中间经过一系列的过程最终挪动到B,然后将柱子A最后一个盘子挪动到柱子C上,最后将中间柱子B上的整体经过一系列的过程挪动到C。同样的,这个中间过程我们不关心,因为N=3时已经帮我们处理好了。
N=5, … , n时,依此类推。
力扣上有很不错的总结:当你在思考这个问题的时候,就将最初的 n 个盘子从 A 移到 C 的问题,转化成了将 n - 1 个盘子从 A 移到 C 的问题, 依次类推,直至转化成 1 个盘子的问题时,问题也就解决了。这就是分治的思想。
而实现分治思想的常用方法就是递归。不难发现,如果原问题可以分解成若干个与原问题结构相同但规模较小的子问题时,往往可以用递归的方法解决。具体解决办法如下:
- n = 1 时,直接把盘子从 A 移到 C;
- n > 1 时,
- 先把上面 n - 1 个盘子从 A 移到 B(子问题,递归);
- 再将最大的盘子从 A 移到 C;
- 再将 B 上 n - 1 个盘子从 B 移到 C(子问题,递归)。
作者:腐烂的橘子
链接:https://leetcode.cn/problems/hanota-lcci/solutions/
来源:力扣(LeetCode)
2.代码实现
定义函数:
// n: 要移动n个碟子 A: 柱子A B: 柱子B C: 柱子C
// 要注意A, B, C的顺序:函数的是从A借助B移动到C
void move(int n, vector<int>& A, vector<int>& B, vector<int>& C);
移动的过程:
// STEP1: 先把上面 n - 1 个盘子从 A 移到 B(期间借助C)
move(n-1, A, C, B);
// STEP2: 再将最大的盘子从 A 移到 C
C.push_back(A.back());
A.pop_back();//A此时为空
// STEP3: 最后将 B 上 n - 1 个盘子从 B 移到 C(期间借助A)
move(n-1, B, A, C);
当N=1时也是递归的结束条件
if(n == 1)
{
C.push_back(A.back());
A.pop_back();
}
最终代码实现如下:
class Solution {
public:
void hanota(vector<int>& A, vector<int>& B, vector<int>& C) {
int n = A.size();
move(n, A, B, C);
}
void move(int n, vector<int>& A, vector<int>& B, vector<int>& C)
{
if(n == 1)
{
C.push_back(A.back());
A.pop_back();
return;
}
// STEP1
move(n-1, A, C, B);
// STEP2
C.push_back(A.back());
A.pop_back();
// STEP3
move(n-1, B, A, C);
}
};
3.总结
时间复杂度:O(N^2)
N | 移动次数f(N) |
---|---|
1 | f(1) = 1 |
2 | f(2) = 2*f(1) + 1 = 2 + 1 = 3 |
3 | f(3) = 2*f(2) + 1 = 6 + 1 = 7 |
4 | f(4) = 2*f(3) + 1 = 14 + 1 = 15 |
… | … |
N | f(N) = 2*f(N-1) + 1 |
空间复杂度:O(1)