백준 - 주사위(Swift)

백준 - 주사위(Swift)

 

 

 문제 설명

 

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

1041번: 주사위

 

 나의 풀이

 

다른 사람의 풀이를 보기전에는 문제를 잘 이해하지도 못했다.

N*N*N개를 이용해서 N*N*N크기의 정육면체를 만들라고?

 

하지만 이는 조금만 생각하면 이해할 수 있었다. N이 2라면 8개로 8크기(1층에 1x1x1을 4칸, 2층에 1x1x1을 4칸)를 만들면 된다.

 

문제의 포커스는 보이는 면의 최소값이다. 그럼 보이는 면의 개수가 몇인지만 체크하면 된다.

 

N이 1일 때를 제외하면(1일때는 5면만 보임) 1면이 보일때, 2면이 보일때, 3면이 보일때로 나뉘어진다. 

즉 1면일때 2면일때 3면일때의 최소값을 구한뒤 N이 변함에따라 개수가 몇개인지를 구하고 곱해주면 된다.

 

여기서 중요한 것은 마주보는 면 중에 더 작은 값을 넣어야한다는 것이다.

A면은 F면과 마주보기 때문에 더 작은 값을 넣어줘야 최소값을 가지는 면이된다. 나머지도 마찬가지이다.

 

그리고 개수를 구하는데 공식을 만들어보면 (자세한 건 주석에 있습니다)

1면일때 (N-1)*(N-1) + 4*(N-1)*(N-2)가 되고

2면일때 4*(N-2) + 4*(N-1)이 되고

3면일때는 4가 된다.(맨 위의 4개 모서리)

 

그래서 이 개수와 최소값이 나오는 면의 수를 곱해주면 보이는 면의 최소 개수가 된다.

 

import Foundation

let N = Int(readLine()!)!
let dice = readLine()!.components(separatedBy:" ").map{Int($0)!}

if N == 1 {
    var dices = dice.sorted(by: <)
    dices.removeLast()
    print(dices.reduce(0, +))
    exit(0)
}


// 마주보는 면 중에 작은 값들을 삽입.
var min_dice = [Int]()
min_dice.append(min(dice[0], dice[5]))
min_dice.append(min(dice[1], dice[4]))
min_dice.append(min(dice[2], dice[3]))
min_dice.sort(by: <)

// 면의 1면, 2면, 3면일때의 최소값
let min_one = min_dice[0]
let min_two = min_dice[0] + min_dice[1]
let min_three = min_dice.reduce(0, +)

// 면이 1면, 2면, 3면 각각 보여지는 경우 주사위의 개수
let one_cnt = (N-2)*(N-2) + 4*(N-2)*(N-1) // 맨위의 개수 + 옆면에서의 개수
let two_cnt = 4*(N-2) + 4*(N-1) // 맨 위의 개수 + 옆면에서의 개수
let three_cnt = 4 // 맨 위 모서리 4개만


var sum = 0
sum += min_one * one_cnt
sum += min_two * two_cnt
sum += min_three * three_cnt

print(sum)