글 읽기 - 테스트케이스 5번이 이해가 안갑니다

qls0860 4년 전 0

초기상태

1 2 3 4 5 6
2 3 4 5 6 7
3 4 5 6 7 8
4 5 6 7 8 9

2 1 4 입력

(2, 4번째 원판을 반시계 방향으로 4칸 돌린다)

1 2 3 4 5 6
6 7 2 3 4 5
3 4 5 6 7 8
8 9 4 5 6 7

인접한 수가 없으니 평균을 구하고

평균보다 작은 수는 1 더하고 큰 수는 1을 뺍니다.

여기서 평균은 5

2 3 4 5 5 5
5 6 3 4 5 5
4 5 5 5 6 7
7 8 5 5 5 6

3 0 1 입력

(3번째 원판을 시계방향으로 1칸 돌린다)

2 3 4 5 5 5
5 6 3 4 5 5
7 4 5 5 5 6
7 8 5 5 5 6

인접한 곳을 지웁니다.

2 3 4 0 0 0
5 6 3 4 0 0
0 4 0 0 0 0
0 8 0 0 0 0

2 1 2 입력

(2, 4번째 원판을 반시계 방향으로 2칸 돌린다)

2 3 4 0 0 0
3 4 0 0 5 6
0 4 0 0 0 0
0 0 0 0 0 8

인접한 곳을 지웁니다.

2 3 4 0 0 0
3 0 0 0 5 6
0 0 0 0 0 0
0 0 0 0 0 8

총 합은 31 아닌가요?

제 코드에 무슨 문제가 있나요?

package Main;

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.Arrays;

public class Main {
	
	static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
	static BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
	static int N, M, K, circle[][];
	static boolean[][] erase;
	
	public static void main(String[] args) throws Exception {
		String[] s = br.readLine().split(" ");
		N = Integer.parseInt(s[0]);
		M = Integer.parseInt(s[1]);
		K = Integer.parseInt(s[2]);
		
		circle = new int[N + 2][M + 2]; // 인접 조건 따지기 좋게 크게 잡기
		erase = new boolean[N + 2][M + 2];
		
		int i, j, l, x, d, k, avg = 0, cnt = 0, tmp[] = new int[M + 1];
		
		for (i = 1; i <= N; i++) {
			s = br.readLine().split(" ");
			for (j = 1; j <= M; j++)
				circle[i][j] = Integer.parseInt(s[j - 1]);
		}
		
		for (i = 0; i < K; i++) {
			s = br.readLine().split(" ");
			x = Integer.parseInt(s[0]);
			d = Integer.parseInt(s[1]);
			k = Integer.parseInt(s[2]);
			
			k = d == 0 ? k : (M - k); 
			
			// x의 부호인 원판을 d방향으로 k칸 돌리기
			for (j = x; j <= N; j += x) { 
				for (l = 1; l <= M; l++) tmp[l] = circle[j][l]; // 해당 원판 복사
				
				for (l = 1; l <= M; l++) {
					int idx = (l + k) % M;
					idx = idx == 0 ? M : idx;
					
					circle[j][idx] = tmp[l]; // 원판 돌리기
				}
			}
			
			// 인접조건 따지기
			boolean isNo = true;
			
			for (j = 1; j <= N; j++) {
				for (l = 1; l <= M; l++) {
					if (circle[j][l] <= 0) continue;
					
					// 행 조건
					if (circle[j - 1][l] != 0 && circle[j][l] == circle[j - 1][l]) erase[j][l] = erase[j - 1][l] = true;
					if (circle[j + 1][l] != 0 &&  circle[j][l] == circle[j + 1][l]) erase[j][l] = erase[j + 1][l] = true;
					
					// 열 조건
					if (circle[j][l - 1] != 0 &&  circle[j][l] == circle[j][l - 1]) erase[j][l] = erase[j][l - 1] = true;
					if (circle[j][l + 1] != 0 &&  circle[j][l] == circle[j][l + 1]) erase[j][l] = erase[j][l + 1] = true;
				}
				
				// 양 끝 인접
				if (circle[j][1] != 0 && circle[j][N] != 0 && circle[j][1] == circle[j][N]) erase[j][1] = erase[j][N] = true;
			}
			
			
			for (j = 1; j <= N; j++) {
				for (l = 1; l <= M; l++) {
					if (erase[j][l]) {
						isNo = false;
						circle[j][l] = 0;
					}
					erase[j][l] = false;
				}
			}
			
			// 인접한 곳이 없을 때
			if (isNo) {
				avg = 0;
				cnt = 0;
				
				for (j = 1; j <= N; j++) { 
					for (l = 1; l <= M; l++) {
						if (circle[j][l] <= 0) continue;
						avg += circle[j][l];
						cnt++;
					}
				}
				
				avg /=  cnt; // 평균
				
				for (j = 1; j <= N; j++) {
					for (l = 1; l <= M; l++) {
						if (circle[j][l] <= 0) continue;
						if (circle[j][l] > avg) { // 크면 1빼고
							circle[j][l]--;
						} else if (circle[j][l] < avg) { // 작으면 1 더한다
							circle[j][l]++;
						}
					}
				}
			}
		}
		
		int ans = 0;
		
		for (i = 1; i <= N; i++)
			for (j = 1; j <= M; j++)
				ans += circle[i][j];
		
		bw.write(ans + "");
		bw.close();
	}
}

portableangel 4년 전 1

3 0 1 입력시 두 번째 줄의 첫 수 5는 두 번째 줄의 마지막 수 5와 인접하고 같으니 지워져야 합니다.

qls0860 4년 전 0

아 그러네요 감사합니다

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

테스트케이스 5번이 이해가 안갑니다

qls0860 4년 전 0

portableangel 4년 전 1

qls0860 4년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말