[백준] G4 15685 드래곤 커브 (java)

15685번: 드래곤 커브

풀이

 

1) 좌표, 시작방향, 세대 정보를 담을 클래스를 만든다

2) 모든 방향을 리스트에 담음

3) 방향이 담긴 리스트로 맵을 그린다

4) 네모를 찾는다.

 

주의사항

 

1) 방향을 담을때 반대로 담아야함 (직접 해봐야 암)

2) 리스트를 만들땐 스택을 이용해야함

 

package com.BackJoon;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.StringReader;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Stack;
import java.util.StringTokenizer;

public class BJ_G4_15685_드래곤커브 {
    static BufferedReader input = new BufferedReader(new InputStreamReader(System.in));
    static StringTokenizer tokens;
    static int N,map[][] = new int[101][101], answer;
    static List<Node> list = new ArrayList<>();
    public static void main(String[] args) throws NumberFormatException, IOException {
        //input = new BufferedReader(new StringReader(src));
        N = Integer.parseInt(input.readLine());
        for(int n=0; n<N; n++) {
            tokens = new StringTokenizer(input.readLine());
            int x = Integer.parseInt(tokens.nextToken());
            int y = Integer.parseInt(tokens.nextToken());
            int d = Integer.parseInt(tokens.nextToken());
            int g = Integer.parseInt(tokens.nextToken());
            dirStore(y,x,d,g);
        }

        getSquare();
        System.out.println(answer);
    }

        // 모든 작업이 끝난 후 정사각형의 개수를 구하는 함수
    private static void getSquare() { 
        for(int r=0; r<map.length-1; r++) {
            for(int c=0; c<map.length-1; c++) {
                if(map[r][c] == 1 && map[r][c] == map[r+1][c] && map[r][c] == map[r][c+1] && map[r][c]== map[r+1][c+1]) {
                    answer++;
                }
            }    
        }
    }

    // 방향을 리스트에 저장하는 함수
    private static void dirStore(int r, int c, int d, int g) { 
        int cnt = 0;
        List<Integer> dir = new ArrayList<>();
        Stack<Integer> stack = new Stack<>();
        dir.add(d);

        while(cnt < g) {
            for(int i=0; i<dir.size(); i++) {
                stack.push(dir.get(i));
            }

            while(!stack.isEmpty()) {
                switch (stack.pop()) {
                case 0:
                    dir.add(1);
                    break;
                case 1:
                    dir.add(2);
                    break;
                case 2:
                    dir.add(3);
                    break;
                case 3:
                    dir.add(0);
                    break;
                }
            }
            cnt++;
        }
        makeMap(r,c,dir);
    }

    // 방향(dir)을 받아 지도를 그리는 함수
    private static void makeMap(int r, int c, List<Integer> dir) {  
        map[r][c] = 1;
        for(int i=0; i<dir.size(); i++) {
            switch (dir.get(i)) {
            case 0: 
                c = c+1; 
                if(isIn(r, c)) {

                }
                map[r][c] = 1;
                break;
            case 1:
                r = r-1; 
                if(isIn(r, c)) {
                    map[r][c] = 1;
                }
                break;
            case 2:
                c = c-1; 
                if(isIn(r, c)) {
                    map[r][c] = 1;
                }
                break;
            case 3:
                r = r+1; 
                if(isIn(r, c)) {
                    map[r][c] = 1;
                }
                break;

            }
        }

    }

        // 배열 밖으로 나가는지 검사
    static boolean isIn(int r, int c) {
        return (r>=0 && c>=0 && r<map.length && c<map.length);
    }

    static class Node{
        int r;
        int c;
        int d;
        int g;
        public Node(int r, int c, int d, int g) {
            super();
            this.r = r;
            this.c = c;
            this.d = d;
            this.g = g;
        }
        @Override
        public String toString() {
            return "Node [r=" + r + ", c=" + c + ", d=" + d + ", g=" + g + "]";
        }
    }

    static String src =
            "3\r\n" + 
            "3 3 0 1\r\n" + 
            "4 2 1 3\r\n" + 
            "4 2 2 1";
}