两数之和

一、要求

给定一个整数数组和一个目标值,找出数组中和为目标值的两个数。

你可以假设每个输入只对应一种答案,且同样的元素不能被重复利用。

示例:

给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

二、思路

这道题挺简单的,暴力的话,只要两层for循环即可,但复杂度是O(N^2)。在使用了HashMap之后,由于HashMap高效的查找效率,可以大大缩减执行时间。有关HashMap的底层原理,可以参考我的另外一篇博客HashMap底层实现原理浅谈


三、代码实现

(1)两层for循环

public class day1106 {
    public int[] twoSum(int[] nums, int target) {
        for (int i = 0; i < nums.length; i++) {
            for (int j = i + 1; j < nums.length; j++) {
                if ((nums[i] + nums[j]) == target) {
                    return new int[]{i, j};
                }
            }
        }
        return null;
    }

(2)使用HashMap

    public int[] twoSum2(int[] nums, int target) {
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            int p = target - nums[i];
            if (map.containsKey(p)) {
                return new int[]{i, map.get(p)};
            }
            map.put(nums[i], i);
        }
        return null;
    }