两数之和(two sum)

LeetCode第一题
题目描述:给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。

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

  1. 暴力破解:遍历数组,时间复杂度O( n 2 n^2 ),空间复杂度O(1)。
public static 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[j]==target-nums[i]){
                    return new int[]{i,j};
                }
            }
        }
        return null;
    }

  1. 两遍哈希表:先报key-value put进HashMap,再遍历数组。时间复杂度O(n),空间复杂度O(n)。
public static int[] twoSum(int[] nums, int target) {
        HashMap<Integer,Integer> map = new HashMap();
        for(int i= 0;i<nums.length;++i)
        {
            map.put(nums[i],i);
        }
        for(int i=0;i<nums.length;++i){
            if(map.containsKey(target-nums[i])&&map.get(target - nums[i])!=i){
                return new int[]{i,map.get(target-nums[i])};
            }
        }
        return null;
    }
  1. 一遍哈希表:一边put key-value,一边判断。时间复杂度O(n),空间复杂度O(n)。
    这里有个坑,HashMap的key值不能相同,如果先put再判断,遇到有相同两个数的和等于target(例如:[3,3],target:6),因为后面put的值会覆盖前面put的值,所以无法得到正确的结果。
    错误代码如下
public static int[] twoSum(int[] nums, int target) {
        HashMap<Integer,Integer> map = new HashMap();
        for(int i= 0;i<nums.length;++i)
        {
            map.put(nums[i],i);
            if(map.containsKey(target - nums[i]) && map.get(target -nums[i])!=i){
                return new int[]{i,map.get(target - nums[i])};
            }
        }
        return null;
    }

正确代码

public static int[] twoSum(int[] nums, int target) {
        HashMap<Integer,Integer> map = new HashMap();
        for(int i= 0;i<nums.length;++i)
        {
            if(map.containsKey(target - nums[i])){
                return new int[]{i,map.get(target - nums[i])};
            }
            map.put(nums[i],i);
        }
        return null;
    }
发布了12 篇原创文章 · 获赞 5 · 访问量 614

猜你喜欢

转载自blog.csdn.net/qq_43657590/article/details/103945819
今日推荐