leetcode-1

题目

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

你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。

示例

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

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

思路

• 拿到题目后，首先想到的是双层循环，时间复杂度O(n^2)。

public int[] twoSum(int[] nums, int target) {
    int[] result = new int[2];
    for (int i = 0; i < nums.length - 1; i++) {
        for (int j = i + 1; j < nums.length; j++) {
            if ((nums[i] + nums[j]) == target) {
                result[0] = i;
                result[1] = j;
                break;
            }
        }
        if (result[1] != 0) {
            break;
        }
    }
    return result;
}

• 此方法还是会存在效率的问题，提供一个利用map存储的思路。key用于存储数据，value用于存储脚标，此时注意，在对Key赋值时，不要覆盖value。

public static int[] twoSum(int[] nums, int target) {
        int[] result = new int[2];
        Map<Integer, Integer> memory = new HashMap<Integer, Integer>();

        for (int i = 0; i < nums.length; i ++) {
            if (memory.containsKey(target - nums[i])) {
                result[0] = memory.get(target - nums[i]);
                result[1] = i;
                break;
            } else {
                if (!memory.containsKey(nums[i])) memory.put(nums[i], i);
            }
        }
        return result;
    }