본문 바로가기
알고리즘/프로그래머스

[프로그래머스 / 두 개 뽑아서 더하기 / JAVA]

by KDW999 2023. 3. 4.

https://school.programmers.co.kr/learn/courses/30/lessons/68644

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

문제 접근

같은 레벨 1의 문제라도 난이도 차이가 들쑥날쑥인 것 같다.

 

배열 안의 두 수를 뽑아서 더하는 모든 경우의 수를 위해 이중 배열 형태로 첫 배열은 i=0부터 배열의 크기-1 까지 두 번째 배열은 j=i+1부터 배열의 크기까지 크기를 지정하였다.

뽑아서 더한 수는 중복을 걸러주는 Set에다가 저장 / Set은 중복은 걸러주나 순서없이 저장된다.

그렇기에 인덱스가 없는 객체에서 다음 요소를 반환해주는 Iterator 사용

배열 answer에 요소를 하나씩 저장 후 정렬해주기 

import java.util.*;

class Solution {
    public int[] solution(int[] numbers) {
		
		HashSet<Integer> hs = new HashSet<Integer>();
		
		for(int i=0; i<numbers.length-1; i++) {
			for(int j=i+1; j<numbers.length; j++) {
				hs.add(numbers[i]+numbers[j]);
			}
		}
        
		Iterator<Integer> it = hs.iterator();
        
		int[] answer = new int[hs.size()];
        
		for(int i=0; i<hs.size(); i++) {
			answer[i] = it.next();

		}
        Arrays.sort(answer);
		return answer;
	}
}

 

다른 사람 풀이

Set을 HashSet말고 TreeSet으로 만들면 저장한 값들이 알아서 정렬되서 마지막에 배열을 또 정렬해주는 작업을 안해줘도 된다.

import java.util.*;

public 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]);
            }
        }

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

        return answer;
    }
}

댓글