본문 바로가기

코딩테스트

[Java] 프로그래머스 - 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입니다.

 

이 문제는 채점항목에 효율성이 없어서 통과했다..

뭔가 똥같이 코딩한거같은데 채점했을때 다 맞아서 신기했음ㅋㅋㅋㅋㅋㅋㅋ

이 느낌같아서 찝-찌입

암튼 통과한 코드는 다음과 같다.

import java.util.Arrays;

class Solution {
    public int[] solution(int[] array, int[][] commands) {

        // array = {1,5,2,6,3,7,4} / commands = {{2,5,3},{4,4,1}};
        int[] answer = new int[commands.length];

        int start = 0;
        int end = 0;
        int index = 0;

        // 여기서 i인 commands.length는 commands 배열안에 배열의 갯수를 나타낸다. -> 2
        for (int i = 0; i < commands.length; i++) {
            int[] inArr = commands[i];

            // 여기서 j인 commands[i].length는 배열안에 배열에 숫자의 갯수를 나타낸다. -> 3
            for (int j = 0; j < inArr.length; j++) {

                // {i,j,k} -> i번째 숫자부터 j번째 숫자까지 자르고 정렬했을 때, k번째에 있는 수
                // 문제에 따라 0일 경우 start 변수에, 1일 경우 end 변수에, 2일 경우 index 변수에 넣어준다.
                if (j == 0) {
                	start = inArr[j];
                }
                if (j == 1) {
                	end = inArr[j];
                }

                if (j == 2) {
                    index = inArr[j];

                    // 변수 세팅이 다 되었으면 copyOfRange 메서드를 통해 배열을 잘라준다.
                    int[] cutArray = Arrays.copyOfRange(array, start-1, end);

                    // 문제에 따라 자른 배열을 정렬하고
                    Arrays.sort(cutArray);

                    // return 값인 배열변수에 넣어준다. 
                    // (-1을 해주는 이유는 java의 index는 0 부터 시작이기 때문)
                    answer[i] = cutArray[index-1];
                }
            }
        }
        return answer;
    }
}

주석을 달아봤는데 나중에 봤을때 이해를 했으면 좋겠다!

 

아 글고 나는 코딩할때 아래 같은 코드가 왜케 손에 안익지

if (j == 0) {
	start = inArr[j];
}

 

if (j == 0) 
	start = inArr[j];

둘다 똑같은 코드인데 아래 코드는 눈에 안익어서 코딩할때 자꾸 위에 코드처럼 한당 대가리 극극