K번째 큰 수(Swift)

K번째 큰 수(Swift)

 

 

 문제 설명

 

현수는 1부터 100사이의 자연수가 적힌 N장의 카드를 가지고 있습니다. 같은 숫자의 카드가 여러장 있을 수 있습니다. 현수는 이 중 3장을 뽑아 각 카드에 적힌 수를 합한 값을 기록하려 고 합니다. 3장을 뽑을 수 있는 모든 경우를 기록합니다. 기록한 값 중 K번째로 큰 수를 출력 하는 프로그램을 작성하세요.

만약 큰 수부터 만들어진 수가 25 25 23 23 22 20 19......이고 K값이 3이라면 K번째 큰 값 은 22입니다.

 

▣ 입력설명
첫 줄에 자연수 N(3<=N<=100)과 K(1<=K<=50) 입력되고, 그 다음 줄에 N개의 카드값이 입력 된다.

 

▣ 출력설명
첫 줄에 K번째 수를 출력합니다. K번째 수는 반드시 존재합니다.

 

▣ 입력예제 1
10 3
13 15 34 23 45 65 33 11 26 42

 

▣ 출력예제

143

 

 

 나의 풀이

 

문제를 보자마자 조합을 떠올렸다.

하지만 "3장을 뽑아"라는 조건을 보자마자 간단한 for문을 돌리면 시간초과가 일어날까? 라는 생각부터 하는 게 좋았을 것 같다. 

 

문제의 예시 입력값에 대해서는 결과값이 맞지만 결과적으로 내가 한 풀이는 틀렸다. 

맨 마지막의 "25 25 23 23 22 20 19......이고 K값이 3이라면 K번째 큰 값 은 22입니다."은 중복을 제거한다는 뜻이기 때문이다. 

 

그래서 문제의 저자처럼 for문 + Set자료구조를 이용해서 다시 풀어보았다

import Foundation

func combination(_ targetArr: [Int], _ k: Int) -> [[Int]] {
  var results: [[Int]] = []
  func cycle(_ index: Int, _ arr: [Int]) {
    if arr.count == k {
      results.append(arr)
      return
    }
    for i in index..<targetArr.count {
      cycle(i + 1, arr + [targetArr[i]])
    }
  }

  cycle(0, [])
  return results
}

let input = readLine()!
let results = input.components(separatedBy: " ").map{ Int($0)! }
let arrInput = readLine()!
let arrResult = arrInput.components(separatedBy: " ").map{ Int($0)! }


print(combination(arrResult, 3).map{ $0.reduce(0, +) }.sorted(by: >)[results[1]-1])

 

 

 새로운 풀이

 

import Foundation

let input = readLine()!
let results = input.components(separatedBy: " ").map{ Int($0)! }
let arrInput = readLine()!
let arrResult = arrInput.components(separatedBy: " ").map{ Int($0)! }

var answers:Set<Int> = []
print(arrResult)
for i in 0..<arrResult.count {
  for j in i+1..<arrResult.count {
    for z in j+1..<arrResult.count {
      answers.insert(arrResult[i] + arrResult[j] + arrResult[z])
    }
  }
}
print(answers.map{$0}.sorted(by: >)[results[1]-1])

 

 피드백

중복 제거에는 Set이 최고인듯 하다  ✨