글 읽기 - 테케 다맞고 모든 반례 다맞고 주의할 점도 다 고쳐봤는데 틀립니다ㅠㅠ

kp6069 3년 전 0

뭐가 문제일까요.... 알려주시면 감사하겠습니다

import java.io.*;
import java.util.*;

public class N_17822 {
	static LinkedList<Integer>[] A;

	static void clockwise(int index, int num) {
		for (int i = 0; i < num; i++) {
			int a = A[index].remove(A[index].size() - 1);
			A[index].add(0, a);
		}
	}

	static void counterclockwise(int index, int num) {
		for (int i = 0; i < num; i++) {
			int a = A[index].remove(0);
			A[index].add(a);
		}
	}

	public static void main(String[] args) {
		// TODO Auto-generated method stub
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		try {
			String[] input = br.readLine().split(" ");
			int N = Integer.parseInt(input[0]); // 원판의 개수
			int M = Integer.parseInt(input[1]); // 원판 내 수의 개수
			int T = Integer.parseInt(input[2]); // 몇 번 회전시킬건지

			A = (LinkedList<Integer>[]) new LinkedList[N + 1];
			for (int i = 1; i <= N; i++) {
				String[] arr = br.readLine().split(" ");
				A[i] = new LinkedList<Integer>();
				for (int j = 0; j < M; j++) {
					A[i].add(Integer.parseInt(arr[j]));
				}
			}

			for (int i = 0; i < T; i++) { // T번 회전
				String[] arr = br.readLine().split(" ");
				int x = Integer.parseInt(arr[0]);
				int d = Integer.parseInt(arr[1]); // 0:시계 방향, 1: 반시계 방향
				int k = Integer.parseInt(arr[2]); // k칸 회전

				// 원판 회전
				for (int j = x; j <= N; j += x) {
					if (d == 0) {
						clockwise(j, k);
					} else if (d == 1) {
						counterclockwise(j, k);
					}

				}

				// 인접하면서 수가 같은 것 찾기(지워진 수는 -1로 표시)
				int[][] B = new int[N + 1][M]; // 지워진 수 표시 하기 위함
				for (int j = 1; j <= N; j++) {
					if (A[j].get(0) == A[j].get(M - 1) && A[j].get(0) != -1) {
						B[j][0] = -1;
						B[j][M - 1] = -1;
					}
				}

				for (int j = 1; j <= N; j++) {
					for (int m = 0; m < M - 1; m++) {
						if (A[j].get(m) == A[j].get(m + 1) && A[j].get(m) != -1) {
							B[j][m] = -1;
							B[j][m + 1] = -1;
						}
					}
				}

				for (int j = 0; j < M; j++) {
					for (int m = 1; m < N; m++) {
						if (A[m].get(j) == A[m + 1].get(j) && A[m].get(j) != -1) {
							B[m][j] = -1;
							B[m + 1][j] = -1;
						}
					}
				}

				// 원판에 지워진 수 반영
				int flag = 0;
				for (int j = 1; j <= N; j++) {
					for (int m = 0; m < M; m++) {
						if (B[j][m] == -1) {
							flag = 1;
							A[j].set(m, -1);
						}
					}
				}

				if (flag == 0) {
					// 평균 구하기
					int sum = 0;
					int count = 0;
					for (int j = 1; j <= N; j++) {
						for (int m = 0; m < M; m++) {
							if (A[j].get(m) != -1) {
								count++;
								sum += A[j].get(m);
							}
						}
					}
					double aver = (double) sum / count;
					// 수 빼고 더하기
					for (int j = 1; j <= N; j++) {
						for (int m = 0; m < M; m++) {
							if (A[j].get(m) != -1) {
								if (A[j].get(m) < aver) {
									int a = A[j].get(m);
									A[j].set(m, a + 1);
								} else if (A[j].get(m) > aver) {
									int a = A[j].get(m);
									A[j].set(m, a - 1);
								}
							}
						}
					}
				}
			}

			// 원판에 적힌 수의 총 합
			int total = 0;
			for (int i = 1; i <= N; i++) {
				for (int j = 0; j < M; j++) {
					if (A[i].get(j) != -1) {
						total += A[i].get(j);
					}
				}
			}
			System.out.println(total);
		} catch (IOException e) {
			e.printStackTrace();
		}
	}

}

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

문제

문제

출처

ICPC

테케 다맞고 모든 반례 다맞고 주의할 점도 다 고쳐봤는데 틀립니다ㅠㅠ

kp6069 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말