程序设计思维与实践 Week10 限时大模拟 B-东东转魔方

题目链接：B-东东转魔方

题目描述：
东东有一个二阶魔方，即2×2×2的一个立方体组。立方体由八个角组成。
魔方的每一块都用三维坐标(h, k, l)标记，其中h, k, l∈{0,1}。六个面的每一个都有四个小面，每个小面都有一个正整数。
对于每一步，东东可以选择一个特定的面，并把此面顺时针或逆时针转90度。
请你判断，是否东东可以在一个步骤还原这个魔方（每个面没有异色）。

Input：
输入的第一行包含一个整数N（N≤30），这是测试用例的数量。
对于每个测试用例， 第 1~4 个数描述魔方的顶面，这是常见的2×2面，由（0，0，1），（0，1，1），（1，0，1），（1，1，1）标记。四个整数对应于上述部分。
第 5~8 个数描述前面，即（1，0，1），（1，1，1），（1，0，0），（1，1，0）的公共面。四个整数 与上述各部分相对应。
第 9~12 个数描述底面，即（1，0，0），（1，1，0），（0，0，0），（0，1，0）的公共面。四个整数与上述各部分相对应。
第 13~16 个数描述背面，即（0，0，0），（0，1，0），（0，0，1），（0，1），（0，1，1）的公共面。四个整数与上述各部分相对应。
第 17~20 个数描述左面，即（0，0，0），（0，0，1），（1，0，0），（1，0，1）的公共面。给出四个整数与上述各部分相对应。
第 21~24 个数描述了右面，即（0，1，1），（0，1，0），（1，1，1），（1，1，0）的公共面。给出四个整数与上述各部分相对应。
换句话说，每个测试用例包含24个整数a、b、c到x。你可以展开表面以获得平面图
如下所示。

Output：
对于每个测试用例，魔方如果可以至多 “只转一步” 恢复，输出YES，则输出NO。

Sample Input：
4
1 1 1 1 2 2 2 2 3 3 3 3 4 4 4 4 5 5 5 5 6 6 6 6
6 6 6 6 1 1 1 1 2 2 2 2 3 3 3 3 5 5 5 5 4 4 4 4
1 4 1 4 2 1 2 1 3 2 3 2 4 3 4 3 5 5 5 5 6 6 6 6
1 3 1 3 2 4 2 4 3 1 3 1 4 2 4 2 5 5 5 5 6 6 6 6

Sample Output：
YES
YES
YES
NO

Hint：
如果能思考一下解题框架的设计是最好的，一上来就莽很痛苦

思路：
对于一次旋转，六个面进行顺时针或者逆时针的情况共12种，但是例如上面顺时针转就和底面逆时针转相同，因此就只有六种旋转情况，对于每一种旋转情况要判断每个面数字的相同情况，然后得出是哪一种转法，如果不是六种转法中的一个就输出-1，还要特判一下处于还原状态的魔法。

总结：
设计解题框架，然后考虑六种转法，最后发现是很痛苦的在莽。

代码：

