【leetcode】力扣刷题(1):两数之和(go语言)

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/u010168781/article/details/89499367
一、问题描述

给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

二、go语言
1、暴力法
// leetcode-1
package main

import (
	"fmt"
)

func twoSum(nums []int, target int) []int {
	for i := 0; i < len(nums); i++ {
		for j := i + 1; j < len(nums); j++ {
			if nums[i]+nums[j] == target {
				return []int{i, j}
			}
		}
	}
	return []int{}
}

func main() {
	nums := []int{1, 3, 5, 2, 7, 11, 15}
	for index, value := range nums {
		fmt.Printf("index = %d, value = %d\n", index, value)
		fmt.Println("index = %d, value = %d", index, value) //不支持格式化参数
	}
	result := twoSum(nums, 9)
	fmt.Println(result)
}

输出结果

index = 0, value = 1
index = %d, value = %d 0 1
index = 1, value = 3
index = %d, value = %d 1 3
。。。
[3 4]
2、两遍哈希表
// leetcode-1
package main

import (
	"fmt"
)

func twoSum(nums []int, target int) []int {
	mapNum := make(map[int]int)
	size := len(nums)
	for i := 0; i < size; i++ {
		mapNum[nums[i]] = i
	}

	for i := 0; i < size; i++ {
		complement := target - nums[i]
		if v, ok := mapNum[complement]; ok {
			if v != i {
				return []int{v, i}
			}
		}
	}
	return []int{}
}

func main() {
	nums := []int{1, 3, 5, 2, 7, 11, 15}
	result := twoSum(nums, 9)
	fmt.Println(result)
}

输出

[4 3]
3、一遍哈希表
// leetcode-1
package main

import (
	"fmt"
)

func twoSum(nums []int, target int) []int {
	mapNum := make(map[int]int)
	size := len(nums)
	for i := 0; i < size; i++ {
		complement := target - nums[i]
		if v, ok := mapNum[complement]; ok {
			//此处不用判断v != i,因为此时mamNum中还没有添加i值
			return []int{v, i}
		}
		mapNum[nums[i]] = i
	}
	return []int{}
}

func main() {
	nums := []int{1, 3, 5, 2, 7, 11, 15}
	result := twoSum(nums, 9)
	fmt.Println(result)
}

输出

[3 4]

猜你喜欢

转载自blog.csdn.net/u010168781/article/details/89499367