LeetCode算法第一题--两数之和

LeetCode算法第一题–两数之和

---

问题描述

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/two-sum

---

原因分析：

问题一：数组中同一元素在答案里不能重复出现

例如：
nums = [3,2,4]
target = 6
如果使用第一次的代码：

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

输出结果为[0,0]
预期输出应该是[1,2]

---

解决问题：

将第二个for循环进行修改

class Solution {
    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 new int[0];  
    } 
}

该方法时间复杂度为O(n^2)，其中 n 是数组中的元素数量。最坏情况下数组中任意两个数都要被匹配一次。

改进算法：

使用哈希表。
降低时间复杂度为O(n)，个人理解为将哈希表的索引当作第二个for循环。

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])) { //判断哈希表的键是否存在target-nums[i]。如果存在，返回通过Map的get方法获得对应键的值（即数组的索引）和当前数组的索引
                return new int[]{hashtable.get(target - nums[i]), i};
            }
            hashtable.put(nums[i], i); //将数组的值作为键，索引作为值放入到哈希表中
        }
        return new int[0];
    }
}

总结

第一次简单算法题过程艰难 2022-3-29