每日一题（六）

1.两数之和

• 循环生成数组

# enumerate枚举，生成索引和对应数字
# 第一个i是对数组的操作，i,x in...是生成枚举数组
a = [i for i, x in enumerate(nums) if x == another]

---

"""
给定一个整数数组nums和一个整数目标值target，请你在该数组中找出和为目标值target的那两个整数，并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。

示例 1：

输入：nums = [2,7,11,15], target = 9
输出：[0,1]
解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。

示例 2：

输入：nums = [3,2,4], target = 6
输出：[1,2]

示例 3：

输入：nums = [3,3], target = 6
输出：[0,1]

"""

class Solution(object):
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        for i in range(len(nums)):
            another = target - nums[i]
            print('预计另一个数是', another, '第', i, '轮')
            a = [j for j, x in enumerate(nums) if x == another]
            if len(a) != 0 and a[0] != i:
                return [i, a[0]]
            elif len(a) > 1 and a[0] != i:
                return [i, a[1]]


if __name__ == '__main__':
    S = Solution()
    nums = [2, 7, 11, 15]
    target = 9
    print(S.twoSum(nums, target))