[Java] 프로그래머스 Lv1. K번째 수 풀이

문제 설명

배열 array의 i번째 숫자부터 j번째 숫자까지 자르고 정렬했을 때, k번째에 있는 수를 구하려 합니다.

예를 들어 array가 [1, 5, 2, 6, 3, 7, 4], i = 2, j = 5, k = 3이라면

1. array의 2번째부터 5번째까지 자르면 [5, 2, 6, 3]입니다.
2. 1에서 나온 배열을 정렬하면 [2, 3, 5, 6]입니다.
3. 2에서 나온 배열의 3번째 숫자는 5입니다.

배열 array, [i, j, k]를 원소로 가진 2차원 배열 commands가 매개변수로 주어질 때, commands의 모든 원소에 대해 앞서 설명한 연산을 적용했을 때 나온 결과를 배열에 담아 return 하도록 solution 함수를 작성해주세요.

제한사항

• array의 길이는 1 이상 100 이하입니다.
• array의 각 원소는 1 이상 100 이하입니다.
• commands의 길이는 1 이상 50 이하입니다.
• commands의 각 원소는 길이가 3입니다.

나의 풀이

이 문제는 배열에 관한 문제로, 배열을 특정 인덱스에 맞추어 자르고, 이를 정렬하는 것이 포인트이다.
copyOfRange(int[] original array, int from, int to) 라는 함수를 이용해 해결할 수 있다. 자르고자 하는 원본 배열을 from 인덱스부터 to 인덱스까지 잘라 새로운 배열을 반환하는 메소드이다. 단, to 로 지정한 인덱스는 포함하지 않는다.

import java.util.*; 
 
class Solution {
    public int[] solution(int[] array, int[][] commands) {
        int[] answer = new int[commands.length];
        for (int i = 0; i < answer.length; i++) {            
            int[] answer2 = Arrays.copyOfRange(array, commands[i][0]-1, commands[i][1]);
            Arrays.sort(answer2);
            answer[i] = answer2[commands[i][2]-1];
        }        
        return answer;
    }
}

참고) 문자열을 자르는 방법

copyOfRange()는 문자열을 인덱스에 맞추어 자르는 substring()과 유사하다. 문자열을 자르는 메소드를 간단히 정리하겠다.
특정 문자를 기준으로 문자열을 자를 때에 indexOf()와 substring()을 사용하며,
특정 문자가 반복되는 경우에는 split()을 활용한다.

• indexOf(): 특정 문자열의 인덱스 값을 리턴한다.

• substring(int from, int to): from 인덱스부터 to 인덱스까지 잘라 문자열을 리턴한다. 단, to로 지정한 인덱스는 포함하지 않는다. from 인덱스만을 인자로 받을 경우, from 인덱스부터 문자열 끝까지 잘라 리턴한다. 이 경우에는 마지막 문자까지 포함한다.

  String a = "abcde";
  String b = a.substring(0, 3);
  b의 출력값: abc

• split(): 반복되는 특정 문자를 기준으로 문자열을 잘라 배열로 리턴한다.

  String tel = "010-1234-1515";
    String[] number = tel.split("-");
    for (int i = 0; i < number.length; i++) {
        System.out.println(number[i]);
    }
  출력값: 
  010
  1234
  1515

다른 사람의 풀이

나의 풀이와 동일하다. 추천수가 많은 풀이법이랑 동일하게 풀었다니 기분 좋다! :-)