LeetCode第一题Two Sum

最新推荐文章于 2022-02-15 12:11:10 发布

原创最新推荐文章于 2022-02-15 12:11:10 发布 · 168 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #leetcode

笔记专栏收录该内容

10 篇文章

订阅专栏

本文探讨了在解决“两数之和”问题时如何利用HashMap进行高效查找的方法。通过对比两种不同实现方式，展示了如何将时间复杂度从O(n^2)优化到O(n)，并分享了编程实践中使用HashMap提高效率的经验。

一开始没想到HashMap，看了别人的答案之后才知道用hashmap更快捷，第一次的代码如下：

class Solution {
    public int[] twoSum(int[] nums, int target) {
        int[] result = new int[2];
        Map<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){
                result[0]=i;
                result[1]=map.get(target-nums[i]);
                return result;
            }
        }
        return null;
        
    }
}

这样其实复杂度还不是最优的情况，因为有两个for循环，于是改变思路，只用一个for，代码如下：

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

以后尽量在需要使用查找的时候想一想能不能使用map来定位。