BOJ 1041 : 주사위

• 문제 링크 : boj.kr/1041
• 난이도 : G5
• 태그 : 그리디

---

코드

#include <bits/stdc++.h>

#define all(x) (x).begin(), (x).end()

#define INF 987654321
#define int long long

using namespace std;

using ld = long double;
using pii = pair<int,int>;

template <typename T1, typename T2>
istream& operator>>(istream & ist, pair<T1,T2> &p) {
    ist >> p.first >> p.second;
    return ist;
}
template <typename T>
istream& operator>>(istream & ist, vector<T> &arr) {
    for(auto &u : arr) ist >> u;
    return ist;
}

int32_t main()
{
    ios::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);

    int n;
    cin >> n;
    vector<int> arr(6);
    cin >> arr;

    int a = min(arr[0], arr[5]);
    int b = min(arr[1], arr[4]);
    int c = min(arr[2], arr[3]);

    if(n == 1) {
        int M = 0, sum = 0;
        for(int i = 0; i < 6; i++) {
            sum += arr[i];
            M = max(M, arr[i]);
        }
        cout << sum - M << "\n";
        return 0;
    }
    int ans = 0;
    ans += (a+b+c)*4;
    ans += (a+b+c-max({a,b,c}))*(8*n-12);
    ans += min({a,b,c})*((n-2)*(n-2)*5 + 4*(n-2));

    cout << ans;

    return 0;
}

---

풀이

$n \ge 2$일때, 3면이 보이는 주사위의 수는 $4$개, 2면이 보이는 주사위의 수는 $4*(n-1)+4*(n-2)$개, 1면이 보이는 주사위의 수는 $5*(n-2)^2+4*(n-2)$개이다.

같은 수의 면이 보이는 주사위는 보이는 수의 합이 같다. 각 면이 보일 때 수의 합의 최소를 구하는 것은 가능한 모든 경우의 수를 다 해보는 것도 방법이다. 하지만 3면 이하의 면이 보일 때 서로 반대에 있는 두 면이 동시에 보이지 않는다는 점을 이용하면 쉽게 구할 수 있다.

$n=1$인 경우에만 예외처리를 해주면 된다.