leetcode-1.two sum（两数之和）

最新推荐文章于 2022-11-09 11:23:01 发布

原创最新推荐文章于 2022-11-09 11:23:01 发布 · 166 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法

Java基础专栏收录该内容

10 篇文章

订阅专栏

本文分享了一个编程爱好者在元旦期间通过解决LeetCode上的“两数之和”问题来保持编程状态的经历。从最初的暴力穷举法到使用哈希表优化，详细介绍了算法的优化过程和结果对比，旨在探讨更高效的问题解决方案。

前言：

元旦放假实在无聊，写点代码娱乐自己吧，平时工作负责项目管理方面的东西相对多了一些，作为一个热爱编程的人实在是手痒，趁着假期练练手，当然了，不是为了刷题，就是为了让自己保持一个编程的状态。个人觉得为了刷题而刷题会失去很多写代码的乐趣，由于自己水平有限，如果代码有问题或者有什么更好的办法也欢迎各位大神提意见，感谢。

题目来源：https://leetcode-cn.com/problems/two-sum/submissions/

正文：

给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那两个整数，并返回他们的数组下标。

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

示例:

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

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

1：第一次随意写写，使用了最简单的暴力穷举法，代码如下：

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

该算法的时间复杂度为O(n2),结果如下：

2.由于上面的算法实在是太慢，自己优化了一下，大概想法是使用hash匹配，代码如下：

class Solution {
    public int[] twoSum(int[] nums, int target) {
        Map<Integer, Integer> hashMap = new HashMap();
        //int record = 0;
        for (int i = 0; i < nums.length; i++) {
            hashMap.put(nums[i], i);
        }

        for (int j = 0; j< nums.length; j++) {
            int record = target - nums[j];
            Integer result = hashMap.get(record);
            if (result != null && result != j) {
                return new int[] { j, result };
            }
        }

        return null;
    }
}

该算法的时间复杂度为O(n)，结果如下：