美文网首页Android开发程序员Android开发经验谈
面试:Java 实现查找旋转数组的最小数字

面试:Java 实现查找旋转数组的最小数字

作者: nanchen2251 | 来源:发表于2018-07-03 17:57 被阅读136次

    在算法面试中,面试官总是喜欢围绕链表、排序、二叉树、二分查找来做文章,而大多数人都可以跟着专业的书籍来做到倒背如流。而面试官并不希望招收的是一位记忆功底很好,但不会活学活用的程序员。所以学会数学建模和分析问题,并用合理的算法或数据结构来解决问题相当重要。

    面试题:打印出旋转数组的最小数字

    题目:把一个数组最开始的若干个元素搬到数组的末尾,我们称之为数组的旋转。输入一个递增排序的数组的一个旋转,输出旋转数组的最小元素。例如数组 {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,我们的查找范围都会缩小为原来的一半,接下来我们再用更新的下标去重复新一轮的查找。直到最后两个下标相邻,也就是我们的循环结束条件。

    说了一堆,似乎已经绕的云里雾里了,我们不妨就拿题干中的这个输入来模拟验证一下我们的算法。

    1. input:{3,4,5,1,2}
    2. 此时 low = 0,high = 4,mid = 2,对应的值分别是:num[low] = 3,num[high] = 2,num[mid] = 5
    3. 由于 num[mid] > num[low],所以 num[mid] 应该是在左边的递增子数组中。
    4. 更新 low = mid = 2,num[low] = 5,mid = (low+high)/2 = 3,num[mid] = 1;
    5. high - low ≠ 1 ,继续更新
    6. 由于 num[mid] < num[high],所以断定 num[mid] = 1 位于右边的自增子数组中;
    7. 更新 high = mid = 3,由于 high - mid = 1,所以结束循环,得到最小值 num[high] = 1;

    我们再来看看 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. low = 0, high = 4, mid = 2, nums[low] = 1, nums[high] = 1,nums[mid] = 1;
    2. 由于 nums[mid] >= nums[low],故认定 nums[mid] = 1 在左边递增子数组中;
    3. 所以更新 high = mid = 2,mid = (low+high)/2 = 1;
    4. nums[low] = 1,nums[mid] = 1,nums[high] = 1;
    5. high - low ≠ 1,继续循环;
    6. 由于 nums[mid] >= nums[low],故认定 nums[mid] = 1 在左边递增子数组中;
    7. 所以更新 high = mid = 1,由于 high - low = 1,故退出循环,得到 result = 1;

    但我们一眼了然,明显我们的最小值不是 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 原生态基本都封装了常用算法,对面试这些无作用的算法表示抗议,其实这是相当愚蠢的。我们不求死记硬背算法的实现,但求学习到其中巧妙的思想。只有不断地提升自己的思维能力,才能助自己收获更好的职业发展。

    这也大概是大家一直到处叫大佬,埋怨自己工资总是跟不上别人的一方面原因吧。

    相关文章

      网友评论

      • 假装程序猿:针对array8的运行分析中:
        由于 nums[mid] >= nums[low],故认定 nums[mid] = 1 在左边递增子数组中;
        所以更新 high = mid = 2,mid = (low+high)/2 = 1;
        mid对应在左边递增子数组,应该更新low=mid=2吧 :flushed:

      本文标题:面试:Java 实现查找旋转数组的最小数字

      本文链接:https://www.haomeiwen.com/subject/yfujuftx.html