hello-algo/en/docs/chapter_divide_and_conquer/hanota_problem.md
2024-05-06 14:40:36 +08:00

11 KiB
Raw Blame History

comments
true

12.4   Tower of Hanoi Problem

In both merge sorting and building binary trees, we decompose the original problem into two subproblems, each half the size of the original problem. However, for the Tower of Hanoi, we adopt a different decomposition strategy.

!!! question

Given three pillars, denoted as `A`, `B`, and `C`. Initially, pillar `A` is stacked with $n$ discs, arranged in order from top to bottom from smallest to largest. Our task is to move these $n$ discs to pillar `C`, maintaining their original order (as shown in Figure 12-10). The following rules must be followed during the disc movement process:

1. A disc can only be picked up from the top of a pillar and placed on top of another pillar.
2. Only one disc can be moved at a time.
3. A smaller disc must always be on top of a larger disc.

Example of the Tower of Hanoi{ class="animation-figure" }

Figure 12-10   Example of the Tower of Hanoi

We denote the Tower of Hanoi of size i as $f(i)$. For example, f(3) represents the Tower of Hanoi of moving 3 discs from A to C.

1.   Consider the base case

As shown in Figure 12-11, for the problem f(1), i.e., when there is only one disc, we can directly move it from A to C.

=== "<1>" Solution for a problem of size 1{ class="animation-figure" }

=== "<2>" hanota_f1_step2{ class="animation-figure" }

Figure 12-11   Solution for a problem of size 1

As shown in Figure 12-12, for the problem f(2), i.e., when there are two discs, since the smaller disc must always be above the larger disc, B is needed to assist in the movement.

  1. First, move the smaller disc from A to B.
  2. Then move the larger disc from A to C.
  3. Finally, move the smaller disc from B to C.

=== "<1>" Solution for a problem of size 2{ class="animation-figure" }

=== "<2>" hanota_f2_step2{ class="animation-figure" }

=== "<3>" hanota_f2_step3{ class="animation-figure" }

=== "<4>" hanota_f2_step4{ class="animation-figure" }

Figure 12-12   Solution for a problem of size 2

The process of solving the problem f(2) can be summarized as: moving two discs from A to C with the help of B. Here, C is called the target pillar, and B is called the buffer pillar.

2.   Decomposition of subproblems

For the problem f(3), i.e., when there are three discs, the situation becomes slightly more complicated.

Since we already know the solutions to f(1) and f(2), we can think from a divide-and-conquer perspective and consider the two top discs on A as a unit, performing the steps shown in Figure 12-13. This way, the three discs are successfully moved from A to C.

  1. Let B be the target pillar and C the buffer pillar, and move the two discs from A to B.
  2. Move the remaining disc from A directly to C.
  3. Let C be the target pillar and A the buffer pillar, and move the two discs from B to C.

=== "<1>" Solution for a problem of size 3{ class="animation-figure" }

=== "<2>" hanota_f3_step2{ class="animation-figure" }

=== "<3>" hanota_f3_step3{ class="animation-figure" }

=== "<4>" hanota_f3_step4{ class="animation-figure" }

Figure 12-13   Solution for a problem of size 3

Essentially, we divide the problem f(3) into two subproblems f(2) and one subproblem $f(1)$. By solving these three subproblems in order, the original problem is resolved. This indicates that the subproblems are independent, and their solutions can be merged.

From this, we can summarize the divide-and-conquer strategy for solving the Tower of Hanoi shown in Figure 12-14: divide the original problem f(n) into two subproblems f(n-1) and one subproblem f(1), and solve these three subproblems in the following order.

  1. Move n-1 discs with the help of C from A to B.
  2. Move the remaining one disc directly from A to C.
  3. Move n-1 discs with the help of A from B to C.

For these two subproblems f(n-1), they can be recursively divided in the same manner until the smallest subproblem f(1) is reached. The solution to f(1) is already known and requires only one move.

Divide and conquer strategy for solving the Tower of Hanoi{ class="animation-figure" }

Figure 12-14   Divide and conquer strategy for solving the Tower of Hanoi

3.   Code implementation

In the code, we declare a recursive function dfs(i, src, buf, tar) whose role is to move the i discs on top of pillar src with the help of buffer pillar buf to the target pillar tar:

=== "Python"

```python title="hanota.py"
def move(src: list[int], tar: list[int]):
    """Move a disc"""
    # Take out a disc from the top of src
    pan = src.pop()
    # Place the disc on top of tar
    tar.append(pan)

def dfs(i: int, src: list[int], buf: list[int], tar: list[int]):
    """Solve the Tower of Hanoi problem f(i)"""
    # If only one disc remains on src, move it to tar
    if i == 1:
        move(src, tar)
        return
    # Subproblem f(i-1): move the top i-1 discs from src with the help of tar to buf
    dfs(i - 1, src, tar, buf)
    # Subproblem f(1): move the remaining one disc from src to tar
    move(src, tar)
    # Subproblem f(i-1): move the top i-1 discs from buf with the help of src to tar
    dfs(i - 1, buf, src, tar)

def solve_hanota(A: list[int], B: list[int], C: list[int]):
    """Solve the Tower of Hanoi problem"""
    n = len(A)
    # Move the top n discs from A with the help of B to C
    dfs(n, A, B, C)
```

