在算法面試中,面試官總是喜歡圍繞鏈表、排序、二叉樹、二分查找來做文章,而大多數人都可以跟著專業的書籍來做到倒背如流。而面試官并不希望招收的是一位記憶功底很好,但不會活學活用的程序員。所以學會數學建模和分析問題,并用合理的算法或數據結構來解決問題相當重要。
面試題:打印出旋轉數組的最小數字
題目:把一個數組最開始的若干個元素搬到數組的末尾,我們稱之為數組的旋轉。輸入一個遞增排序的數組的一個旋轉,輸出旋轉數組的最小元素。例如數組 {3,4,5,1,2} 為數組 {1,2,3,4,5} 的一個旋轉,該數組的最小值為 1。
要想實現這個需求很簡單,我們只需要遍歷一遍數組,找到最小的值后直接退出循環。代碼實現如下:
public class Test08 { public static int getTheMin(int nums[]) { if (nums == null || nums.length == 0) { throw new RuntimeException("input error!"); } int result = nums[0]; for (int i = 0; i < nums.length - 1; i++) { if (nums[i + 1] < nums[i]) { result = nums[i + 1]; break; } } return result; } public static void main(String[] args) { // 典型輸入,單調升序的數組的一個旋轉 int[] array1 = {3, 4, 5, 1, 2}; System.out.println(getTheMin(array1)); // 有重復數字,并且重復的數字剛好的最小的數字 int[] array2 = {3, 4, 5, 1, 1, 2}; System.out.println(getTheMin(array2)); // 有重復數字,但重復的數字不是第一個數字和最后一個數字 int[] array3 = {3, 4, 5, 1, 2, 2}; System.out.println(getTheMin(array3)); // 有重復的數字,并且重復的數字剛好是第一個數字和最后一個數字 int[] array4 = {1, 0, 1, 1, 1}; System.out.println(getTheMin(array4)); // 單調升序數組,旋轉0個元素,也就是單調升序數組本身 int[] array5 = {1, 2, 3, 4, 5}; System.out.println(getTheMin(array5)); // 數組中只有一個數字 int[] array6 = {2}; System.out.println(getTheMin(array6)); // 數組中數字都相同 int[] array7 = {1, 1, 1, 1, 1, 1, 1}; System.out.println(getTheMin(array7)); }}
打印結果沒什么毛病。不過這樣的方法顯然不是最優的,我們看看有沒有辦法找出更加優質的方法處理。
有序,還要查找?
找到這兩個關鍵字,我們不免會想到我們的二分查找法,但不少小伙伴肯定會問,我們這個數組旋轉后已經不是一個真正的有序數組了,不過倒像是兩個遞增的數組組合而成的,我們可以這樣思考。
我們可以設定兩個下標 low 和 high,并設定 mid = (low + high)/2,我們自然就可以找到數組中間的元素 array[mid],如果中間的元素位于前面的遞增數組,那么它應該大于或者等于 low 下標對應的元素,此時數組中最小的元素應該位于該元素的后面,我們可以把 low 下標指向該中間元素,這樣可以縮小查找的范圍。
同樣,如果中間元素位于后面的遞增子數組,那么它應該小于或者等于 high 下標對應的元素。此時該數組中最小的元素應該位于該中間元素的前面。我們就可以把 high 下標更新到中位數的下標,這樣也可以縮小查找的范圍,移動之后的 high 下標對應的元素仍然在后面的遞增子數組中。
不管是更新 low 還是 high,我們的查找范圍都會縮小為原來的一半,接下來我們再用更新的下標去重復新一輪的查找。直到最后兩個下標相鄰,也就是我們的循環結束條件。
說了一堆,似乎已經繞的云里霧里了,我們不妨就拿題干中的這個輸入來模擬驗證一下我們的算法。
我們再來看看 Java 中如何用代碼實現這個思路:
public class Test08 { public static int getTheMin(int nums[]) { if (nums == null || nums.length == 0) { throw new RuntimeException("input error!"); } // 如果只有一個元素,直接返回 if (nums.length == 1) return nums[0]; int result = nums[0]; int low = 0, high = nums.length - 1; int mid; // 確保 low 下標對應的值在左邊的遞增子數組,high 對應的值在右邊遞增子數組 while (nums[low] >= nums[high]) { // 確保循環結束條件 if (high - low == 1) { return nums[high]; } // 取中間位置 mid = (low + high) / 2; // 代表中間元素在左邊遞增子數組 if (nums[mid] >= nums[low]) { low = mid; } else { high = mid; } } return result; } public static void main(String[] args) { // 典型輸入,單調升序的數組的一個旋轉 int[] array1 = {3, 4, 5, 1, 2}; System.out.println(getTheMin(array1)); // 有重復數字,并且重復的數字剛好的最小的數字 int[] array2 = {3, 4, 5, 1, 1, 2}; System.out.println(getTheMin(array2)); // 有重復數字,但重復的數字不是第一個數字和最后一個數字 int[] array3 = {3, 4, 5, 1, 2, 2}; System.out.println(getTheMin(array3)); // 有重復的數字,并且重復的數字剛好是第一個數字和最后一個數字 int[] array4 = {1, 0, 1, 1, 1}; System.out.println(getTheMin(array4)); // 單調升序數組,旋轉0個元素,也就是單調升序數組本身 int[] array5 = {1, 2, 3, 4, 5}; System.out.println(getTheMin(array5)); // 數組中只有一個數字 int[] array6 = {2}; System.out.println(getTheMin(array6)); // 數組中數字都相同 int[] array7 = {1, 1, 1, 1, 1, 1, 1}; System.out.println(getTheMin(array7)); // 特殊的不知道如何移動 int[] array8 = {1, 0, 1, 1, 1}; System.out.println(getTheMin(array8)); }}
前面我們提到在旋轉數組中,由于是把遞增排序數組的前面的若干個數字搬到數組后面,因為第一個數字總是大于或者等于最后一個數字,而還有一種特殊情況是移動了 0 個元素,即數組本身,也是它自己的旋轉數組。這種情況本身數組就是有序的了,所以我們只需要返回第一個元素就好了,這也是為什么我先給 result 賦值為 nums[0] 的原因。
上述代碼就完美了嗎?我們通過測試用例并沒有達到我們的要求,我們具體看看 array8 這個輸入。先模擬計算機運行分析一下:
但我們一眼了然,明顯我們的最小值不是 1 ,而是 0 ,所以當 array[low]、array[mid]、array[high] 相等的時候,我們的程序并不知道應該如何移動,按照目前的移動方式就默認 array[mid] 在左邊遞增子數組了,這顯然是不負責任的做法。
我們修正一下代碼:
public class Test08 { public static int getTheMin(int nums[]) { if (nums == null || nums.length == 0) { throw new RuntimeException("input error!"); } // 如果只有一個元素,直接返回 if (nums.length == 1) return nums[0]; int result = nums[0]; int low = 0, high = nums.length - 1; int mid = low; // 確保 low 下標對應的值在左邊的遞增子數組,high 對應的值在右邊遞增子數組 while (nums[low] >= nums[high]) { // 確保循環結束條件 if (high - low == 1) { return nums[high]; } // 取中間位置 mid = (low + high) / 2; // 三值相等的特殊情況,則需要從頭到尾查找最小的值 if (nums[mid] == nums[low] && nums[mid] == nums[high]) { return midInorder(nums, low, high); } // 代表中間元素在左邊遞增子數組 if (nums[mid] >= nums[low]) { low = mid; } else { high = mid; } } return result; } /** * 查找數組中的最小值 * * @param nums 數組 * @param start 數組開始位置 * @param end 數組結束位置 * @return 找到的最小的數字 */ public static int midInorder(int[] nums, int start, int end) { int result = nums[start]; for (int i = start + 1; i <= end; i++) { if (result > nums[i]) result = nums[i]; } return result; } public static void main(String[] args) { // 典型輸入,單調升序的數組的一個旋轉 int[] array1 = {3, 4, 5, 1, 2}; System.out.println(getTheMin(array1)); // 有重復數字,并且重復的數字剛好的最小的數字 int[] array2 = {3, 4, 5, 1, 1, 2}; System.out.println(getTheMin(array2)); // 有重復數字,但重復的數字不是第一個數字和最后一個數字 int[] array3 = {3, 4, 5, 1, 2, 2}; System.out.println(getTheMin(array3)); // 有重復的數字,并且重復的數字剛好是第一個數字和最后一個數字 int[] array4 = {1, 0, 1, 1, 1}; System.out.println(getTheMin(array4)); // 單調升序數組,旋轉0個元素,也就是單調升序數組本身 int[] array5 = {1, 2, 3, 4, 5}; System.out.println(getTheMin(array5)); // 數組中只有一個數字 int[] array6 = {2}; System.out.println(getTheMin(array6)); // 數組中數字都相同 int[] array7 = {1, 1, 1, 1, 1, 1, 1}; System.out.println(getTheMin(array7)); // 特殊的不知道如何移動 int[] array8 = {1, 0, 1, 1, 1}; System.out.println(getTheMin(array8)); }}
我們再用完善的測試用例放進去,測試通過。
總結
本題其實考察的點挺多的,實際上就是考察對二分查找的靈活運用,不少小伙伴死記硬背二分查找必須遵從有序,而沒有學會這個二分查找的思想,這樣會導致只能想到循環查找最小值了。
不少小伙伴在面試中表態,Android 原生態基本都封裝了常用算法,對面試這些無作用的算法表示抗議,其實這是相當愚蠢的。我們不求死記硬背算法的實現,但求學習到其中巧妙的思想。只有不斷地提升自己的思維能力,才能助自己收獲更好的職業發展。
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持VeVb武林網。
新聞熱點
疑難解答
圖片精選