[백준] 17822 - 원판 돌리기

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

17822번: 원판 돌리기

회전 시의 인덱스 변화를 잘 신경쓴다면 어렵지 않은 구현 문제..

 

#include <cstdio>
using namespace std;
 
int n, m, t;
int arr[52][52], rot[52], check[52][52];
int dv[2] = {1, -1};
 
inline int idx(int i){
    int t = ((i % m) + m) % m; 
    if(t == 0) return m;
    else return t;
}
 
void near(){
    for(int i=1; i<=n; i++){
        for(int j=1; j<=m; j++){
            check[i][j] = 0;
        }
    }
    for(int i=1; i<=n; i++){
        // 좌우 근접
        for(int j=1; j<m; j++){
            if(arr[i][j] == arr[i][j+1]){
                check[i][j] = check[i][j+1] = 1;
            }
        }
        if(arr[i][1] == arr[i][m]) check[i][1] = check[i][m] = 1;
        
        // 상하 근접
        for(int j=1; j<=m; j++){
            if(arr[i][j] == arr[i+1][idx(j + rot[i] - rot[i+1])]){
                check[i][j] = check[i+1][idx(j + rot[i] - rot[i+1])] = 1;
            }
        }
    }
    
    bool change = false;
    for(int i=1; i<=n; i++){
        for(int j=1; j<=m; j++){
            if(check[i][j] == 1){
                if(arr[i][j] != 0) change = true;
                arr[i][j] = 0;
            }
        }
    }
 
    if(!change){
        int cnt = 0, s = 0;
        for(int i=1; i<=n; i++){
            for(int j=1; j<=m; j++){
                if(arr[i][j] != 0){
                    cnt++;
                    s += arr[i][j];
                }
            }
        }
        double aver = (double)s / cnt;
        for(int i=1; i<=n; i++){
            for(int j=1; j<=m; j++){
                if(arr[i][j] != 0 && arr[i][j] > aver) arr[i][j]--;
                else if(arr[i][j] != 0 && arr[i][j] < aver) arr[i][j]++;
            }
        }
    }
}
 
int main(){
    scanf("%d %d %d", &n, &m, &t);
    for(int i=1; i<=n; i++){
        for(int j=1; j<=m; j++){
            scanf("%d", &arr[i][j]);
        }
    }
    for(int v=0; v<t; v++){
        int x, d, k;
        scanf("%d %d %d", &x, &d, &k);
        for(int i=x; i<=n; i+=x){
            rot[i] += dv[d] * k;
        }
        near();
    }
 
    int re = 0;
    for(int i=1; i<=n; i++){
        for(int j=1; j<=m; j++){
            re += arr[i][j];
        }
    }
    printf("%d", re);
 
    return 0;
}