[LeetCode] Two Sum (with Go)

최근 Go 에 관심을 가지고 있습니다.

알고리즘 문제를 Java 혹은 Go로 풀어볼 생각입니다.

LeetCode에서 기초적인 문제부터 풀어보았습니다.

 

Given an array of integers nums and an integer target, return indices of the two numbers such that they add up to target.

You may assume that each input would have exactly one solution, and you may not use the same element twice.

You can return the answer in any order.

 

Example 1:

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

 

 

nums의 배열 중 2개의 합이 target이 되는

배열의 인덱스를 배열에 담아 응답하면 됩니다.

 

아래는 go로 풀어본 답입니다.

java를 하던 버릇도 있고 아직은 헷갈리네요

 

 

func twoSum(nums []int, target int) []int {
    
    answer := []int{}
    for i, number := range nums{
        for j, second := range nums{
            
            if i != j{
                isTwoSum := isSame(number, second, target)
                if isTwoSum{
                    answer = append(answer, i)
                    answer = append(answer, j)
                    return answer
                }
            }
        }   
    }
    return answer
}

func isSame(first, second, target int) bool{
    if first + second == target{
        return true
    }
    return false
}

 

 

 

문제 출처 : https://leetcode.com/problems/two-sum/

Two Sum - LeetCode