面试题 08.06. 汉诺塔问题

面试题 08.06. 汉诺塔问题

难度简单106

在经典汉诺塔问题中，有 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中盘子的数目不大于14个。

解答

就是定义一个递归函数，先把A移动n-1个到B，然后把A的底盘移到C，最后把B的n-1个移动到C。

class Solution {
public:
    void hanota(vector<int>& A, vector<int>& B, vector<int>& C) {
        int n=A.size();
        dfs(A,B,C,n);//定义辅助函数，A移到C n个,B是空的
    }
    void dfs(vector<int>&A,vector<int>&B,vector<int>& C,int n)
    {
        if(n==1)
        {
            C.push_back(A.back());
            A.pop_back();
        }
        if(n>1)
        {
            dfs(A,C,B,n-1);
            C.push_back(A.back());
            A.pop_back();
            dfs(B,A,C,n-1);
        }
    }
};