Leetcode 1 - two sum (python)

Input으로 들어온 리스트의 원소 중 두 원소를 합쳐서 target 숫자를 만드는 문제.

 

temp에 target에서 input 리스트의 원소를 순서대로 빼기를 하여 

해당 temp 숫자가 input 리스트에 있는지 확인하는 방식으로 풀었다.

 

ex 1.

Input: nums = [2,7,11,15], target = 9
Output: [0,1]
Explanation: Because nums[0] + nums[1] == 9, we return [0, 1].

ex 2.

Input: nums = [3,2,4], target = 6
Output: [1,2]

 

from typing import List

def two_sum(nums:List[int], target:int):
    result = []

    for i, num in enumerate(nums):
        temp = target - num

        if temp in nums[i+1:]:
            # result.append(i)
            result.append(nums.index(num))
            result.append(nums[i+1:].index(temp) + (i+1))
    #print(result)
    return(result)

two_sum([3,3], 6)

 

https://leetcode.com/problems/two-sum/

Two Sum - LeetCode