两数之和【LeetCode】

计划从这周开始,每周刷一道leetcode,目的在于练习基础算法,同时每次强制用golang实现,以此达到学习golang的目的。由于刚开始,所以这一题也比较简单。

题目

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

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

示例:

给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

实现思路:

  • 从前往后遍历数组,每次那当前数组和后面所有的数字进行相加比较,如果符合条件,返回两数索引。

这里贴一下java实现

public static int[] twoSum(int[] nums, int target) {

    int[] index = new int[2];
    for (int i = 0; i < nums.length; i++) {
        for (int j = i + 1; j < nums.length; j++) {
            if (nums[i] + nums[j] == target) {
                index[0] = i;
                index[1] = j;
                return index;
            }
        }
    }
    return index;
}

最近想了想,突然觉得这道题的解应该是使用Hash将算法复杂度降低道O(n)

public static int[] twoSum2(int[] nums, int target) {

    Map number = new HashMap<>(nums.length);
    number.put(target - nums[0], 0);
    for (int i = 1; i < nums.length; i++) {
        int num = nums[i];
        Integer j = number.get(target - num);
        if (j == null) {
            number.put(num, i);
        } else {
            return new int[]{j, i};
        }
    }
    return null;
}

奇怪的是,我把同样的逻辑翻译为go之后,居然跑的比java还慢,也是醉了:

func towSum(nums []int, target int) []int  {
    var index = [2]int{}
    for i, a := range nums {
        for j, b := range nums[i + 1:] {
            if a + b == target {
                index[0] = i;
                index[1] = i + j + 1
                return index[:]
            }
        }
    }
    return index[:]
}

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

你可能感兴趣的:(两数之和【LeetCode】)