[ 백준 15662 ] 톱니바퀴(2) (C++)

백준의 톱니바퀴2 (15662) 문제이다.

[ 문제 바로가기 ]

[ 문제풀이 ]

1) 톱니바퀴1과 전체적인 문제는 똑같고 톱니바퀴의 수만 더 증가한 문제이다.

   [ 톱니바퀴1 문제 풀이 바로가기(Click) ]

   본인은 전체적인 로직 또한 톱니바퀴1과 굉장히 비슷하게 풀었다.

   단지, 톱니바퀴1에서는 바퀴가 4개뿐이라서, 4개를 하나하나 다 체크해가면서 톱니바퀴를 돌려준 반면, 이 문제에서는

   for문으로 돌려야 하는 톱니바퀴를 기준으로 왼쪽과 오른쪽으로 나눠서 돌려야 할 톱니바퀴들을

   따로 저장 후 실제로 돌려주는 과정만 더 추가해 주었다.

  

[ 소스코드 ]

#include<iostream>
#include<vector>
#include<string>
 
#define endl "\n"
#define MAX 1010
using namespace std;
 
int T, K;
vector<int> V[MAX];
vector<pair<int, int>> Cmd;
 
void Input()
{
    cin >> T;
    for (int i = 1; i <= T; i++)
    {
        string S; cin >> S;
        for (int j = 0; j < S.length(); j++)
        {
            V[i].push_back(S[j] - '0');
        }
    }
    cin >> K;
    for (int i = 0; i < K; i++)
    {
        int a, b; cin >> a >> b;
        Cmd.push_back(make_pair(a, b));
    }
}
 
int Reverse(int Dir)
{
    if (Dir == 1) return -1;
    else return 1;
}
 
vector<pair<int,int>> Find_Turn_Wheel(int Idx, int Dir)
{
    vector<pair<int, int>> R;
    R.push_back(make_pair(Idx, Dir));
    int Side1 = V[Idx].at(2);
    int Side2;
    int Cur_Dir = Dir;
 
    for (int i = Idx + 1; i <= T; i++)
    {
        Side2 = V[i].at(6);
        if (Side1 != Side2)
        {
            R.push_back(make_pair(i, Reverse(Cur_Dir)));
            Side1 = V[i].at(2);
            Cur_Dir = Reverse(Cur_Dir);
        }
        else break;
    }
 
    Side1 = V[Idx].at(6);
    Side2 = -1;
    Cur_Dir = Dir;
    for (int i = Idx - 1; i >= 1; i--)
    {
        Side2 = V[i].at(2);
        if (Side1 != Side2)
        {
            R.push_back(make_pair(i, Reverse(Cur_Dir)));
            Side1 = V[i].at(6);
            Cur_Dir = Reverse(Cur_Dir);
        }
        else break;
    }
    
    return R;
}
 
void Turning(int Idx, int Dir)
{
    if (Dir == 1)
    {
        int Temp = V[Idx].at(7);
        for (int i = 6; i >= 0; i--)
        {
            V[Idx].at(i + 1) = V[Idx].at(i);
        }
        V[Idx].at(0) = Temp;
    }
    else
    {
        int Temp = V[Idx].at(0);
        for (int i = 1; i <= 7; i++)
        {
            V[Idx].at(i - 1) = V[Idx].at(i);
        }
        V[Idx].at(7) = Temp;
    }
}
 
void Actual_Turning(vector<pair<int, int>> Wheel)
{
    for (int i = 0; i < Wheel.size(); i++)
    {
        int Idx = Wheel[i].first;
        int Dir = Wheel[i].second;
        Turning(Idx, Dir);
    }
}
 
void Solution()
{
    for (int i = 0; i < K; i++)
    {
        int W_Idx = Cmd[i].first;
        int Dir = Cmd[i].second;
 
        vector<pair<int, int>> Turn_Wheel;
        Turn_Wheel = Find_Turn_Wheel(W_Idx, Dir);
        Actual_Turning(Turn_Wheel);
    }
 
    int Cnt = 0;
    for (int i = 1; i <= T; i++)
    {
        if (V[i].at(0) == 1) Cnt++;
    }
    cout << Cnt << endl;
}
 
void Solve()
{
    Input();
    Solution();
}
 
int main(void)
{
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
 
    //freopen("Input.txt", "r", stdin);
    Solve();
 
    return 0;
}