LeetCode的开端:1. 两数之和

经典的OJ题,放假了,刷这题作为开始。

题目:

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

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

示例:

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

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

我的解答:

第一个想到的自然是暴力解法:

class Solution {
    public int[] twoSum(int[] nums, int target) {
    int[] results = {0, 0};
    for (int i = 0; i < nums.length - 1; i++) {
    for (int j = i + 1; j < nums.length; j++) {
        int sum = nums[i] + nums[j];
        if (sum == target) {
            results[0] = i;
            results[1] = j;
        }
    }
    }
        return results;
    }
}

然而看到网上的解答后,才知道是可以用哈希表优化速度,于是又写了一个:

class Solution {
    public int[] twoSum(int[] nums, int target) {
        Map<Integer, Integer> map = new HashMap<>();
        int[] result = {0, 0};
        for (int i = 0; i < nums.length; i++) {
            map.put(nums[i], i);
        }
        for (int i = 0; i < nums.length; i++) {
            int v = target - nums[i];
            if (map.containsKey(v) && i != map.get(v)) {
                result[0] = map.get(v);
                result[1] = i;
            }
        }
        return result;
    }
}



猜你喜欢

转载自blog.csdn.net/SoulOH/article/details/81017361