一、题目描述
- 有 A,B,C 三根柱子,A 上面有 n 个盘子,想把 A 上面的盘子移动到 C 上,但是要满足以下三个条件:
-
-
-
- 请编写程序,用栈将所有盘子从第一根柱子移到最后一根柱子。
- 需要原地修改栈。
输入:A = [2, 1, 0], B = [], C = []
输出:C = [2, 1, 0]
输入:A = [1, 0], B = [], C = []
输出:C = [1, 0]
二、解题思路:递归与分治
- 这是一道递归方法的经典题目,乍一想还挺难理清头绪的,我们不妨先从简单的入手:
-
- 假设 n = 1,只有一个盘子,很简单,直接把它从 A 中拿出来,移到 C 上;
-
- 如果 n = 2 呢?这时候就要借助 B 了,因为小盘子必须时刻都在大盘子上面,共需要 4 步。
- 如果 n > 2 呢?思路和上面是一样的,把 n 个盘子也看成两个部分,一部分有 1 个盘子,另一部分有 n - 1 个盘子:
- 观察上图,可能会产生疑问:那么 n - 1 个盘子是怎么从 A 移到 C 的呢?
- 注意,当你在思考这个问题的时候,就将最初的 n 个盘子从 A 移到 C 的问题,转化成了将 n - 1 个盘子从 A 移到 C 的问题, 依次类推,直至转化成 1 个盘子的问题时,问题也就解决,这就是分治的思想。
- 而实现分治思想的常用方法就是递归。不难发现,如果原问题可以分解成若干个与原问题结构相同但规模较小的子问题时,往往可以用递归的方法解决。具体解决办法如下:
-
-
- n > 1 时,先把上面 n - 1 个盘子从 A 移到 B(子问题,递归);再将最大的盘子从 A 移到 C;再将 B 上 n - 1 个盘子从 B 移到 C(子问题,递归)。
三、算法示例
class Solution:
def hanota(self, A: List[int], B: List[int], C: List[int]) -> None:
n = len(A)
self.move(n, A, B, C)
def move(self,n, A, B, C):
if n == 1:
C.append(A[-1])
A.pop()
return
else:
self.move(n-1, A, C, B)
C.append(A[-1])
A.pop()
self.move(n-1,B, A, C)
class Solution {
public void hanota(List<Integer> A, List<Integer> B, List<Integer> C) {
int N = A.size();
move(N, A, B, C);
}
void move(int N, List<Integer> A, List<Integer> B, List<Integer> C) {
if (N == 1) {
C.add(0, A.remove(0));
return;
}
move(N - 1, A, C, B);
C.add(0, A.remove(0));
move(N - 1, B, A, C);
}
}
四、复杂度分析
- 时间复杂度:O(2n ?1)。一共需要移动的次数。
- 空间复杂度:O(1)。
|