34-gjsk132

[gjsk132]

🔗 문제 링크

백준 1041 : 주사위

✔️ 소요된 시간

30분

✨ 수도 코드

🎯 문제 이해

문제 요약
아래 이미지와 같이 주사위의 6면에 정해진 숫자가 있다.

아래와 같이 주사위를 여러 개 쌓아준다.

이때, 표면에 있는 숫자의 합이 최소가 되는 값

N이 주어질 때, 꼭짓점, 모서리, 면에 오는 값이 정해져 있다.

3면을 보이는 주사위는 항상 4개
2면 이상 보이는 주사위는 모서리에 위치한 (모서리길이 -1) * 8
1면 이상 보이는 주사위는 표면에 나와있는 주사위 수와 같다.
( 보이는 모든 면에서, 2면 이상 보이는 주사위 수와 3면 이상 보이는 주사위의 수를 빼면 된다. )

주사위는 최대 3면까지 보이므로 표면에 보일 3개의 숫자를 정해야한다.

여기서 주의할 점은 마주보는 면에서 1개씩 골라야 한다.
주사위를 놓으면, 마주보는 면 중에서 한 면은 못 보기 때문이다.

이를 구하기 위해 마주보는 면끼리 묶으면서 오름차순 정렬하고, 묶음을 작은 수는 작을수록, 만약 작은 수가 같다면 큰 수가 클수록 먼저 사용하도록 정렬을 해준다. ( 큰 수를 가지고 있는 쌍을 먼저 사용하기 위해서이다. )

또 하나 더 유의할 점은, 1x1x1을 만들 때이다. 쉽게 말해 주사위가 1개 있을 때에는 가장 큰 수를 제외한 나머지 숫자를 합해야 한다.

🔍코드 설명

size : 주사위를 쌓을 크기
dice_nums : 주사위의 각 면에 쓰일 숫자
find_dice_sum() : 주사위를 1보다 많이 쌓을 때, 표면 숫자의 최소 합을 구해주는 함수
dice_sets : 주사위를 마주보는 면끼리 묶어 정렬하는 리스트
vertex : 꼭짓점 면의 수
corner : 모서리 면의 수
surface : 표면의 면의 수

우선 주어진 입력값을 받아준다.

size = int(input())
dice_nums = list(map(int,input().split()))

만약 size가 1이라면, 가장 큰 수를 제외한 표면 수의 총합을 출력하고,
1보다 큰 수라면, find_dice_set()함수를 실행한다.

if size == 1:
    print(sum(dice_nums)-max(dice_nums))
else:
    print(find_dice_sum())

find_dice_set()함수에서는 일단 마주보는 면끼리 정렬하여 묶고,
마주보는 면에서 작은 수는 작을수록, 큰 수는 클수록 앞에 오도록 정렬한다.
( 마주보는 면의 인덱스의 합이 2로 균일하여 아래와 같이 코드를 작성 )

def find_dice_sum():
    dice_sets = [sorted([dice_nums[2-i], dice_nums[3+i]])for i in range(3)]
    dice_sets = sorted(dice_sets, key=lambda x : x[1], reverse = True) #reverse는 반대로 정렬한 것이다.
    dice_sets = sorted(dice_sets, key=lambda x : x[0])

size를 통하여 vertex, cormer, surface에 해당하는 면이 몇 개인지 정의한다.

    vertex = 4
    corner = (size-1)*8
    surface = pow(size, 2)*5 - corner - vertex 

가장 많은 surface, cormer, vertex 순으로 가장 작은 값을 곱해서 총합을 구한뒤 return해준다.

    return dice_sets[0][0]*surface + dice_sets[1][0]*corner + dice_sets[2][0]*vertex

전체 코드

input = open(0).readline

size = int(input())
dice_nums = list(map(int,input().split()))

def find_dice_sum():
    dice_sets = [sorted([dice_nums[2-i], dice_nums[3+i]])for i in range(3)]
    dice_sets = sorted(dice_sets, key=lambda x : x[1], reverse = True)
    dice_sets = sorted(dice_sets, key=lambda x : x[0])

    vertex = 4
    corner = (size-1)*8
    surface = pow(size, 2)*5 - corner - vertex 

    return dice_sets[0][0]*surface + dice_sets[1][0]*corner + dice_sets[2][0]*vertex

if size == 1:
    print(sum(dice_nums)-max(dice_nums))
