之前用的python,后面简单学了下go,打算之后的leetcode都用python和go分别做一遍了。
题目描述
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
go
package main
import "fmt"
func twoSum(nums []int, target int) []int {
for k1, _ := range nums {
for k2 := k1 + 1; k2 < len(nums); k2++ {
if target == nums[k1]+nums[k2] {
return []int{k1, k2}
}
}
}
return []int{}
}
func main() {
nums := []int{2, 7, 11, 15}
var target int = 26
fmt.Printf("%d\n", twoSum(nums, target))
}
python
class Solution(object):
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
# 从第一个元素开始遍历:
for i in range(len(nums)):
# 从第二个元素开始遍历:
for j in range(i+1,len(nums)):
# 根据索引取列表元素的值
if nums[i] + nums[j] == target:
# 返回索引
return [i,j]
s = Solution()
print(s.twoSum([11, 7, 2, 15],9))