BOJ16235_나무 재테크

https://www.acmicpc.net/problem/16235

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.Comparator;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;
import java.util.StringTokenizer;
 
public class Main {
 
    static class tree {
        int r;
        int c;
        int age;
 
        public tree(int r, int c, int age) {
            this.r = r;
            this.c = c;
            this.age = age;
        }
    }
 
    private static int N;
    private static int M;
    private static int K;
    private static int[][] ground;
    private static int[][] A;
    private static List<tree> list;
    private static List<tree> dead_list;
    private static List<Integer> delete_list;
    private static int[] dr = {-1,-1,-1,0,0,1,1,1};
    private static int[] dc = {-1,0,1,-1,1,-1,0,1};
 
    public static void main(String[] args) throws Exception {
        // *************입력 시작****************
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine().trim(), " ");
        N = Integer.parseInt(st.nextToken()); // 땅크기
        M = Integer.parseInt(st.nextToken()); // 나무 몇그루?
        K = Integer.parseInt(st.nextToken()); // 몇년?
        ground = new int[N + 1][N + 1]; // 땅의 양분상태 저장 배열 , 계속 변함
        A = new int[N + 1][N + 1]; // 각 땅에 공급해주는 양분 정보
        for (int i = 1; i <= N; i++) {
            st = new StringTokenizer(br.readLine().trim(), " ");
            for (int j = 1; j <= N; j++) {
                A[i][j] = Integer.parseInt(st.nextToken());
            }
        }
 
        list = new ArrayList<>();
        dead_list = new ArrayList<>();
        delete_list = new ArrayList<>();
        for (int i = 0; i < M; i++) {
            st = new StringTokenizer(br.readLine(), " ");
            int tree_r = Integer.parseInt(st.nextToken());
            int tree_c = Integer.parseInt(st.nextToken());
            int tree_age = Integer.parseInt(st.nextToken());
            list.add(new tree(tree_r, tree_c, tree_age));
        }
 
        for (int i = 1; i <= N; i++) {
            for (int j = 1; j <= N; j++) {
                ground[i][j] = 5;
            }
        }
        // **************입력 끝****************
        for (int i = 0; i < K; i++) {
            // 하나의 칸에 여러개의 나무 있을 경우 어린 나무부터 양분을 먹는 경우를 위해 현재 나무가 담긴 list를 sort해준다
            tree_sort();
            spring();
            if (dead_list.size() >= 1) {
                summer();
            }
            fall();
            winter();
        }
        System.out.println(list.size());
 
    } // end of main
 
    private static void tree_sort() {
        Collections.sort(list, new Comparator<tree>() {
            @Override
            public int compare(tree o1, tree o2) {
                return o1.age - o2.age;
            }
        });
    }
    
    private static void spring() {
        int list_size = list.size();
        delete_list = new ArrayList<>();
        for (int i = 0; i < list_size; i++) {
            int r = list.get(i).r;
            int c = list.get(i).c;
            int age = list.get(i).age;
            // 현재땅의 양분이 나무의 나이보다 크거나 같으면 땅양분상태와 나무나이를 갱신해준다
            if (ground[r][c] >= age) {
                ground[r][c] -= age;
                list.set(i, new tree(r, c, ++age));
            // 반대의 경우 여름에 양분으로 사용할 죽은 나무 리스트에 add해주고 현재 나무 리스트에서 삭제 해준다
            } else if (ground[r][c] < age) {
                delete_list.add(i);
                dead_list.add(new tree(r, c, age));
            }
        }
        Collections.sort(delete_list, new Comparator<Integer>() {
            @Override
            public int compare(Integer o1, Integer o2) {
                return o2-o1;
            }
        });
        for (int i = 0; i < delete_list.size(); i++) {
            int index = delete_list.get(i);
            list.remove(index);
        }
    }
    
    private static void summer() {
        // 죽은 나무만큼 현재땅의 양분상태를 갱신해준다
        for (int i = 0; i < dead_list.size(); i++) {
            int r = dead_list.get(i).r;
            int c = dead_list.get(i).c;
            int age = dead_list.get(i).age;
            ground[r][c] += age / 2;
        }
        dead_list.clear();
    }
    
    private static void fall() {
        // 나무의 나이가 5의 배수면 인접한 8칸에 나무나이=1인 나무를 생성해준다
        int list_size = list.size();
        for (int i = 0; i < list_size; i++) {
            int r = list.get(i).r;
            int c = list.get(i).c;
            int age = list.get(i).age;
            if(age % 5 == 0) {
                for (int j = 0; j < 8; j++) {
                    int nr = r + dr[j];
                    int nc = c + dc[j];
                    if(nr>0 && nr<=N && nc>0 && nc<=N) {
                        list.add(new tree(nr,nc,1));
                    }
                }
            }
            else 
                continue;
        }
    }
    
    private static void winter() {
        // 각칸에 해당하는 양분을 더해준다
        for (int i = 1; i <= N; i++) {
            for (int j = 1; j <= N; j++) {
                ground[i][j] += A[i][j];
            }
        }
    }
} // end of class