题目描述:
在经典汉诺塔问题中,有 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]
解题思路:
使用递归来实现
设初始时A上的柱子个数为N,此时若想全部移到C中,可将其划分为子问题:
1. 首先将A中的前N-1个圆盘移动到柱子B上
2. 将A中的第N个圆盘移动到柱子C上,并在A中删除第N个圆盘
3. 将柱子B上的圆盘全部移动到柱子C上
问题的步骤已经解决,下一步需要找到终止条件。
终止条件:当N=1时,直接将A中的一个圆盘移动到柱子C上,并在A中删除第N个圆盘
代码如下:
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==0)
{
return;
}
else if(n==1)
{
C.push_back(A.back());
A.pop_back();
}
else{
move(n-1, A, C, B);
C.push_back(A.back());
A.pop_back();
move(n-1, B, A, C);
}
}
};