程序设计思维与实践 Week10限时模拟 东东的魔方

问题描述

东东有一个二阶魔方，即2×2×2的一个立方体组。立方体由八个角组成。
魔方的每一块都用三维坐标(h, k, l)标记，其中h, k, l∈{0,1}。六个面的每一个都有四个小面，每个小面都有一个正整数。

对于每一步，东东可以选择一个特定的面，并把此面顺时针或逆时针转90度。

请你判断，是否东东可以在一个步骤还原这个魔方（每个面没有异色）。
input

** 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

想法

每次只能转一次，每次只能转90度，每次旋转可以看做是交换了八个点的值，这样按照点的对应关系我们就能写出每个面顺势针旋转90度所对应函数，每次旋转都要判断魔方是否复原，由judge函数实现，逆时针旋转90度相当于顺时针旋转270度，也就是转三次，当对一个面两种旋转都不能未复原就要再转一次恢复原状态换下一个面再判断。

代码

#include <algorithm>
#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
int a[7][5]; 
void upShift()
{
    int t1 = a[2][1], t2 = a[2][2];
    a[2][1] = a[5][2];
    a[2][2] = a[5][4];
    a[5][2] = a[4][4];
    a[5][4] = a[4][3];
    a[4][4] = a[6][3];
    a[4][3] = a[6][1];
    a[6][3] = t1;
    a[6][1] = t2;
}
void leftShift()
{
    int t1 = a[1][1], t2 = a[1][3];
    a[1][1] = a[4][1];
    a[1][3] = a[4][3];
    a[4][1] = a[3][1];
    a[4][3] = a[3][3];
    a[3][1] = a[2][1];
    a[3][3] = a[2][3];
    a[2][1] = t1;
    a[2][3] = t2;
}
void frontShift()
{
    int t1 = a[1][1], t2 = a[1][2];
    a[1][1] = a[5][1];
    a[1][2] = a[5][2];
    a[5][1] = a[3][4];
    a[5][2] = a[3][3];
    a[3][4] = a[6][1];
    a[3][3] = a[6][2];
    a[6][1] = t1;
    a[6][2] = t2;
}
bool Judge()
{
    for (int i = 1; i <= 6; i++)
        if (count(a[i] + 1, a[i] + 4 + 1, a[i][1]) != 4)
            return false;
    return true;
} 
int main()
{
    freopen("input.txt", "r", stdin);
    freopen("out.txt", "w", stdout);
    int N;
    cin >> N;
    while (N--)
    {
        for (int i = 1; i <= 6; i++)
            for (int j = 1; j <= 4; j++)
                cin >> a[i][j];
        if (Judge())
        {
            cout << "YES" << endl;
            continue;
        }

        upShift();
        if (Judge())
        {
            cout << "YES" << endl;
            continue;
        }
        upShift();
        upShift();
        if (Judge())
        {
            cout << "YES" << endl;
            continue;
        }
        upShift();

        leftShift();
        if (Judge())
        {
            cout << "YES" << endl;
            continue;
        }
        leftShift();
        leftShift();
        if (Judge())
        {
            cout << "YES" << endl;
            continue;
        }
        leftShift();

        frontShift();
        if (Judge())
        {
            cout << "YES" << endl;
            continue;
        }
        frontShift();
        frontShift();
        if (Judge())
        {
            cout << "YES" << endl;
            continue;
        }
        frontShift();
        cout << "NO" << endl;
    }
    fclose(stdin);
    fclose(stdout);
    return 0;
}