缺失数字

题目描述

从0,1,2,...,n这n+1个数中选择n个数，组成有序数组，找出这n个数中缺失的那个数，要求O(n)尽可能小。

示例1

输入

[0,1,2,3,4,5,7]

返回值

6

class Solution {
        public int missingNumber(int[] nums) {
//            return missingNumberI(nums);
//            return missingNumberII(nums);
            //方法三：二分法
            //比较nums[mid]和mid大小，如果nums[mid]>mid，说明缺失的数字在mid左边，否则在右边
            //终止条件是left==right，如果数字遍历结束还是没有找到，说明缺失的元素是n，返回left
            //时间复杂度O(logN)，空间复杂度O(1)
            if (nums == null || nums.length == 0) {
                return 0;
            }
            int left = 0, right = nums.length;
            while (left < right) {
                int mid = left + (right - left) / 2;
                if (nums[mid] > mid) {
                    right = mid;
                } else {
                    left = mid + 1;
                }
            }
            return left;
        }

        //方法二：求和法
        //将数组中所有元素求和得出sum
        //根据求和公式sum(n)=n*(n+1)/2，sum(n)-sum结果就是缺少的元素
        //时间复杂度O(n)，空间复杂度O(1)
        private int missingNumberII(int[] nums) {
            if (nums == null || nums.length == 0) {
                return 0;
            }
            int n = nums.length;
            int sum = 0;
            for (int i = 0; i < n; i++) {
                sum += nums[i];
            }
            return n * (n + 1) / 2 - sum;
        }

        //方法一：依次遍历
        //遍历数组，判断nums[i]==i是否成立，不成立直接返回i
        //如果数组遍历结束还没找到，说明0~n-1缺少的是n-1
        //时间复杂度O(n)，空间复杂度O(1)
        private int missingNumberI(int[] nums) {
            if (nums == null || nums.length == 0) {
                return 0;
            }
            for (int i = 0; i < nums.length; i++) {
                if (nums[i] != i) {
                    return i;
                }
            }
            return nums.length;
        }
    }