14891 톱니바퀴

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

K번 톱니바퀴를 회전시킨 후 점수의 합을 구하는 문제이다. 매번 톱니바퀴를 돌리면 서로 맞닿은 극이 다를 경우에만 회전하게 된다. 현재 상태를 기준으로 다음 회전할 방향을 미리 저장한 후 move함수를 이용하여 상태를 변화시킨다.

#include <cstdio>
#include <cstring>
 
// w[i][j]: i번째 톱니바퀴의 j번째 점수
// nd[i]: i번째 톱니바퀴의 다음 이동 방향
int w[5][9],nd[5],k,n,d;
 
int sum()
{
    int s=0;
    for(int i=1,j=1;i<=4;i++,j*=2)
        if(w[i][1]) s+=j;
    return s;
}
 
void move()
{
    for(int i=1;i<=4;i++)
    {
        if(nd[i]==1)
        {
            int t=w[i][8];
            for(int j=8;j>=2;j--)
                w[i][j]=w[i][j-1];
            w[i][1]=t;
        }
        else if(nd[i]==-1)
        {
            int t=w[i][1];
            for(int j=1;j<=7;j++)
                w[i][j]=w[i][j+1];
            w[i][8]=t;
        }
    }
}
 
int rd(int d)
{
    if(d==1) return -1;
    else return 1;
}
 
int main()
{
    for(int i=1;i<=4;i++) for(int j=1;j<=8;j++)
        scanf("%1d",&w[i][j]);
    scanf("%d",&k);
    while(k--)
    {
        memset(nd,0,sizeof(nd));
        scanf("%d %d",&n,&d);
        nd[n]=d;
        if(n==1)
        {
            for(int i=2;i<=4;i++)
            {
                if(w[i][7]!=w[i-1][3]) nd[i]=rd(nd[i-1]);
                else break;
            }
        }
        else if(n==2)
        {
            if(w[1][3]!=w[2][7]) nd[1]=rd(nd[2]);
            for(int i=3;i<=4;i++)
            {
                if(w[i][7]!=w[i-1][3]) nd[i]=rd(nd[i-1]);
                else break;
            }
        }
        else if(n==3)
        {
            if(w[3][3]!=w[4][7]) nd[4]=rd(nd[3]);
            for(int i=2;i>=1;i--)
            {
                if(w[i][3]!=w[i+1][7]) nd[i]=rd(nd[i+1]);
                else break;
            }
        }
        else if(n==4)
        {
            for(int i=3;i>=1;i--)
            {
                if(w[i][3]!=w[i+1][7]) nd[i]=rd(nd[i+1]);
                else break;
            }
        }
        move();
    }
    printf("%d\n",sum());
    return 0;
}