[Swift] 프로그래머스 연습 문제(Level 1) : K번째수

실제 코딩테스트의 문제를 통하여 알고리즘 분석과 코딩 능력을 향상시킵니다.

 

 

#. 구독 대상

• 기본 알고리즘을 코딩 테스트 문제를 통하여 학습하고 싶으신 분
• 취업 및 이직을 준비하고 계신 개발자
• Swift를 통하여 코딩 테스트 문제를 살펴보고 이해를 하고 싶으신 분
• 코딩 테스트에 대한 거부감을 없애기 위하여 기초부터 하나씩 공부해보고 싶으신 분
• 기타 알고리즘과 문제 해결 능력에 대해서 관심이 있는 모든 개발자분

참고 사항

본 코딩 테스트 문제에 대한 설명 및 해결 방안은 최적의 답이 아닐 수 있습니다.

본 강의에서 지향하는 목표는 바로 특정 문제에 대한 최적의 해결 방법을 찾기보다는 특정한 문제에 대해서 충분히 이해할 수 있고 다양한 방법을 통하여 해결하는 방법을 찾고 향상시키는데 그 목적이 있습니다.

좀 더 좋은 알고리즘 및 코드가 있으시다면 언제든지 본 게시물의 댓글을 통해서 제시해주시면 감사하겠습니다.

 

 

 

---

 

 

 

코딩 테스트 문제

 

먼저 오늘 살펴볼 문제에 대해서 먼저 살펴보겠습니다.

 

programmers.co.kr/learn/courses/30/lessons/42748

코딩테스트 연습 - 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입니다.

 

 

 

 

 

 

 

 

 

---

 

 

 

 

입출력 예

array	commands	return
[1, 5, 2, 6, 3, 7, 4]	[[2, 5, 3], [4, 4, 1], [1, 7, 3]]	[5, 6, 3]

 

입출력 예 설명

 

[1, 5, 2, 6, 3, 7, 4]를 2번째부터 5번째까지 자른 후 정렬합니다. [2, 3, 5, 6]의 세 번째 숫자는 5입니다.

[1, 5, 2, 6, 3, 7, 4]를 4번째부터 4번째까지 자른 후 정렬합니다. [6]의 첫 번째 숫자는 6입니다.
[1, 5, 2, 6, 3, 7, 4]를 1번째부터 7번째까지 자릅니다. [1, 2, 3, 4, 5, 6, 7]의 세 번째 숫자는 3입니다.

 

 

 

 

---

 

 

문제 분석

 

문제는 array 라는 배열이 주어졌을 때 commands 라는 배열 0번째 값부터 1번째 값까지의 array 배열을 내림차순으로 정렬된 배열로 변환합니다.

 

그리고 해당 정렬된 배열에서 특정한 인덱스(command 배열에서 2번째 값)의 값만을 갖는 배열로 다시 한번 변환하는 것이 최종 원하는 답입니다.

 

 

 

 

---

 

 

 

알고리즘

 

위의 내용을 기반으로 하여 본 문제를 해결하기 위한 알고리즘을 살펴보면 아래와 같습니다.

 

• 최종 정답을 반활할 정답 배열을 하나 만듭니다.
• 입력 배열(array)에서 원하는 인덱스의 범위를 갖는 배열을 가져옵니댜.(시작 인덱스 : command[0], 끝 인덱스 : command[1])
• 추출된 배열을 내림차순으로 정렬합니다.
• 정렬된 배열에서 특정 인덱스(command[2])의 값을 가지는 배열로 다시 추출합니다.
• 최종 추출된 배열을 반환합니다.

 

 

 

 

---

 

 

 

코드 설명

 

그렇다면 위의 알고리즘에 대해서 하나씩 살펴보면서 코드로 작성을 해보도록 하겠습니다.

 

 

• 최종 정답을 반환할 정답 배열을 하나 만듭니다.
• 입력 배열(array)에서 원하는 인덱스의 범위를 갖는 배열을 가져옵니댜.(시작 인덱스 : command[0], 끝 인덱스 : command[1])

 

 

 

 

var answers = [Int]()

for n in 0...commands.count-1{
  // 시작 인덱스
  let i = commands[n][0]

  // 종료 인덱스
  let j = commands[n][1]

  var list = [Int]()

  for number in i-1...j-1{
  // 인덱스 범위의 수를 리스트 배열에 추가
  list.append(array[number])
  }
   
  ... 
}

 

 

• 추출된 배열을 내림차순으로 정렬합니다.

 

 

 

// 내림차순 정렬
list.sort(by: <)

 

 

• 정렬된 배열에서 특정 인덱스(command[2])의 값을 가지는 배열로 다시 추출합니다.

 

 

// 원하는 인덱스의 값을 추출
let k = commands[n][2]

// 정답 배열에 최종 결과값 추가
answers.append(list[k-1])

 

 

• 최종 추출된 배열을 반환합니다.

 

 

// 최종 정답 배열 반환
return answers

 

 

 

위의 코드들을 모두 조합하여 최종 코드를 완성하면 아래와 같습니다.

 

 

최종 코드

import Foundation

func solution(_ array:[Int], _ commands:[[Int]]) -> [Int] {
    var answers = [Int]()
        
    for n in 0...commands.count-1{
        // 시작 인덱스
        let i = commands[n][0]
        
        // 종료 인덱스
        let j = commands[n][1]
        
        var list = [Int]()
        
        for number in i-1...j-1{
            // 인덱스 범위의 수를 리스트 배열에 추가
            list.append(array[number])
        }
        
        // 내림차순 정렬
        list.sort(by: <)
        
        // 원하는 인덱스의 값을 추출
        let k = commands[n][2]
        
        // 정답 배열에 최종 결과값 추가
        answers.append(list[k-1])
    }
    
    // 최종 정답 배열 반환
    return answers
 }

 

위의 코드들을 Swift의 고차함수를 통하여 아래와 같이 코드 라인을 줄일 수도 있습니다.

아래와 같이 고차함수 map을 활용하여 for 문 및 불필요한 변수 선없없이 매우 깔끔한 형태로 변경이 가능합니다.

import Foundation

func solution(_ array:[Int], _ commands:[[Int]]) -> [Int] {
    var answers = [Int]()
{
    // array 배열에서 commands 0, 1 번째의 인덱스의 배열로 추출 
    // : commands.map( { array[($0[0]-1) ... ($0[1]-1)] } )
    // 추출된 배열을 내림차순으로 정렬
    // : array[($0[0]-1) ... ($0[1]-1)].sorted()
    // 정렬된 배열에서 command 2번째 값을 인덱스를 갖는 배열만 추출
    // : [$0[2]-1]
    let answers = commands.map( { array[($0[0]-1) ... ($0[1]-1)].sorted()[$0[2]-1] })
    
    return answers
 }

 

 

 

 

---

 

 

 

이상으로 오늘 제시한 문제에 대해서 분석 및 코드를 작성해 보았습니다.

 

 

 

감사합니다.

 

 

 

 

---

[참고 자료(References)]

 

[1] 프로그래머스 - K번째수 : programmers.co.kr/learn/courses/30/lessons/42862programmers.co.kr/learn/courses/30/lessons/42748