[프로그래머스] Lv.1 두 개 뽑아서 더하기

문제 링크
https://programmers.co.kr/learn/courses/30/lessons/68644

코딩테스트 연습 - 두 개 뽑아서 더하기

 

내 풀이

• 한 배열 안에서 두 개의 정수의 합을 구하는 과정을 두 반복문으로 구현
• 결과 값에 중복이 존재할 수 있으므로, contains 메소드로 검사.
• 결과 배열의 원소 갯수가 가변적이므로 배열 대신 arraylist를 사용 후 배열로 다시 만듬.
• 정렬은 Collections 클래스의 sort메소드 활용

import java.util.ArrayList;
import java.util.Collections;
class Solution {
    public int[] solution(int[] numbers) {
        ArrayList<Integer> list = new ArrayList<>();
        for(int i=0; i<numbers.length-1; i++){
            for(int j=i+1; j<numbers.length; j++){
                int sum = numbers[i]+numbers[j];
                if(!list.contains(sum)){
                    list.add(sum);
                }
            }
        }
        Collections.sort(list);
        int[] answer=new int[list.size()];
        for(int i=0; i<list.size();i++){
            answer[i]=list.get(i);
        }
        
        return answer;
    }
}

 

다른 풀이의 아이디어

• 정렬과 중복 제거를 제공하는 treeset을 사용.
• 요소 접근에 iterator 사용

import java.util.Iterator;
import java.util.Set;
import java.util.TreeSet;

class Solution {
    public int[] solution(int[] numbers) {
        Set<Integer> sumNumber = new TreeSet();

        for(int i = 0; i < numbers.length-1; i++){
            for(int j = i+1; j < numbers.length; j++){
                sumNumber.add(numbers[i] + numbers[j]);
            }
        }

        int[] answer = new int[sumNumber.size()];
        int index = 0;
        Iterator itor = sumNumber.iterator();
        while(itor.hasNext()){
            answer[index] = (int)itor.next();
            index++;
        }

        return answer;
    }
}