LeetCode-001  Python&Go版本_Python&Go

闲来无事刷题吧

题目要求:

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

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

示例:

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

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

解题思路

给定数组arr,初始化字典dic,目标值target,a,b为数组的元素
如果存在这样的关系a+b=target,如果a的下标为a_1,则可以得到dic[target-a]=a_1。
简单数就是,当前的元素的下标和目标值减去当前元素获取的值做一个映射,如果映射存在就可以找到这两个数了。

代码

Python Version

from typing import List
class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        d =dict()
        for index,item in enumerate(nums):
            if item in d:
                return [d[item],index]
            else:
                d[target-item] =  index 

            #或者   
            # if target-item in d:
            #     return [d[target-item],index]
            # d[item] = index

if __name__ == "__main__":
    l:list = [2,7,11,15,12]
    target:int = 19
    s = Solution()
    print(s.twoSum(l,target))            

Go Version

package main
import "fmt"

func solution(nums []int,target int) []int{
    m:=make(map[int]int)
    for index,value := range nums {
        if w,ok:=m[value];ok{
            return []int{w,index}
        } else{
            m[target-value] = index
        }
    }
    return nil
}
func main() {
    nums:=[]int{2,7,11,15,12}
    fmt.Println("nums",nums)
    target:=19
    fmt.Println(solution(nums,target))

}