else:
    print(find_dice_sum())

📚 새롭게 알게된 내용

[9kyo-hwang]

이게 대체 뭔 문제지.... 하고 한참 생각해다가 PR 보고 겨우 이해했네요 ㅋㅋ
결국 정육면체를 만들면

• 한 면이 보이는 주사위
• 두 면이 보이는 주사위
• 세 면이 보이는 주사위

이렇게 3종류가 존재하므로, 세 면이 보이는 주사위의 (개수 * 최소 눈 조합) + 두 면이 보이는 주사위의 (개수 * 최소 눈 조합) + 한 면이 보이는 주사위의 (개수 * 최소 눈 조합)을 찾는 거였네요...

그리고 두 면/세 면이 보이는 주사위의 경우 "마주보는 면"에 대한 예외 처리를 잡아주면 되는 거였구요.

뭔가 되게 수학스러운데 그리디 관점이 접목된 느낌이네요...

#include <iostream>
#include <vector>
#include <algorithm>
#include <numeric>

using namespace std;

int main()
{
    cin.tie(nullptr)->sync_with_stdio(false);
    
    int64_t N; cin >> N;
    vector<int> Dice(6, 0);
    for(int& Pip : Dice)
    {
        cin >> Pip;
    }
    
    int SumPip = reduce(Dice.begin(), Dice.end(), 0);
    int MaxPip = *max_element(Dice.begin(), Dice.end());

    if(N == 1)
    {
        cout << SumPip - MaxPip;
        return 0;
    }
    
    int Min1D = *min_element(Dice.begin(), Dice.end()), Min2D = 101, Min3D = 151;
    for(int i = 0; i < 6; ++i)
    {
        for(int j = i + 1; j < 6; ++j)
        {
            if(i + j == 5)  // 마주보는 면은 두 면은 동시에 보일 수 없음
            {
                continue;
            }
            
            Min2D = min(Min2D, Dice[i] + Dice[j]);
            
            for(int k = j + 1; k < 6; ++k)
            {
                if(i + k == 5 || j + k == 5)  // 마주보는 면은 두 면은 동시에 보일 수 없음
                {
                    continue;
                }
                
                Min3D = min(Min3D, Dice[i] + Dice[j] + Dice[k]);
            }
        }
    }
    
    int64_t Answer = 0;
    Answer += 4 * Min3D;
    Answer += (8 * N - 12) * Min2D;
    Answer += (5 * N * N - 16 * N + 12) * Min1D;
    cout << Answer;
    
    return 0;
}

[xxubin04]

제가 엄청 복잡하게 풀려고 했던 것 같네요...
저는 3면, 2면, 1면이 보이는 주사위 개수를 구하는 공식을 도출해서 풀려고 했는데, 뭔가 중간에 오류가 있는 것 같습니다..
나중에 고칠 수 있지 않을까 하여 계속 미루다가 이제서야 리뷰하게 되었네용..🧎🏻‍♀️

참고사진

틀렸지만 혹시나 하여 올리는 코드..!

input = open(0).readline

n = int(input())
num = list(map(int, input().split()))
dice = []

for i in enumerate(num):
    dice.append(i)

select = []
if dice[2][1] > dice[3][1]:
    select.append(dice.pop(3)[1])
    dice.pop(2)
else:
    select.append(dice.pop(2)[1])
    dice.pop(3)
dice.sort(key=lambda x: x[1])
select.append((a := dice.pop(0))[1])
for j in range(3):
    if a[0] == 1 and dice[j][0] != 4:
        select.append(dice[j][1])
        break
    elif a[0] == 4 and dice[j][0] != 1:
        select.append(dice[j][1])
        break
    elif a[0] == 0 and dice[j][0] != 5:
        select.append(dice[j][1])
        break
    elif a[0] == 5 and dice[j][0] != 0:
        select.append(dice[j][1])
        break

select.sort()
min_1 = select[0]
min_2 = select[1]
min_3 = select[2]
ans = 8 * (min_1 + min_2 + min_3) + 12 * (n - 2) * (min_1 + min_2) + 6 * (n - 2) ** 2 * min_1
ans -= (min_1 * (n - 2) ** 2 + min_2 * 4 * (n - 2) + min_3 * 4)
print(ans)

테케들은 잘 통과하지만, min_1, min_2, min_3으로 구하지 않고 마주보는 면을 기준으로 풀도록 해봐야겠습니다!!