백준 - 주사위게임 2476번 (Swift)

백준 - 주사위게임 2476번 (Swift)

 

 

 문제 설명

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

2476번: 주사위 게임

 

 

 나의 풀이

 

몇 가지 테스트케이스를 실패하는 풀이이다. 아마도 문제의 조건의 최대값에서 filter와 map을 연속으로 돌려서 그런 것 같다.

import Foundation

func calculate(_ a: Int, _ b: Int, _ c: Int) -> Int {
  var ch = [Int](repeating: 0, count: 7)
  ch[a] += 1
  ch[b] += 1
  ch[c] += 1
  if ch.max()! == 3 {
    let val = ch.enumerated().filter{ $0.element == 3 }.map{ $0.offset }[0]
    return 10000+(val*1000)
  } else if ch.max()! == 2 {
    let val = ch.enumerated().filter{ $0.element == 2 }.map{ $0.offset }[0]
    return 1000+(val*100)
  } else {
    return ch.enumerated().sorted(by: >)[0].offset * 100
  }
}

let n = Int(readLine()!)!
var maxPrice = 0
for _ in 0..<n {
  let input = readLine()!
  let result = input.components(separatedBy: " ").map{Int($0)!}
  let price = calculate(result[0], result[1], result[2])
  print(price)
  maxPrice = max(maxPrice, price)
}

print(maxPrice)

 

다른 풀이

 

문제는 3개의 주사위만 돌리기 때문에 3개를 if문으로 쪼개서 풀었다.

이렇게 풀면 n번만큼만 로직을 돌리기 때문에 시간복잡도가 현저히 줄어든다.

import Foundation

let n = Int(readLine()!)!
var maxPrice = 0
for _ in 0..<n {
  let input = readLine()!
  let result = input.components(separatedBy: " ").map{Int($0)!}
  let a = result[0], b = result[1], c = result[2]
  var price = 0
  if a==b && b==c {
      price = 10000+(a*1000)
  } else if a==b || a==c {
      price = 1000+(a*100)
  } else if b==c {
      price = 1000+(b*100)
  } else {
      var maxVal = 0
      maxVal = max(a, b)
      maxVal = max(maxVal, c)
      price = maxVal * 100
  }
  maxPrice = max(maxPrice, price)
}

print(maxPrice)

 

 

 피드백

 

문제의 크기가 작다면 하드코딩으로 해결하는 게 때로는 좋다