삼성 역량 기출문제[17822 원판 돌리기]

[youbeen2798]

문제 링크 : https://www.acmicpc.net/problem/17822

이거 2프로에서 틀렸다고 하는데 혹시 왜 틀렸는지 아시는분 있나요?!ㅜㅜ

게시글 반례도 다 맞고, 왜 틀렸는지 모르겠습니당..ㅜㅠ

#include <iostream>
#include <cstring>
#include <vector>
using namespace std;

int n, m, t;
int xi, di, ki;
long sum = 0;
int arr[51][51];
bool check[51][51];
bool flag = false;

void change(int x, int k, int d) {
	//번호가 x인 줄을 k만큼 회전시킴
	//2만큼 회전시킴

	vector <int> v;

	if (d == 0) { //시계방향이라면

		for (int i = m - k + 1; i <= m; i++) {
			v.push_back(arr[x][i]);
		}

		for (int i = 1; i <= m - k; i++) {
			v.push_back(arr[x][i]);
		}

		for (int i = 1; i <= m; i++) {
			arr[x][i] = v[i - 1];
		}
	}
	else { //반시계방향이라면

		for (int i = k + 1; i <= m; i++) {
			v.push_back(arr[x][i]);
		}
		for (int i = 1; i <= k; i++) {
			v.push_back(arr[x][i]);
		}

		for (int i = 1; i <= m; i++) {
			arr[x][i] = v[i - 1];
		}
	}


}


void erase() {
	//인접한 수들의 boolean 값을 true로 만들어서 지우는 함수

	flag = false;
	memset(check, false, sizeof(check));

	for (int i = 1; i < n; i++) {
		for (int j = 1; j < m; j++) {
			if (arr[i][j] != -1 && arr[i][j] == arr[i][j + 1]) {
				check[i][j] = true;
				check[i][j + 1] = true;
				flag = true;
			}
			if (arr[i][j] != -1 && arr[i][j] == arr[i + 1][j]) {
				check[i][j] = true;
				check[i + 1][j] = true;
				flag = true;
			}
		}
	}

	for (int i = 1; i < m; i++) {
		if (arr[n][i] != -1 && arr[n][i] == arr[n][i + 1]) {
			check[n][i] = true;
			check[n][i + 1] = true;
		}
	}

	for (int i = 1; i < n; i++) {
		if (arr[i][m] != -1 && arr[i][m] == arr[i + 1][m]) {
			check[i][m] = true;
			check[i + 1][m] = true;
		}
	}


	if (flag) { //없어져야 할 수가 존재한다면
		for (int i = 1; i <= n; i++) {
			for (int j = 1; j <= m; j++) {
				if (check[i][j]) {
					arr[i][j] = -1;
				}
			}
		}
	}
	else {
		//없어져야 할 수가 존재하지 않으면 평균구하기
		long sum = 0;
		long cnt = 0;
		for (int i = 1; i <= n; i++) {
			for (int j = 1; j <= m; j++) {
				if (arr[i][j] != -1) { //지워진 원소가 아니면
					sum += arr[i][j];
					cnt++;
				}
			}
		}

		double  average = (double)sum / (double)cnt; //평균

		for (int i = 1; i <= n; i++) {
			for (int j = 1; j <= m; j++) {
				if (arr[i][j] != -1) {
					if (arr[i][j] > average) { //평균보다 크면 -1하기
						arr[i][j]--;
					}
					else if (arr[i][j] < average) { //평균보다 작으면 +1하기
						arr[i][j]++;
					}
				}
			}
		}
	}
}

void rotate(int xi, int di, int ki) {
	//번호가 xi의 배수인 원판을 di방향으로 ki칸 회전시킴
	memset(check, false, sizeof(check));
	flag = false;


	for (int i = xi; i <= n; i += xi) {
		change(i, ki, di); //배열 바꾸기
	}
	erase(); //배열 지우기 
}


void input() {
	//입력 받기
	cin >> n >> m >> t;

	for (int i = 1; i <= n; i++) {
		for (int j = 1; j <= m; j++) {
			cin >> arr[i][j];
		}
	}

	for (int i = 0; i < t; i++) {
		cin >> xi >> di >> ki;

		rotate(xi, di, ki);
	}


}
int main() {
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);

	input();

	long answer = 0;
	for (int i = 1; i <= n; i++) {
		for (int j = 1; j <= m; j++) {
			if (arr[i][j] != -1) { //-1인 수만 제외하고 합을 구한다.
				answer += arr[i][j];
			}
		}
	}

	cout << answer;
}