From 6bcb66bf006d681c5f1408880b52b7d6eb608ae4 Mon Sep 17 00:00:00 2001 From: ductnn Date: Tue, 28 Nov 2023 00:09:06 +0700 Subject: [PATCH] update sol --- leetcode/01.two_sum/two_sum.go | 18 +++++++++++++++++- 1 file changed, 17 insertions(+), 1 deletion(-) diff --git a/leetcode/01.two_sum/two_sum.go b/leetcode/01.two_sum/two_sum.go index 8a6c978..31faf70 100644 --- a/leetcode/01.two_sum/two_sum.go +++ b/leetcode/01.two_sum/two_sum.go @@ -19,8 +19,24 @@ func twoSum(nums []int, target int) []int { return results } +func twoSum1(nums []int, target int) []int { + m := map[int]int{} + + for i := range nums { + temp := nums[i] + search := target - temp + + if j, ok := m[search]; ok { + return []int{i, j} + } + + m[temp] = i + } + return []int{} +} + func main() { nums := []int{2, 7, 11, 15} target := 9 - fmt.Println(twoSum(nums, target)) + fmt.Println(twoSum1(nums, target)) }