您的位置 首页 golang

LeetCode1.两数之和


问题描述:

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

示例:

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

解题思路:

1.暴力法

利用两层循环查找和为目标值的两个整数,并返回下标。

2.HashMap两层循环

先将数组中的值和下标作为key和value存到HashMap中,然后遍历数组,计算差值是否在HashMap中,若存在,返回对应两个目标值的下标。

注意:有缺陷,对于数组中存在相同元素,得到的下标,有问题。例如

[3, 3] 6 结果应为[0,0],实际为[1,1]

3.HashMap一次循环

将数组元素和下标依次存到HashMap中,同时寻找当前元素的目标差值是否已经在HashMap中,若存在,则返回目标差值的下标和当前循环下标。

代码展示:

Java版:

import java.util.HashMap;/* * @lc app=leetcode.cn id=1 lang=java * * [1] 两数之和 */class Solution {    //HashMap一次循环    public int[] twoSum(int[] nums, int target) {        HashMap<Integer, Integer> map = new HashMap<>();                for(int j = 0; j < nums.length; j++){            int sub = target - nums[j];            if(map.containsKey(sub)){                return new int[]{map.get(sub), j};            }            map.put(nums[j], j);        }        return new int[2];    }    //HashMap两次循环    public int[] twoSum1(int[] nums, int target) {        HashMap<Integer, Integer> map = new HashMap<>();                for(int i = 0; i < nums.length; i++){            map.put(nums[i], i);        }        for(int j = 0; j < nums.length; j++){            int sub = target - nums[j];            if(map.containsKey(sub) && sub != nums[j]){                return new int[]{map.get(sub), j};            }        }        return new int[2];    }    //暴力法    public int[] twoSum2(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[2];    }}

Go版:

package leetcode// 暴力法func twoSum(nums []int, target int) []int {    result := make([]int, 2)    for i := 0; i < len(nums); i++ {        for j := i + 1; j < len(nums); j++ {            if nums[i]+nums[j] == target {                result[0] = i                result[1] = j                return result            }        }    }    return result}// HashMap一次循环func twoSum2(nums []int, target int) []int {    result := make([]int, 2)    m := make(map[int]int)    for j := 0; j < len(nums); j++ {        sub := target - nums[j]        v, ok := m[sub]        if ok {            result[0] = v            result[1] = j            return result        }        m[nums[j]] = j    }    return result}// HashMap两次循环,有缺陷func twoSum3(nums []int, target int) []int {    result := make([]int, 2)    m := make(map[int]int)    for i := 0; i < len(nums); i++ {        m[nums[i]] = i    }    for j := 0; j < len(nums); j++ {        sub := target - nums[j]        v, ok := m[sub]        if ok && v != nums[j] {            result[0] = v            result[1] = j            return result        }    }    return result}

文章来源:智云一二三科技

文章标题:LeetCode1.两数之和

文章地址:https://www.zhihuclub.com/778.shtml

关于作者: 智云科技

热门文章

网站地图