[JAVA] 백준 11004번 : K번째 수

https://www.acmicpc.net/problem/11004

정렬을 이용하는 문제이다. 문제를 푸는 데 어려움이 크게 없었다.

하지만, 이 문제에서 N이 필요가 없다고 생각하였고, 결과값도 똑같이 나왔다. 그 코드는 아래와 같다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;

public class Main {
    public static void main(String[] arsg) throws IOException{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] input = br.readLine().split(" ");
        int N = Integer.parseInt(input[0]);
        int K = Integer.parseInt(input[1]); //앞에서부터 K번째 있는 수

        String[] input2 = br.readLine().split(" ");
        Arrays.sort(input2); //정렬
        System.out.println(input[K-1]);

        br.close();
    }
}

하지만, 시간 초과로 실패가 떴고, 그래서 N을 사용하도록 배열을 하나 선언해주어 선언한 배열에 string값을 형 변환 해서 넣어주었다. 그 코드는 아래와 같다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;

public class Main {
    public static void main(String[] arsg) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] input = br.readLine().split(" ");
        int N = Integer.parseInt(input[0]);
        int K = Integer.parseInt(input[1]); //앞에서부터 K번째 있는 수

        String[] input2 = br.readLine().split(" ");
        int[] nums = new int[N];
        for (int i = 0; i < N; i++) {
            nums[i] = Integer.parseInt(input2[i]);
        }
        Arrays.sort(nums); //정렬
        System.out.println(nums[K - 1]);

        br.close();
    }
}