Merge pull request halfrost#42 from JessonChan/master

halfrost · web-flow · commit 6cb397f0025a · 2020-08-14T07:28:49.000+08:00
use range-for
diff --git a/leetcode/0001.Two-Sum/1. Two Sum.go b/leetcode/0001.Two-Sum/1. Two Sum.go
@@ -2,12 +2,11 @@ package leetcode
 
 func twoSum(nums []int, target int) []int {
 	m := make(map[int]int)
-	for i := 0; i < len(nums); i++ {
-		another := target - nums[i]
-		if _, ok := m[another]; ok {
-			return []int{m[another], i}
+	for k, v := range nums {
+		if idx, ok := m[target-v]; ok {
+			return []int{idx, k}
 		}
-		m[nums[i]] = i
+		m[v] = k
 	}
 	return nil
 }

Original file line number	Diff line number	Diff line change
`@@ -2,12 +2,11 @@ package leetcode`
`2`	`2`
`3`	`3`	`func twoSum(nums []int, target int) []int {`
`4`	`4`	`m := make(map[int]int)`
`5`		`- for i := 0; i < len(nums); i++ {`
`6`		`- another := target - nums[i]`
`7`		`- if _, ok := m[another]; ok {`
`8`		`- return []int{m[another], i}`
	`5`	`+ for k, v := range nums {`
	`6`	`+ if idx, ok := m[target-v]; ok {`
	`7`	`+ return []int{idx, k}`
`9`	`8`	`}`
`10`		`- m[nums[i]] = i`
	`9`	`+ m[v] = k`
`11`	`10`	`}`
`12`	`11`	`return nil`
`13`	`12`	`}`