백준 1930번 정사면체 C++ (골드4)

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

 

a,b,c,d순으로, 역삼각형 모양의 전개도로 입력이 들어왔다고 가정하면

a가 밑면일때 b,c,d가 시계방향으로 배치되어있고

b가 밑면일때 a,d,c

c가 밑면일때 d,a,b

d가 밑면일때 c,b,a

로 배치되어있다. 위의 경우들은 전부 같은 모양이다.

위의 경우들을 저장해놓는다.

 

뒤에 들어오는 4개의 입력중 첫 입력은 밑면이고, 위에서 저장해놓은 밑면들과 비교하여 만약 밑면이 같다면 나머지 3개의 입력을 비교하여 같은지를 판별한다. 물론 (x,y,z), (y,z,x), (z,x,y)는 순서만 다를뿐 회전시키면 같은 모양이기 때문에 이 부분도 고려하여 비교하였다.

 

#include <iostream>
#include <string>
#include <vector>
#include <math.h>
#include <algorithm>
#include <utility>
#include <stack>
#include <queue>
#include <math.h>
#include <set>
#include <map>
#include <list>
#include <unordered_map>
#include <unordered_set>
#include <iomanip>
#include <limits.h>

using namespace std;
using int64 = long long;

void Init(int a, int b, int c, int d, vector<pair<int, vector<int>>>& pv)
{
    vector<int> v1;
    v1.push_back(b);
    v1.push_back(c);
    v1.push_back(d);

    vector<int> v2;
    v2.push_back(a);
    v2.push_back(d);
    v2.push_back(c);

    vector<int> v3;
    v3.push_back(d);
    v3.push_back(a);
    v3.push_back(b);

    vector<int> v4;
    v4.push_back(c);
    v4.push_back(b);
    v4.push_back(a);

    pv.push_back(make_pair(a, v1));
    pv.push_back(make_pair(b, v2));
    pv.push_back(make_pair(c, v3));
    pv.push_back(make_pair(d, v4));
}

bool CompareSame(vector<int>& v, vector<int>& comp)
{
    bool ret = false;
    for (int i = 0; i < comp.size(); i++)
    {
        bool tmp = true;
        int addIdx = i;
        for (int j = 0; j < v.size(); j++)
        {
            if (v[j] != comp[(j + addIdx) % comp.size()]) tmp = false;
        }
        ret |= tmp;
    }
    return ret;
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    int k;
    cin >> k;
    while (k--)
    {
        int a, b, c, d;
        vector<pair<int, vector<int>>> pv;
        cin >> a >> b >> c >> d;
        Init(a, b, c, d, pv);

        int ans = 0;

        cin >> a >> b >> c >> d;

        vector<int> comp{ b,c,d };
        for (auto am : pv)
        {
            if (am.first != a) continue;
            vector<int>& v = am.second;
            bool res = CompareSame(v, comp);
            if (res) ans = 1;
        }
        cout << ans << "\n";
    }
}