【每日LeetCode】1.两数之和

题目

给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那 两个 整数，并返回他们的数组下标。

你可以假设每种输入只会对应一个答案。但是，数组中同一个元素不能使用两遍。

示例:

给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

答案

解题思路

最先想到的就是暴力解法，不断的取出当前下标后面的值，与当前值进行相加，即双重循环。
时间复杂度：O(n²)

代码

public class twoSum {

    public static void main(String[] args) {
        int nums[] = {2, 7, 11, 15};
        int[] solution = Solution1(nums, 9);
        System.out.println(Arrays.toString(solution));
    }

    /**
     * @description: 暴力解法，双重循环依次判断数组中每两个值的和，时间复杂度：O(n方)
     * @Param: [nums, target]
     * @Return: int[]
     * @Author: WHB
     * @Date: 2020/7/4 21:58
     */
    public static int[] Solution1(int[] nums, int target) {
        //  下标
        int result[] = new int[2];
        for (int i = 0; i < nums.length; i++) {
            for (int j = i + 1; j < nums.length; j++) {
                if (nums[i] + nums[j] == target) {
                    result[0] = i;
                    result[1] = j;
                    return result;
                }
            }
        }
        return null;
    }
}

优化写法

对于 x+y=z 在已知z的情况下，可以通过x的值推算出y的值，既然必有正确答案，则z-x的值必定存在与数组中，所以可以使用一个map来存储数组中数值和坐标，通过差值找到相应的数据

代码

public class twoSum {

    public static void main(String[] args) {
        int nums[] = {2, 7, 11, 15};
        int[] solution = Solution2(nums, 9);
        System.out.println(Arrays.toString(solution));
    }

    public static int[] Solution2(int[] nums, int target) {
        int result[] = new int[2];
        //  key存储差值，value存储下标
        HashMap<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            //  map中存在对应的差值，即当前数据为所求
            //  例：nums[1,4,6,2,7] target= 7
            //  map{6,0;    3,1;    1,2;    5,3;    0,4}
            //  执行到6的位置时，发现Map中存在，返回value，即6对应的数组下标
            if (map.containsKey(nums[i])) {
                result[0] = map.get(nums[i]);
                result[1] = i;
                return result;
            }
            map.put(target - nums[i], i);
        }
        return null;
    }
}