글 읽기 - 어떤 부분이 틀렸을까요..?

mrseos 3년 전 0

안녕하세요?

어디가 틀렸는지 잘 모르겠어서 질문을 올리게 됐습니다.

24~25 라인은 x배수의 원판을 d방향으로, k만큼 시계 또는 반시계 방향으로 움직이는 부분입니다.

28~37 라인은 아래의 그림처럼 동 서 남 북으로 각각의 인접한 수를 검사하는 부분입니다.

38~54 라인은 아래의 그림처럼 같은 원판내의 인접한 수를 검사하는 부분입니다.

마지막 인덱스인 M-1의 경우 0번째와 비교할 수 있게 했습니다.

55~63은 위에서 이차원 체크배열을 통해 지워져야하는 수가 true인 부분을 찾아 지워줍니다.

만약 지울 수 없는 수가 있다면 isSet 변수는 false가 됩니다.

64~86은 isSet이 false인 경우 수의 평균을 구하고, 평균보다 작은 값은 +1, 큰 값은 -1로 처리해주는 부분입니다.

혹시 잘못된 부분이 보이신다면... 알려주시면 감사하겠습니다!

import java.io.*;
import java.util.*;
public class Main {
	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(br.readLine()," ");
		int N = Integer.valueOf(st.nextToken());
		int M = Integer.valueOf(st.nextToken());
		int T = Integer.valueOf(st.nextToken());
		ArrayList<ArrayList<Integer>> board = new ArrayList<ArrayList<Integer>>();
		for(int i=0; i<N; i++) {
			ArrayList<Integer> number =  new ArrayList<Integer>();
			st = new StringTokenizer(br.readLine(), " ");
			for(int j=0; j<M; j++) {
				number.add(Integer.valueOf(st.nextToken()));
			}
			board.add(number);
		}
		for(int tc=0; tc<T; tc++) {
			st = new StringTokenizer(br.readLine(), " ");
			int x = Integer.valueOf(st.nextToken());
			int d = Integer.valueOf(st.nextToken());
			int k = Integer.valueOf(st.nextToken());
			for(int i=x; i<=N; i+=x) {
				Collections.rotate(board.get(i-1), (d==0)?k:-k);
			}
			boolean check[][] = new boolean[N][M];
			for(int i=0; i<M; i++) { // 그림에서 나온 원판의 동 서 남 북으로 인접수 찾기
				for(int j=0; j<N-1; j++) {
					if(board.get(j).get(i)!=0) {
						if(board.get(j).get(i)==board.get(j+1).get(i)) {
							check[j][i] = true;
							check[j+1][i] = true;
						}
					}
				}
			}
			for(int i=0; i<N; i++) { // 같은 번호 원판 내의 인접수 찾기
				for(int j=0; j<M; j++) {
					if(board.get(i).get(j)!=0) {
						if(j==M-1) {
							if(board.get(i).get(M-1)==board.get(i).get(0)) {
								check[i][M-1] = true;
								check[i][0] = true;
							}
						} else {
							if(board.get(i).get(j)==board.get(i).get(j+1)) {
								check[i][j] = true;
								check[i][j+1] = true;
							}
						}
					}
				}
			}
			boolean isSet = false;
			for(int i=0; i<N; i++) { //지울 수가 있는지 검사하고 지움
				for(int j=0; j<M; j++) {
					if(check[i][j]) {
						board.get(i).set(j, 0);
						isSet = true;
					}
				}
			}
			if(!isSet) {  //지울 수가 없다면 평균을 구하고 평균보다 크면 -1, 작으면 +1
				double sum=0, count=0;
				for(int i=0; i<N; i++) {
					for(int j=0; j<M; j++) {
						if(board.get(i).get(j)>0) {
							count++;
							sum += board.get(i).get(j);
						}
					}
				}
				double avg = sum/count;
				for(int i=0; i<N; i++) {
					for(int j=0; j<M; j++) {
						if(board.get(i).get(j)!=0) {
							if(board.get(i).get(j)>avg) {
								board.get(i).set(j, board.get(i).get(j)-1);
							} else if(board.get(i).get(j)<avg) {
								board.get(i).set(j, board.get(i).get(j)+1);
							}
						}
					}
				}
			}
		}
		int answer = 0; //최종 정답
		for(int i=0; i<N; i++) {
			for(int j=0; j<M; j++) {
				answer += board.get(i).get(j);
			}
		}
		System.out.println(answer);
	}
}

5 5 1
1 2 3 4 5
3 4 5 1 2
2 3 4 5 1
5 1 2 3 4
4 5 1 2 3
2 0 2

output 45

5 5 5
1 2 3 4 5
3 4 5 1 2
2 3 4 5 1
5 1 2 3 4
4 5 1 2 3
2 0 2
4 1 1
3 0 1
3 0 1
3 0 1

output 6

5 5 6
1 2 3 4 5
3 4 5 1 2
2 3 4 5 1
5 1 2 3 4
4 5 1 2 3
2 0 2
4 1 1
3 0 1
3 0 1
3 0 1
3 0 1

output : 0

mrseos 3년 전 0

현재 28~54를 아래와 같이 4방향 탐색으로 수정 후 테스트해보고 있습니다.

여전히 틀렸다고 뜨네요 ㅠㅠ

int dir[][] = { {0,1}, {0,-1}, {1,0}, {-1,0}};
for(int i=0; i<N; i++) { // 같은 번호 원판 내의 인접수 찾기
	for(int j=0; j<M; j++) {
		if(board.get(i).get(j)!=0) {
			for(int v=0; v<dir.length; v++) {
				int ni = i + dir[v][0];
				int nj = j + dir[v][1];
				// 같은 원판은 처음과 끝이 이어져있어서 보정해주는 부분
				if(nj>M-1) 	nj=0; 
				if(nj<0) 	nj=M-1;
				// 아래는 인덱스가 넘어가거나 0인 부분
				if(ni<0 || ni>=N || board.get(ni).get(nj)==0) 
					continue;
				// 인접한 부분이 존재할 경우
				if(board.get(ni).get(nj)==board.get(i).get(j)) {
					check[ni][nj] = true;
					check[i][j] = true;
				}
			}
		}
	}
}

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

문제

문제

출처

ICPC

어떤 부분이 틀렸을까요..?

mrseos 3년 전 0

mrseos 3년 전 0

mrseos 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말