[Baekjoon] 11004번 : K 번째 수

1. 문제

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

• 정렬 문제

// 문제
수 N개 A1, A2, ..., AN이 주어진다.
A를 오름차순 정렬했을 때,
앞에서부터 K번째 있는 수를 구하는 프로그램을 작성하시오.

// 입력
첫째 줄에 N(1 ≤ N ≤ 5,000,000)과 K (1 ≤ K ≤ N)이 주어진다.

둘째에는 A1, A2, ..., AN이 주어진다. (-109 ≤ Ai ≤ 109)

// 출력
A를 정렬했을 때, 앞에서부터 K번째 있는 수를 출력한다.

// 예제 입력 1
5 2
4 1 2 3 5

// 예제 출력 1
2

2. 핵심 아이디어

• 데이터의 개수가 최대 5000000개
• 시간 복잡도 O(NlogN)의 정렬 알고리즘을 이용해야 함
• 고급 정렬 알고리즈을 이용해 문제 해결 가능
• 혹은 파이썬의 기본 정렬 라이브러리를 이용해 문제 해결 가능
• 시간적 이점을 위해 PyPy3을 선택해 코드를 제출

3. Python 문제풀이

# Merge Sort(Python3으로 제출시 오답)
import sys
input = sys.stdin.readline

def merge_sort(arr) :
  if len(arr) <= 1 :
    return arr

  mid = len(arr) // 2
  left = merge_sort(arr[:mid])
  right = merge_sort(arr[mid:])

  i, j, k = 0, 0, 0

  while i < len(left) and j < len(right) :
    if left[i] < right[j] :
      arr[k] = left[i]
      i += 1
    else :
      arr[k] = right[j]
      j += 1
    k += 1

  if i == len(left) :
    while j < len(right) :
      arr[k] = right[j]
      j += 1
      k += 1
  elif j == len(right) :
    while i < len(left) :
      arr[k] = left[i]
      i += 1
      k += 1
  return arr

n, k = map(int, input().split())
arr = list(map(int, input().split()))

arr = merge_sort(arr)

print(arr[k-1])

# 기본 정렬 라이브러리
import sys
input = sys.stdin.readline

n, k = map(int, input().split())
arr = list(map(int, input().split()))

arr = sorted(arr)
print(arr[k-1])

4. Java 문제풀이

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

class Main {
  public static void main(String[] args) throws IOException {
    BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    StringTokenizer st = new StringTokenizer(br.readLine());

    int n = Integer.parseInt(st.nextToken());
    int k = Integer.parseInt(st.nextToken());
    int[] arr = new int[n];

    st = new StringTokenizer(br.readLine());

    for (int i=0; i<n; i++) {
      arr[i] = Integer.parseInt(st.nextToken());
    }
    Arrays.sort(arr);
    System.out.println(arr[k-1]);
  }
}