두 개 뽑아서 더하기(solution.ver)

2021.03.16 - [CordingTest] - 두 개 뽑아서 더하기

두 개 뽑아서 더하기

이전에 했던 코딩테스트는 정답은 맞지만 코드의 길이가 너무 길고 비효율적이라

 

실용적이지 못했다. 이번에는 자동으로 오름차순 정렬과 중복을 제거해주는 TreeSet클래스를 사용해서

 

코드의 길이를 줄이고 가독성을 높여봤다. 아래에 코드를 첨부한다.

 

 

 

 

 
class Solution {
    public int[] solution(int[] numbers) {
        TreeSet<Integer> store = new TreeSet<Integer>();    
        int index=0;
        for(int i=0;i<numbers.length;i++) {
            
            for(int j=i+1;j<numbers.length;j++)
                
                store.add(numbers[i]+numbers[j]);
 
        }
        Iterator result =store.iterator();        
        int[] answer= new int[store.size()];
        
        while(result.hasNext()) {
            
            int x=(int)result.next();
            answer[index]=x;
            index++;
        }
        
        
     return answer;     
    }
 
}

코드의 길이가 훨씬 짧아졌다.