글 읽기 - 어디가 틀렸는지 모르겠습니다ㅠㅠ 도움 부탁드립니다.

kwy2868 5년 전 1

예제 8번이 86이 나오는 것 보면 로직을 잘못 짠 것 같은데 어디가 문제인지 못찾고 있습니다 ㅠㅠ

고수님들의 많은 도움 부탁드립니다.

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Scanner;

public class Main {
	static int N;
	static int M;
	static int K;
	static ArrayList<Tree>[][] map;
	static ArrayList<Pair> checkList;
	static int[][] energy;
	static int[][] offer;
	static int[] dr = { -1, -1, -1, 0, 1, 1, 1, 0 };
	static int[] dc = { -1, 0, 1, 1, 1, 0, -1, -1 };

	public static void main(String[] args) {
		Scanner scanner = new Scanner(System.in);
		N = scanner.nextInt();
		M = scanner.nextInt();
		K = scanner.nextInt();
		map = (ArrayList<Tree>[][]) new ArrayList[N][N];
		for (int row = 0; row < N; row++) {
			for (int col = 0; col < N; col++)
				map[row][col] = new ArrayList<Tree>();
		}
		energy = new int[N][N];
		for (int i = 0; i < N; i++)
			Arrays.fill(energy[i], 5);
		offer = new int[N][N];
		for (int row = 0; row < N; row++) {
			for (int col = 0; col < N; col++) {
				offer[row][col] = scanner.nextInt();
			}
		}
		for (int i = 0; i < M; i++) {
			int row = scanner.nextInt() - 1;
			int col = scanner.nextInt() - 1;
			int age = scanner.nextInt();
			Tree tree = new Tree(row, col, age);
			map[row][col].add(tree);
		}

		for (int year = 0; year < K; year++) {
			spring();
			summer();
			fall();
			winter();
		}
		int answer = 0;
		for (int i = 0; i < N; i++) {
			for (int j = 0; j < N; j++) {
				if (map[i][j].size() > 0)
					answer += map[i][j].size();
			}
		}
		System.out.println(answer);
	}

	public static void spring() {
		checkList = new ArrayList<Pair>();
		for (int row = 0; row < N; row++) {
			for (int col = 0; col < N; col++) {
				if (map[row][col].size() > 0) {
					ArrayList<Tree> treeList = map[row][col];
					for (int i = treeList.size() - 1; i >= 0; i--) {
						Tree tree = treeList.get(i);
						if (tree.age > energy[row][col]) {
							tree.isAlive = false;
						} else {
							energy[row][col] -= tree.age;
							tree.age++;
						}
					}
					checkList.add(new Pair(row, col));
				}
			}
		}
	}

	public static void summer() {
		for(int index=0; index<checkList.size(); index++) {
			Pair pair = checkList.get(index);
			int row = pair.row;
			int col = pair.col;
			ArrayList<Tree> treeList = map[row][col];
			for (int i = treeList.size() - 1; i >= 0; i--) {
				Tree tree = treeList.get(i);
				if (!tree.isAlive) {
					energy[row][col] += tree.age / 2;
					treeList.remove(i);
				}
			}
		}
	}

	public static void fall() {
		for(int index=0; index<checkList.size(); index++) {
			Pair pair = checkList.get(index);
			int row = pair.row;
			int col = pair.col;
			ArrayList<Tree> treeList = map[row][col];
			for (int i = 0; i < treeList.size(); i++) {
				Tree tree = treeList.get(i);
				if (tree.age % 5 == 0) {
					int tRow = tree.row;
					int tCol = tree.col;
					for (int j = 0; j < 8; j++) {
						int nextRow = tRow + dr[j];
						int nextCol = tCol + dc[j];
						if (nextRow < 0 || nextRow > N - 1 || nextCol < 0 || nextCol > N - 1)
							continue;
						Tree t = new Tree(nextRow, nextCol, 1);
						map[nextRow][nextCol].add(0, t);
					}
				}
			}
		}
	}

	public static void winter() {
		for (int row = 0; row < N; row++) {
			for (int col = 0; col < N; col++) {
				energy[row][col] += offer[row][col];
			}
		}
	}
}
class Pair{
	int row;
	int col;
	
	Pair(int row, int col){
		this.row = row;
		this.col = col;
	}
}

class Tree {
	boolean isAlive = true;
	int row;
	int col;
	int age;

	Tree(int row, int col, int age) {
		this.row = row;
		this.col = col;
		this.age = age;
	}
}

qkrckd45 5년 전 2

Spring 부분로직이 잘못된것같은데요?

map에는 나이가 어린 나무순으로 정렬이 되어있고 나이가 어린 나무순으로 양분을 섭취해야되는데 뒤에서부터 탐색하시네요.

kwy2868 5년 전 1

아아 감사합니다! 리스트 처음에 넣는게 아니고 뒤에서부터 넣는 코드에서 수정을 하다보니

소스가 뒤엉켜있었나보네요ㅠㅠㅠ 수정해서 돌려보니 바로 해결됐습니다. 감사합니다!

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

문제

문제

출처

ICPC

어디가 틀렸는지 모르겠습니다ㅠㅠ 도움 부탁드립니다.

kwy2868 5년 전 1

qkrckd45 5년 전 2

kwy2868 5년 전 1

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말