16939번 2×2×2 큐브

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

https://codeforces.com/contest/887/problem/C

처음에 bool check() 함수 내에서 return false; 하기 전에 init()을 안 해줘서 틀렸었다.

#include <iostream>
#include <queue>
#include <stack>
#include <cstdio>
#include <vector>
#include <cstring>
#include <string>
#include <math.h>
#include <algorithm>
#include <map>
#include <set>
#include <sstream>
#include <tuple>
 
#pragma warning(disable:4996)  
#pragma comment(linker, "/STACK:336777216")
 
using namespace std;
 
int N = 24, cube[25] = { 0, }, arr[25] = { 0, };
 
int cnt = 0;
 
void init()
{
    for (int i = 1; i <= N; i++) cube[i] = arr[i];
}
 
bool check()
{    
    int i = 1;
 
    // init()를 실행하지 않고, return false; 하는 경우 발생 할 수 있음. 물론 이 코드는 고친 후이다.
 
    for (i = 1; i <= 21; i += 4)
    {
        if (cube[i] != cube[i + 1])
        {
            init();
 
            return false;
        }
 
        if (cube[i + 1] != cube[i + 2])
        {
            init();
 
            return false;
        }
 
        if (cube[i + 2] != cube[i + 3])
        {
            init();
 
            return false;
        }
 
        if (cube[i + 3] != cube[i])
        {
            init();
 
            return false;
        }
    }
 
    init();
 
    return true;
}
 
bool ru()
{
    int temp = cube[6];
    cube[6] = cube[2];
    cube[2] = cube[23];
    cube[23] = cube[12];
    cube[12] = temp;
 
    temp = cube[8];
    cube[8] = cube[4];
    cube[4] = cube[21];
    cube[21] = cube[10];
    cube[10] = temp;
 
    return check();
}
 
bool rd()
{
    int temp = cube[6];
    cube[6] = cube[12];
    cube[12] = cube[23];
    cube[23] = cube[2];
    cube[2] = temp;
 
    temp = cube[8];
    cube[8] = cube[10];
    cube[10] = cube[21];
    cube[21] = cube[4];
    cube[4] = temp;
 
    return check();
}
 
bool lu()
{
    int temp = cube[5];
    cube[5] = cube[1];
    cube[1] = cube[24];
    cube[24] = cube[9];
    cube[9] = temp;
 
    temp = cube[7];
    cube[7] = cube[3];
    cube[3] = cube[22];
    cube[22] = cube[11];
    cube[11] = temp;
 
    return check();
}
 
bool ld()
{
    int temp = cube[5];
    cube[5] = cube[9];
    cube[9] = cube[24];
    cube[24] = cube[1];
    cube[1] = temp;
 
    temp = cube[7];
    cube[7] = cube[11];
    cube[11] = cube[22];
    cube[22] = cube[3];
    cube[3] = temp;
 
    return check();
}
 
bool ur()
{
    int temp = cube[18];
    cube[18] = cube[1];
    cube[1] = cube[15];
    cube[15] = cube[12];
    cube[12] = temp;
 
    temp = cube[20];
    cube[20] = cube[2];
    cube[2] = cube[13];
    cube[13] = cube[11];
    cube[11] = temp;
 
    return check();
}
 
bool ul()
{
    int temp = cube[20];
    cube[20] = cube[11];
    cube[11] = cube[13];
    cube[13] = cube[2];
    cube[2] = temp;
 
    temp = cube[18];
    cube[18] = cube[12];
    cube[12] = cube[15];
    cube[15] = cube[1];
    cube[1] = temp;
 
    return check();
}
 
bool br()
{
    int temp = cube[19];
    cube[19] = cube[4];
    cube[4] = cube[14];
    cube[14] = cube[9];
    cube[9] = temp;
 
    temp = cube[17];
    cube[17] = cube[3];
    cube[3] = cube[16];
    cube[16] = cube[10];
    cube[10] = temp;
 
    return check();
}
 
bool bl()
{
    int temp = cube[19];
    cube[19] = cube[9];
    cube[9] = cube[14];
    cube[14] = cube[4];
    cube[4] = temp;
 
    temp = cube[17];
    cube[17] = cube[10];
    cube[10] = cube[16];
    cube[16] = cube[3];
    cube[3] = temp;
 
    return check();
}
 
bool fr()
{
    int temp = cube[20];
    cube[20] = cube[24];
    cube[24] = cube[16];
    cube[16] = cube[8];
    cube[8] = temp;
 
    temp = cube[19];
    cube[19] = cube[23];
    cube[23] = cube[15];
    cube[15] = cube[7];
    cube[7] = temp;
 
    return check();
}
 
bool fl()
{
    int temp = cube[20];
    cube[20] = cube[8];
    cube[8] = cube[16];
    cube[16] = cube[24];
    cube[24] = temp;
 
    temp = cube[19];
    cube[19] = cube[7];
    cube[7] = cube[15];
    cube[15] = cube[23];
    cube[23] = temp;
 
    return check();
}
 
bool rr()
{
    int temp = cube[18];
    cube[18] = cube[22];
    cube[22] = cube[14];
    cube[14] = cube[6];
    cube[6] = temp;
 
    temp = cube[17];
    cube[17] = cube[21];
    cube[21] = cube[13];
    cube[13] = cube[5];
    cube[5] = temp;
 
    return check();
}
 
bool rl()
{
    int temp = cube[18];
    cube[18] = cube[6];
    cube[6] = cube[14];
    cube[14] = cube[22];
    cube[22] = temp;
 
    temp = cube[17];
    cube[17] = cube[5];
    cube[5] = cube[13];
    cube[13] = cube[21];
    cube[21] = temp;
 
    return check();
}
 
int main(void)
{
    for (int i = 1; i <= 24; i++)
    {
        scanf("%d", &cube[i]);
 
        arr[i] = cube[i];
    }
 
    if (ru() || rd() || lu() || ld() || ur() || ul() || br() || bl() || fr() || fl() || rr() || rl())
    {
        printf("1\n");
 
        return 0;
    }
 
    printf("0\n");
 
    return 0;
}