=== "C++"

```cpp title="hanota.cpp"
/* Move a disc */
void move(vector<int> &src, vector<int> &tar) {
    // Take out a disc from the top of src
    int pan = src.back();
    src.pop_back();
    // Place the disc on top of tar
    tar.push_back(pan);
}

/* Solve the Tower of Hanoi problem f(i) */
void dfs(int i, vector<int> &src, vector<int> &buf, vector<int> &tar) {
    // If only one disc remains on src, move it to tar
    if (i == 1) {
        move(src, tar);
        return;
    }
    // Subproblem f(i-1): move the top i-1 discs from src with the help of tar to buf
    dfs(i - 1, src, tar, buf);
    // Subproblem f(1): move the remaining one disc from src to tar
    move(src, tar);
    // Subproblem f(i-1): move the top i-1 discs from buf with the help of src to tar
    dfs(i - 1, buf, src, tar);
}

/* Solve the Tower of Hanoi problem */
void solveHanota(vector<int> &A, vector<int> &B, vector<int> &C) {
    int n = A.size();
    // Move the top n discs from A with the help of B to C
    dfs(n, A, B, C);
}
```

=== "Java"

```java title="hanota.java"
/* Move a disc */
void move(List<Integer> src, List<Integer> tar) {
    // Take out a disc from the top of src
    Integer pan = src.remove(src.size() - 1);
    // Place the disc on top of tar
    tar.add(pan);
}

/* Solve the Tower of Hanoi problem f(i) */
void dfs(int i, List<Integer> src, List<Integer> buf, List<Integer> tar) {
    // If only one disc remains on src, move it to tar
    if (i == 1) {
        move(src, tar);
        return;
    }
    // Subproblem f(i-1): move the top i-1 discs from src with the help of tar to buf
    dfs(i - 1, src, tar, buf);
    // Subproblem f(1): move the remaining one disc from src to tar
    move(src, tar);
    // Subproblem f(i-1): move the top i-1 discs from buf with the help of src to tar
    dfs(i - 1, buf, src, tar);
}

/* Solve the Tower of Hanoi problem */
void solveHanota(List<Integer> A, List<Integer> B, List<Integer> C) {
    int n = A.size();
    // Move the top n discs from A with the help of B to C
    dfs(n, A, B, C);
}
```

=== "C#"

```csharp title="hanota.cs"
[class]{hanota}-[func]{Move}

[class]{hanota}-[func]{DFS}

[class]{hanota}-[func]{SolveHanota}
```

=== "Go"

```go title="hanota.go"
[class]{}-[func]{move}

[class]{}-[func]{dfsHanota}

[class]{}-[func]{solveHanota}
```

=== "Swift"

```swift title="hanota.swift"
[class]{}-[func]{move}

[class]{}-[func]{dfs}

[class]{}-[func]{solveHanota}
```

=== "JS"

```javascript title="hanota.js"
[class]{}-[func]{move}

[class]{}-[func]{dfs}

[class]{}-[func]{solveHanota}
```

=== "TS"

```typescript title="hanota.ts"
[class]{}-[func]{move}

[class]{}-[func]{dfs}

[class]{}-[func]{solveHanota}
```

=== "Dart"

```dart title="hanota.dart"
[class]{}-[func]{move}

[class]{}-[func]{dfs}

[class]{}-[func]{solveHanota}
```

=== "Rust"

```rust title="hanota.rs"
[class]{}-[func]{move_pan}

[class]{}-[func]{dfs}

[class]{}-[func]{solve_hanota}
```

=== "C"

```c title="hanota.c"
[class]{}-[func]{move}

[class]{}-[func]{dfs}

[class]{}-[func]{solveHanota}
```

=== "Kotlin"

```kotlin title="hanota.kt"
[class]{}-[func]{move}

[class]{}-[func]{dfs}

[class]{}-[func]{solveHanota}
```

=== "Ruby"

```ruby title="hanota.rb"
[class]{}-[func]{move}

[class]{}-[func]{dfs}

[class]{}-[func]{solve_hanota}
```

=== "Zig"

```zig title="hanota.zig"
[class]{}-[func]{move}

[class]{}-[func]{dfs}

[class]{}-[func]{solveHanota}
```

As shown in Figure 12-15, the Tower of Hanoi forms a recursive tree with a height of n, each node representing a subproblem, corresponding to an open dfs() function, thus the time complexity is O(2^n), and the space complexity is $O(n)$.

Recursive tree of the Tower of Hanoi{ class="animation-figure" }

Figure 12-15   Recursive tree of the Tower of Hanoi

!!! quote

The Tower of Hanoi originates from an ancient legend. In a temple in ancient India, monks had three tall diamond pillars and $64$ differently sized golden discs. The monks continuously moved the discs, believing that when the last disc is correctly placed, the world would end.

However, even if the monks moved a disc every second, it would take about $2^{64} \approx 1.84×10^{19}$ seconds, approximately 585 billion years, far exceeding current estimates of the age of the universe. Thus, if the legend is true, we probably do not need to worry about the world ending.