#include<iostream>
using namespace std;
int n,cube[20][10],jud[20][10];
bool judge(){
    bool flag=1;
    for(int i=0;i<6;i++)
        for (int j=1;j<=4;j++)
            if (jud[i][j]!=jud[i][1]){
                flag=0;
                break;
            }
    return flag;
}
int main(){
    cin>>n;
    while(n--){
        for(int i=0;i<6;i++)
            for (int j=1;j<=4;j++) {
                cin>>cube[i][j];
                jud[i][j]=cube[i][j];
            }
        if (judge()) {
            cout<<"YES"<<endl;
            continue;
        }
        if(cube[0][1]==cube[0][2]&&cube[0][2]==cube[4][4]&&cube[4][4]==cube[4][3]&&cube[4][2]==cube[4][1]&&cube[4][1]==cube[2][1]&&cube[2][1]==cube[2][2]&&cube[2][3]==cube[2][4]&&cube[2][4]==cube[5][3]&&cube[5][3]==cube[5][4]&&cube[5][1]==cube[5][2]&&cube[5][2]==cube[0][3]&&cube[0][3]==cube[0][4]&&cube[1][1]==cube[1][2]&&cube[1][2]==cube[1][3]&&cube[1][3]==cube[1][4]&&cube[3][1]==cube[3][2]&&cube[3][2]==cube[3][3]&&cube[3][3]==cube[3][4])
            cout<<"YES"<<endl;
        else if(cube[0][1]==cube[0][2]&&cube[0][2]==cube[5][4]&&cube[5][4]==cube[5][3]&&cube[5][2]==cube[5][1]&&cube[5][1]==cube[2][1]&&cube[2][1]==cube[2][2]&&cube[2][3]==cube[2][4]&&cube[2][4]==cube[4][3]&&cube[4][3]==cube[4][4]&&cube[4][1]==cube[4][2]&&cube[4][2]==cube[0][3]&&cube[0][3]==cube[0][4]&&cube[1][1]==cube[1][2]&&cube[1][2]==cube[1][3]&&cube[1][3]==cube[1][4]&&cube[3][1]==cube[3][2]&&cube[3][2]==cube[3][3]&&cube[3][3]==cube[3][4])
            cout<<"YES"<<endl;
        else if(cube[0][1]==cube[0][3]&&cube[0][3]==cube[1][2]&&cube[1][2]==cube[1][4]&&cube[1][1]==cube[1][3]&&cube[1][3]==cube[2][2]&&cube[2][2]==cube[2][4]&&cube[2][1]==cube[2][3]&&cube[2][3]==cube[3][2]&&cube[3][2]==cube[3][4]&&cube[3][1]==cube[3][3]&&cube[3][3]==cube[0][2]&&cube[0][2]==cube[0][4]&&cube[5][1]==cube[5][2]&&cube[5][2]==cube[5][3]&&cube[5][3]==cube[5][4]&&cube[4][1]==cube[4][2]&&cube[4][2]==cube[4][3]&&cube[4][3]==cube[4][4])
            cout<<"YES"<<endl;
        else if(cube[0][1]==cube[0][3]&&cube[0][3]==cube[3][2]&&cube[3][2]==cube[3][4]&&cube[3][1]==cube[3][3]&&cube[3][3]==cube[2][2]&&cube[2][2]==cube[2][4]&&cube[2][1]==cube[2][3]&&cube[2][3]==cube[1][2]&&cube[1][2]==cube[1][4]&&cube[1][1]==cube[1][3]&&cube[1][3]==cube[0][2]&&cube[0][2]==cube[0][4]&&cube[5][1]==cube[5][2]&&cube[5][2]==cube[5][3]&&cube[5][3]==cube[5][4]&&cube[4][1]==cube[4][2]&&cube[4][2]==cube[4][3]&&cube[4][3]==cube[4][4])
            cout<<"YES"<<endl;
        else if(cube[3][1]==cube[3][2]&&cube[3][2]==cube[5][1]&&cube[5][1]==cube[5][3]&&cube[5][2]==cube[5][4]&&cube[5][4]==cube[1][1]&&cube[1][1]==cube[1][2]&&cube[1][3]==cube[1][4]&&cube[1][4]==cube[4][2]&&cube[4][2]==cube[4][4]&&cube[4][1]==cube[4][3]&&cube[4][3]==cube[3][3]&&cube[3][3]==cube[3][4]&&cube[2][1]==cube[2][2]&&cube[2][2]==cube[2][3]&&cube[2][3]==cube[2][4]&&cube[0][1]==cube[0][2]&&cube[0][2]==cube[0][3]&&cube[0][3]==cube[0][4])
            cout<<"YES"<<endl;
        else if(cube[3][3]==cube[3][4]&&cube[3][4]==cube[5][2]&&cube[5][2]==cube[5][4]&&cube[5][3]==cube[5][1]&&cube[5][1]==cube[1][3]&&cube[1][3]==cube[1][4]&&cube[1][1]==cube[1][2]&&cube[1][2]==cube[4][3]&&cube[4][3]==cube[4][1]&&cube[4][4]==cube[4][2]&&cube[4][2]==cube[3][1]&&cube[3][1]==cube[3][2]&&cube[2][1]==cube[2][2]&&cube[2][2]==cube[2][3]&&cube[2][3]==cube[2][4]&&cube[0][1]==cube[0][2]&&cube[0][2]==cube[0][3]&&cube[0][3]==cube[0][4])
            cout<<"YES"<<endl;
        else
            cout<<"NO"<<endl;
    }
    return 0;
}