在经典汉诺塔问题中,有 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中盘子的数目不大于14个。
题目链接:https://leetcode-cn.com/problems/hanota-lcci/
思路
分治思想。时间复杂度是O(n^2-1)。
f(n) = f(n-1) + f(1) + f(n-1)
class Solution {
public:
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;
}
move(n-1, A, C, B);
C.push_back(A.back());
A.pop_back();
move(n-1, B, A, C);
}
void hanota(vector<int>& A, vector<int>& B, vector<int>& C) {
int len = A.size();
if(len==0) return;
move(len, A, B, C);
}
};
如果把操作过程打印出来,则在移动之处加入打印输出。
class Solution {
public:
void print(vector<int>& A, vector<int>& B, vector<int>& C,char a,char b,char c){
cout<<"-----"<<endl;
cout<<a<<": ";
for(int i=0; i<A.size(); ++i){
cout<<A[i]<<" ";
}
cout<<endl<<b<<": ";
for(int i=0; i<B.size(); ++i){
cout<<B[i]<<" ";
}
cout<<endl<<c<<": ";
for(int i=0; i<C.size(); ++i){
cout<<C[i]<<" ";
}
cout<<endl;
}
void move(int n, vector<int>& A, vector<int>& B, vector<int>& C, char a, char b, char c){
if(n==1){
C.push_back(A.back());
A.pop_back();
print(A,B,C,a,b,c);
cout<<"move "<<a <<"->" <<c<<endl;
return;
}
move(n-1, A, C, B, a,c,b);
C.push_back(A.back());
A.pop_back();
print(A,B,C,a,b,c);
cout<<"move "<<a <<"->" <<c <<endl;
move(n-1, B, A, C, b,a,c);
}
void hanota(vector<int>& A, vector<int>& B, vector<int>& C) {
int len = A.size();
if(len==0) return;
move(len, A, B, C,'a','b','c');
}
};