最新情報:所有的遞歸都可以改寫成非遞歸?

file

前言

本文收錄於專輯://dwz.win/HjK,點擊解鎖更多數據結構與演算法的知識。

你好,我是彤哥,一個每天爬二十六層樓還不忘讀源碼的硬核男人。

上一節,我們使用點陣圖介紹了12306搶票演算法的實現,沒有收到推送的同學可以點擊上方專輯查看,或者在公主號歷史消息中查看。

在上一節的最後,彤哥收到最新情報,說是所有的遞歸都可以改寫成非遞歸,是不是真的呢?如何實現呢?有沒有套路呢?

讓我們帶著這些問題進入今天的學習吧。

何為遞歸?

所謂遞歸,是指程式在運行的過程中調用自身的行為。

這種行為也不能無限制地進行下去,得有個出口,叫做邊界條件,所以,遞歸可以分成三個段:前進段、達到邊界條件,返回段,在這三個段我們都可以做一些事,比如前進段對問題規模進行縮小,返回段對結果進行整理。

這麼說可能比較抽象,讓我們看一個簡單的案例:

如何用遞歸實現1到100的相加?

1到100相加使用循環大家都會解,程式碼如下:

public class Sum {
    public static void main(String[] args) {
        System.out.println(sumCircle(1, 100));
    }

    private static int sumCircle(int min, int max) {
        int sum = 0;
        for (int i = min; i <= max; i++) {
            sum += i;
        }
        return sum;
    }
}

那麼,如何使用遞歸實現呢?

如何快速實現遞歸?

首先,我們要找到這道題的邊界條件,1到100相加,邊界條件可以是1,也可以是100,如果從1開始,那麼邊界條件就是100,反之亦然。

找到了邊界條件之後,就是將問題規模縮小,對於這道題,計算1到100相加,那麼,能不能先計算1到99相加再把100加上呢?肯定是可以的,這樣問題的規模就縮小了,直到,問題規模縮小為1到1相加為止。

OK,讓我們看程式碼實現:

private static int sumRecursive(int min, int max) {
    // 邊界條件
    if (min >= max) {
        return min;
    }
    // 問題規模縮小
    int sum = sumRecursive(min, max - 1);
    // 加上當前值
    sum += max;
    // 返回
    return sum;
}

是不是很簡單?還可以更簡單:

private static int sumRecursive2(int min, int max) {
    return min >= max ? min : sumRecursive2(min, max - 1) + max;
}

686?

所以,使用遞歸最重要的就是找到邊界條件,然後讓問題的規模朝著邊界條件的方向一直縮小,直到達到邊界條件,最後依次返回即可,這也是快速實現遞歸的套路。

這麼看來,使用遞歸似乎很簡單,但是,它有沒有什麼缺點呢?

要了解缺點就得從遞歸的本質入手。

遞歸的本質是什麼?

我們知道,JVM啟動的時候有個參數叫做-Xss,它不是表示XSS攻擊哈,它是指每個執行緒可以使用的執行緒棧的大小。

那麼,什麼又是執行緒棧呢?

棧大家都理解了,我們在前面的章節也學習過了,使用棧,可以實現計算器的功能,非常方便。

執行緒棧,顧名思義,就是指執行緒運行過程中使用的棧。

那麼,執行緒在運行的過程中為什麼要使用棧呢?

這就不得不說方法調用的本質了。

舉個簡單的例子:

private static int a(int num) {
    int a = 1;
    return a + b(num);
}

private static int b(int num) {
    int b = 2;
    return c(num) + b;
}

private static int c(int num) {
    int c = 3;
    return c + num;
}

在這段程式碼中,方法a() 調用 方法b(),方法b() 調用 方法c(),在實際運行的過程中,是這樣處理的:調用方法a()時,發現需要調用方法b()才能返回,那就把方法a()及其狀態保存到棧中,然後調用方法b(),同樣地,調用方法b()時,發現需要先調用方法c()才能返回,那就把方法b()及其狀態入棧,然後調用方法c(),調用方法c()時,不需要額外調用別的方法了,計算完畢返回,返回之後,從棧頂取出方法b()及當時的狀態,繼續運行方法b(),方法b()運行完畢,返回,再從棧中取出方法a()及當時的狀態,計算完畢,方法a()返回,程式等待結束。

還是上圖吧:

file

所以,方法調用的本質,就是棧的使用。

同理,遞歸的調用就是方法的調用,所以,遞歸的調用,也是棧的使用,不過,這個棧會變得非常大,比如,對於1到100相加,就有99次入棧出棧的操作。

file

因此,總結起來,遞歸有以下兩個缺點:

  1. 操作耗時,因為牽涉到大量的入棧出棧操作;
  2. 有可能導致執行緒棧溢出,因為遞歸調用佔用了執行緒棧很大的空間。

