[SWEA_모의SW역량테스트] 4013번 - 특이한 자석

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AWIeV9sKkcoDFAVH

SW Expert Academy

SWEA 모의역량테스트 문제다.

이 문제는 기존 기출문제였던 백준 14891번(https://cpp-dev.tistory.com/31?category=852516)과 동일한 문제다.

다시금 연습한다는 자세로 풀어보았다.

역시 높은 정답률답게 한번에 Pass.

 

#include <iostream>
#include <vector>
#include <algorithm>
 
using namespace std;
 
int T, K;
int topnis[4][8];
 
int getanswer(){
    int sum = 0;
    for (int r = 0; r < 4; r++){
        if (topnis[r][0] == 1){
            if (r == 0) sum += 1;
            else if (r == 1) sum += 2;
            else if (r == 2) sum += 4;
            else if (r == 3) sum += 8;
        }
    }
    return sum;
}
 
void Rotate(int which, int dir){
    // 무엇무엇 돌릴지 여기서 계산하고 rotate에서 돌린다.
    vector<pair<int, int> > rot_info;
    rot_info.push_back(make_pair(which, dir));
    // 왼쪽
    int curdir = dir;
    for (int cur = which; cur >= 1; cur--){
        if (topnis[cur][6] != topnis[cur - 1][2]){
            curdir *= -1;
            rot_info.push_back(make_pair(cur - 1, curdir));
        }
        else break;
    }
 
    curdir = dir;
    // 오른쪽
    for (int cur = which; cur <= 2; cur++){
        if (topnis[cur][2] != topnis[cur + 1][6]){
            curdir *= -1;
            rot_info.push_back(make_pair(cur + 1, curdir));
        }
        else break;
    }
    
    for (int i = 0; i < rot_info.size(); i++){
        int cur_which = rot_info[i].first;
        int cur_dir = rot_info[i].second;
        int temp[8];
        // 시계
        if (cur_dir == 1){
            temp[0] = topnis[cur_which][7];
            for (int c = 1; c <= 7; c++) temp[c] = topnis[cur_which][c - 1];
            for (int c = 0; c < 8; c++) topnis[cur_which][c] = temp[c];
        }
        // 반시계
        else if (cur_dir == -1){
            temp[7] = topnis[cur_which][0];
            for (int c = 0; c < 7; c++) temp[c] = topnis[cur_which][c + 1];
            for (int c = 0; c < 8; c++) topnis[cur_which][c] = temp[c];
        }
    }
}
 
int main(){
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    cin >> T;
    // 1은 시계방향, -1은 반 시계방향. 
 
    for (int tc = 1; tc <= T; tc++){
        cin >> K; // 회전 수.
        for (int r = 0; r < 4; r++){
            for (int c = 0; c < 8; c++){
                cin >> topnis[r][c];
            }
        }
        for (int k = 0; k < K; k++){
            int which, dir;
            cin >> which >> dir;
            Rotate(which-1, dir);
        }
        cout << "#" << tc << " " << getanswer() << '\n';
    }
    return 0;
}