面試題:青蛙跳一跳,斐波那切數列

面試題:一直青蛙一次可以跳上 1 級臺階,也可以跳上 2 級,求該青蛙跳上 n 級的臺階總共有多少種跳法。

  一看這道題,好像沒啥思路,感覺和我們的數據結構和常用的算法好像一點都不沾邊。

  但這看起來就像一道數學題,而且似乎就是高考數學的倒數第一題,所以我們就用數學來做吧。

  數學中有個方法叫「數學歸納法」,我們這裏就可以巧妙用到。

  1. 當 n = 1 時,青蛙有 1 種跳法;
  2. 當 n = 2 時,青蛙可以選擇一次跳 1 級,跳兩次;也可以選擇一次跳 2 級;青蛙有 2 種跳法;
  3. 當 n = 3 時,青蛙可以選擇 1-1-1,1-2,2-1,青蛙有 3 種跳法;
  4. 當 n = 4 時,青蛙可以選擇 1-1-1-1,1-1-2,1-2-1,2-1-1,2-2,青蛙有 5 種跳法;似乎能得到 f(3) = f(2) + f(1),f(4) = f(3) + f(2),這是 f(n) = f(n-1) + f(n-2) 的節奏?我們得用 n = 5 驗證一下。
  5. 當 n =5 時,青蛙可以選擇1-1-1-1-1,1-1-1-2,1-1-2-1,1-2-1-1,2-1-1-1,1-2-2,2-1-2,2-2-1,青蛙有 8 種跳法,f(5) = f(4) + f(3) 成立。
  6. 這是最笨的方法了,得出了這確實就是一個典型的斐波那契數列,唯一不一樣的地方就是 n =2 的時候並沒有 f(2) = f(0) + f(1)。

稍微有點思維能力的可能更簡單。

  1. n = 1 ,青蛙有 1 種跳法;
  2. n = 2 ,青蛙有 2 種跳法;
  3. n = 3,青蛙在第 1 級可以跳 1 種,後面 2 級相當於 f(3-1) = f(2),還有一種就是先跳 2 級,然後後面 1 級有 f(3-2) = f(1) 種跳法,可以得出 f(3) = f(2) + f(1);
  4. 當取 n 時,青蛙在第一次跳 1 級,後面的相當於有 f(n-1) 種跳法;假設第一次跳 2 級,後面相當於有 f(n-2) 種跳法;故可以得出 f(n) = f(n-1) + f(n-2);

      這樣思考可能更不容易出錯吧,這就是思維的提煉過程,可見我們高考常考的「數學歸納法」是多麼地有用。

      既然能分析出這是一道典型的斐波那契數列了,我想教科書都教給大家方法了,不過一定要注意 n = 2 的時候,正常的斐波那契數列值應該是 1,而我們是 2。大多數人肯定會寫出下面的代碼:

public class Test09 {

    private static int fn(int n) {
        if (n <= 0)
            return 0;
        if (n == 1)
            return 1;
        if (n == 2)
            return 2;
        else
            return fn(n - 1) + fn(n - 2);
    }

    public static void main(String[] args) {
        System.out.println(fn(1));
        System.out.println(fn(2));
        System.out.println(fn(3));
        System.out.println(fn(4));
    }
}

  我們教科書上反覆用這個問題來講解遞歸函數,但並不能說明遞歸的解法是最適合這個題目的。當我們暗自竊喜完成了這道面試題的時候,或許面試官會告訴我們,上面的這種遞歸解法存在很嚴重的效率問題,並讓我們分析其中的原因。

  我們以求 fn(10) 爲例,要想求得 fn(10),需要先求得 fn(9) 和 fn(8);同樣,要求得 fn(9),需要先求得 fn(8) 和 fn(7)……

  這存在一個很大的問題,我們一定會去重複計算很多值,我們一定得想辦法把這個計算好的值存放起來。
  
避免重複計算

  既然我們找到了問題所在,那改進方法自然是信手拈來了。我們目前的算法是「從大到小」計算,而我們只需要反向「從小到大」計算就可以了。我們根據 fn(1) 和 fn(2) 計算出 fn(3),再根據 fn(2) 和 fn(3) 計算出 fn(4)……

  很容易理解,這樣的算法思路時間複雜度是 O(n),實現代碼如下:

public class Test09 {

    private static long fn(int n) {
        if (n <= 0)
            return 0;
        if (n == 1)
            return 1;
        if (n == 2)
            return 2;
        long prePre = 1, pre = 2;
        long result = 0;
        for (int i = 3; i <= n; i++) {
            result = prePre + pre;
            prePre = pre;
            pre = result;
        }
        return result;
    }

    public static void main(String[] args) {
        System.out.println(fn(1));
        System.out.println(fn(3));
        System.out.println(fn(50));
        System.out.println(fn(100));
    }
}

  上面的代碼,一定要注意做了一點小修改,我們把返回值悄悄地改成了 long ,因爲我們並不能保證客戶端是否會輸入一個比較大的數字,比如:100,這樣,如果返回值爲 int,一定會因爲超出了最大值而顯示錯誤的,解決方案就是把值換爲更大容量的 long。但有時候你會發現,long 的容量也不夠,畢竟整型和長整型,它都會有最大顯示值,在遇到這樣的情況的時候。我們最好和面試官交流一下,(多問兩句總不會有錯)是否處理這樣的情況。如果一定要處理這樣的情況,那麼可能你就得用 String 來做顯示處理了。
  
  題目來源於《劍指 Offer》,本文轉自微信公衆號nanchen

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章