那麼,我們是不是就不要使用遞歸了呢?

當然不是,之所以使用遞歸,就是因為它使用起來非常簡單,能夠快速地解決我們的問題,合理控制遞歸調用鏈的長度,就是一個好遞歸。

既然,遞歸調用的本質,就是棧的使用,那麼,我們能不能自己模擬一個棧,將遞歸調用改成非遞歸呢?

當然可以。

修改遞歸為非遞歸的套路

還是使用上面的例子,現在我們需要把遞歸修改成非遞歸,且不是使用for循環的那種形式,要怎麼實現呢?

首先,我們要自己模擬一個棧;

然後,找到邊界條件;

最後,朝著邊界條件的方向縮小問題規模;

OK,上程式碼:

private static int sumNonRecursive(int min, int max) {
        int sum = 0;
        // 聲明一個棧
        Stack<Integer> stack = new Stack<Integer>();
        stack.push(max);
        while (!stack.isEmpty()) {
            if (max > min) {
                // 要計算max,先計算max-1
                stack.push(--max);
            } else {
                // 問題規模縮小到一定程度,計算返回
                sum += stack.pop();
            }
        }
        return sum;
    }

好了,是不是很簡單,其實跟遞歸的套路是一樣的,只不過改成自己模擬棧來實現。

這個例子可能不是那麼明顯,我們再舉個二叉樹遍歷的例子來看一下。

public class BinaryTree {

    Node root;

    // 插入元素
    void put(int value) {
        if (root == null) {
            root = new Node(value);
        } else {
            Node parent = root;
            while (true) {
                if (value <= parent.value) {
                    if (parent.left == null) {
                        parent.left = new Node(value);
                        return;
                    } else {
                        parent = parent.left;
                    }
                } else {
                    if (parent.right == null) {
                        parent.right = new Node(value);
                        return;
                    } else {
                        parent = parent.right;
                    }
                }

            }
        }
    }

    // 先序遍歷
    void preTraversal(Node x) {
        if (x == null) return;
        System.out.print(x.value + ",");
        preTraversal(x.left);
        preTraversal(x.right);
    }

    static class Node {
        int value;
        Node left;
        Node right;

        public Node(int value) {
            this.value = value;
        }
    }

    public static void main(String[] args) {
        BinaryTree binaryTree = new BinaryTree();
        binaryTree.put(3);
        binaryTree.put(1);
        binaryTree.put(2);
        binaryTree.put(7);
        binaryTree.put(8);
        binaryTree.put(5);
        binaryTree.put(4);
        binaryTree.put(6);
        binaryTree.put(9);
        binaryTree.put(0);

        binaryTree.preTraversal(binaryTree.root);
    }
}

我這裡隨手寫了一顆二叉樹,並實現了其先序遍歷,這個測試用例中的二叉樹長這個樣子:

file

所以,這個二叉樹的先序遍歷結果為3,1,0,2,7,5,4,6,8,9,

可以看到,使用遞歸先序遍歷二叉樹非常簡單,而且程式碼清晰易懂,那麼,它如何修改為非遞歸實現呢?

首先,我們要自己模擬一個棧;

然後,找到邊界條件,為節點等於空時;

最後,縮小問題規模,這裡是先把右子樹壓棧,再把左子樹壓棧,因為先左後右;

好了,來看程式碼實現:

// 先序遍歷非遞歸形式
void nonRecursivePreTraversal(Node x) {
    // 自己模擬一個棧
    Stack<Node> stack = new Stack<Node>();
    stack.push(x);
    while (!stack.isEmpty()) {
        Node tmp = stack.pop();
        // 隱含的邊界條件
        if (tmp != null) {
            System.out.print(tmp.value + ",");
            // 縮小問題規模
            stack.push(tmp.right);
            stack.push(tmp.left);
        }
    }
}

掌握了這個套路是不是把遞歸改寫為非遞歸非常簡單,不過,改寫之後的程式碼顯然沒有遞歸那麼清晰。

好了,遞歸改寫為非遞歸的套路我們就講到這裡,不知道你Get到了沒有呢?你也可以找個遞歸自己來改寫試試看。

後記

本節,我們從遞歸的概念入手,學習了如何快速實現遞歸,以及遞歸的本質,最後,學習了遞歸改寫為非遞歸的套路。

本質上,這也是棧這種數據結構的常規用法。

既然講到了棧,不講隊列是不是有點過分?

所以,下一節,遍歷各種源碼的彤哥將介紹如何實現高性能的隊列,想了解其中的套路嗎?還不快點來關注我!

關注公號主「彤哥讀源碼」,解鎖更多源碼、基礎、架構知識。