Go map解决力扣问题

力扣问题

给定一个整数数组 arr 和一个目标值 target，在该数组中找出加和等于目标值的两个整数，并返回它们在原数组中的下标。假设，原数组中没有重复元素，而且有且只有一组答案。

代码

package main

import (
    "fmt"
    "encoding/json"
)

func twoSum(arr []int, target int) (newArr []int) {
	m := make(map[int]int)
	newArr = []int{0}
	for i := 0; i < len(arr); i++ {
		m[arr[i]] = i
		complement := target - arr[i]
		v, ok := m[complement]
		if ok == true && v != i {
			newArr = []int{v,i}
			return newArr;
		} 
	}
	return newArr
}

func main() {
	arr := []int{ 10, 92, 21, 83, 32, 74 }
	str,err := json.Marshal(arr)
	if err != nil {
		panic(err)
	}
	fmt.Printf("原始数据：%s\n",string(str))
	target := 102
	fmt.Printf("目标值：%v\n",target)
	newArr := twoSum(arr, target)
	newStr,err := json.Marshal(newArr)
	if err != nil {
		panic(err)
	}
	fmt.Printf("数组下标：%s\n",string(newStr))
}

执行

$ go run target.go
原始数据：[10,92,21,83,32,74]
目标值：102
数组下标：[0,1]