K번째 큰 수 찾기

1. 문제

2. 입출력

3. 입출력 예시

4. 문제 설계

1. 문제의 크기가 100,000 이므로 O(n^2)으로는 문제를 해결할 수 없음.
2. 따라서, 삽입 정렬, 선택 정렬, 버블 정렬 보다는 합병정렬, 퀵 정렬 등으로 문제를 해결할 수 있음
3. 하지만 내장 함수인 sort() 혹은 sorted()가 정렬 알고리즘 중에는 가장 빠르므로 해당 함수를 사용함
4. 입력 받은 정렬을 sorted() 함수로 정렬 후 k = 1일 경우 제일 큰 원소 이므로 거꾸로 인덱스를 접근 함으로써 문제를 해결함

5. 전체 코드

//
//  main.swift
//  Knumber
//
//  Created by 이영재 on 2022/02/28.
//MARK: - k번째 큰 수 찾기

//MARK: - Framework
import Foundation

//MARK: - Function
func solution() -> Void {
    //MARK: - input
    guard let input = readLine()?.components(separatedBy: " ") else { return }
    guard let inputArray = readLine()?.components(separatedBy: " ") else { return }
    
    let N: Int = input.map { Int($0) }[0] ?? 0
    let k: Int = input.map { Int($0) }[1] ?? 0
    
    //MARK: - process
    let sortedArray: Array<Int> = inputArray.map { Int($0) ?? 0 }.sorted { $0 < $1 }
    let result: Int = sortedArray[N - k]
    
    //MARK: - output
    print(result)
}
solution()

전체코드는 여기에서 확인할 수 있습니다.