알고리즘/프로그래머스
[프로그래머스] 두 개 뽑아서 더하기
말랑말랑쓰
2021. 4. 7. 21:07
[21-04-07]
오늘은 머리가 안돌아가서 쉬운 문제로...ㅎㅅㅎ...
문제
정수 배열 numbers가 주어집니다. numbers에서 서로 다른 인덱스에 있는 두 개의 수를 뽑아 더해서 만들 수 있는 모든 수를 배열에 오름차순으로 담아 return 하도록 solution 함수를 완성해주세요.
제한사항
- numbers의 길이는 2 이상 100 이하입니다.
- numbers의 모든 수는 0 이상 100 이하입니다.
입출력 예
numbers | result |
[2,1,3,4,1] | [2,3,4,5,6,7] |
[5,0,2,7] | [2,5,7,9,12] |
풀이
import java.util.HashSet;
import java.util.Set;
import java.util.Arrays;
class Solution {
public int[] solution(int[] numbers) {
int[] answer = {};
Set<Integer> set = new HashSet<>();
for(int i=0 ; i<numbers.length-1 ; i++) {
for(int j=i+1 ; j<numbers.length ; j++) {
int sum = numbers[i] + numbers[j];
set.add(sum);
}
}
int index = 0;
answer = new int[set.size()];
for(Integer s : set) {
answer[index++] = s;
}
Arrays.sort(answer);
return answer;
}
}
set에 담고 array로 옮기는 걸 단순하게 반복문을 사용해서 했는데 다른사람들 풀이를 보니
return set.stream().sorted().mapToInt(Integer::intValue).toArray();
와 같이 할 수 있었다. 뭔가 복잡하지만 외워두면 요긴하게 쓸 듯
자바는 이런 저런 utill 함수가 많기도 하고 업데이트도 자주 되서 틈틈히 공부를 해야겠당