哈希表优化求解两数之和

题目

给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。

解法1:暴力穷举法

class Solution {
    
    
    public int[] twoSum(int[] nums, int target) {
    
    
        int[] twoSum = new int[2];
        boolean flage = false;
        for(int i = 0; i < nums.length; i++){
    
    
            for(int j = i+1; j < nums.length; j++ ){
    
    
                if(nums[i] + nums[j] == target){
    
    
                    twoSum[0] = i;
                    twoSum[1] = j;
                    flage = true;
                    break;
                }  
            }
            if(flage){
    
     
                break;
            }  
        }
        return twoSum;
    }
}

上述代码过于冗余,不够简洁。

  1. 不用着急定义一个新的数组进行接收,只需在返回时直接定义要返回的数组即可。
  2. 外循环的 if 并不需要,只需要在内循环 return 可靠,不需要 break。
改进后
class Solution {
    
    
    public int[] twoSum(int[] nums, int target) {
    
    
        int n = nums.length;
        for (int i = 0; i < n; ++i) {
    
    
            for (int j = i + 1; j < n; ++j) {
    
    
                if (nums[i] + nums[j] == target) {
    
    
                    return new int[]{
    
    i, j};
                }
            }
        }
        return new int[0];
    }
}

暴力穷举的时间复杂度为O(N^2),利用哈希表可以将时间复杂度降到O(N)。

解法2:哈希表

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

猜你喜欢

转载自blog.csdn.net/qq_44125965/article/details/128142682