[문제] 프로그래머스 레벨 1: 두 개 뽑아서 더하기

programmers.co.kr/learn/courses/30/lessons/68644

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

풀이

입력받는 숫자들 중 2개를 뽑아서 set에 저장하면 된다. 

이때 나는 재귀를 이용한 조합 함수를 만들었는데, 어차피 2개의 수니까 그럴 필요 없이 for문으로 풀어도 된다.

 

1. 조합 (재귀 이용)

import java.util.Arrays;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Set;

class Solution {
    static int N;
    static Set<Integer> set = new HashSet<>();
    public int[] solution(int[] numbers) {
      N = numbers.length; 
	      combination(0,new int[2],0,numbers);
	        int[] answer = new int[set.size()];
	        Iterator<Integer> it = set.iterator();
	        int i=0;
	        while(it.hasNext()) {
	        	answer[i++] = it.next();
	        }
	        Arrays.sort(answer);
	        return answer;
    }
    static void combination(int toSelect, int[] selected, int startIdx,int[] numbers){
        if(toSelect == 2){
            set.add(selected[0]+ selected[1]);
            return;
        }
        for(int i=startIdx;i<N;i++){
            selected[toSelect] = numbers[i];
            combination(toSelect+1, selected, i+1, numbers);
        }
    }
}

2. for문 이용

import java.util.Arrays;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Set;
class Solution { 
    public int[] solution(int[] numbers) {
        Set<Integer> set = new HashSet<>();
        int N = numbers.length; 
	    for(int i= 0;i<N-1;i++){
            for(int j=i+1;j<N;j++){
                set.add(numbers[i]+numbers[j]);
            }
        }
        int[] answer = new int[set.size()];
        Iterator<Integer> it = set.iterator();
        int i = 0;
        while(it.hasNext()){
            answer[i++] = it.next();
        }
        Arrays.sort(answer);
	    return answer;
    }
   
}