[LeetCode] #1 Two Sum

Two Sum

x
Given an array of integers, return indices of the two numbers such that they add up to a specific target.
​
You may assume that each input would have exactly one solution, and you may not use the same element twice.

 

첫번째 문제 답게 쉬운 문제가 나왔다.

그런데 뭔가 멋있게 풀고싶은 마음이 너무 드글드글해서,

회사 제품 특성상 낮은 jdk버전에도 사용 가능한 코드만 짜다보니 간단하면서도 최근 jdk(라고 해봐야 자바8이지만..ㅎㅎ)에서 사용 가능한 방법으로 풀고싶었다.

 

그래서 우선 스트림을 사용해봤다.

xxxxxxxxxx
import java.util.*;
​
class Solution {
    public int[] twoSum(int[] nums, int target) {
        int[] answer = new int[2];
        ArrayList<Integer> list = new ArrayList<Integer>();
        
        for (int i = 0; i < nums.length; i++) {
            final int temp = target - nums[i];
            if (Arrays.stream(nums).anyMatch(x -> x == temp))
                list.add(i);
        }
        
        if (list.size() == 2) {
            answer[0] = list.get(0);
            answer[1] = list.get(1);
        }
        
        return answer;
    }
}

 

케이스 [3,2,4], 6에서 실패했다.

3은 6의 반값이고.. 3이 배열 안에 또있네? 이러고 값을 리스트에 담았더니,

리스트 사이즈는 3이 돼버리고 빈 배열을 리턴하게되어서 실패.

 

아 그러면 반값일때는 나랑 값이 같으면 패스해주면 되겠다~ 하고 다시 코드를 짰다.

 

xxxxxxxxxx
import java.util.*;
​
class Solution {
    public int[] twoSum(int[] nums, int target) {
        int[] answer = new int[2];
        ArrayList<Integer> list = new ArrayList<Integer>();
        
        for (int i = 0; i < nums.length; i++) {
            final int me = nums[i];
            final int temp = target - nums[i];
            if (Arrays.stream(nums).anyMatch(x -> x == temp && x != me))
                list.add(i);
        }
        
        if (list.size() == 2) {
            answer[0] = list.get(0);
            answer[1] = list.get(1);
        }
        
        return answer;
    }
}

 

결과는~ 또 실패.

아예 본인을 제거했더니, 이제 [3,3], 6 케이스에서 실패했다.

아.. 중복값도 되는거였어? ^^..?

 

이렇게 저렇게 바꿔보다가 결국은 그냥 이중포문으로 해결했다.

 

x
class Solution {
    public int[] twoSum(int[] nums, int target) {
        int[] answer = new int[2];
        
        for (int i = 0; i < nums.length; i++) {
            for (int j = 0; j < nums.length; j++) {
                if (i == j) continue;
                if (nums[i] + nums[j] == target) {
                    answer[0] = i;
                    answer[1] = j;
                    return answer;
                }
            }
        }
        
        return answer;
    }
}

 

답은 맞췄지만 너무나 찝찝하고 ㅠㅠ 조금만 더 고민해보면 답이 나올것같은데.. 싶다.

나중에 다시 스트림으로 도전해봐야겠다.