[백준] G5. 테트리스 (Java)

문제

테트리스는 C열 필드위에서 플레이하는 유명한 게임이다. 필드의 행의 수는 무한하다. 한 번 움직일 때, 아래와 같은 일곱가지 블록 중 하나를 필드에 떨어뜨릴 수 있다.

 

블록을 떨어뜨리기 전에, 플레이어는 블록을 90, 180, 270도 회전시키거나 좌우로 움직일 수 있다. 이때, 블록이 필드를 벗어나지 않으면 된다. 블록을 필드의 바닥이나 이미 채워져있는 칸의 위에 놓여지게 된다.

창영이가 하고있는 테트리스는 일반적인 테트리스와 약간 규칙이 다르다. 블록이 떨어졌을 때, 블록과 블록 또는 블록과 바닥 사이에 채워져있지 않은 칸이 생기면 안 된다.

예를 들어, 아래와 같이 각 칸의 높이가 2, 1, 1, 1, 0, 1인 경우를 생각해보자. 블록 5번을 떨어뜨리는 방법의 수는 아래와 같이 다섯가지이다.

테트리스 필드의 각 칸의 높이와 떨어뜨려야 하는 블록의 번호가 주어진다. 이때, 블록을 놓는 서로 다른 방법의 수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 C와 떨어뜨리는 블록의 번호 P가 주어진다. (1 ≤ C ≤ 100, 1 ≤ P ≤ 7)

둘째 줄에는 각 칸의 높이가 주어진다. 높이는 0보다 크거나 같고, 100보다 작거나 같은 자연수이다.

출력

첫째 줄에 블록을 떨어뜨리는 방법의 수를 출력한다.

풀이

필요 지식

jagged array(들쭉날쭉 배열)

Java에서 int[][][]는 가변 배열이라 각 패턴의 길이가 달라도 문제 없음

int[][][] blockPatterns = {
    {{0}, {0, 0, 0, 0}},   // 블록 1번: 세로, 가로
    {{0, 0}},              // 블록 2번: 정사각형
    ...
};

나의 풀이

풀이 단계

1. 각 블록의 상대적 길이 패턴 만들기
2. 각 블록을 돌면서 패턴 맞는 지 확인하고 맞으면 count 높이기

풀이 코드

1차 코드

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

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        int C = Integer.parseInt(st.nextToken());  // C개 열
        int P = Integer.parseInt(st.nextToken());  // 블록 번호 P
        int[] heights = new int[C];  // 각 칸의 높이를 담을 배열
        st = new StringTokenizer(br.readLine());
        for (int i = 0; i < C; i++) {
            heights[i] = Integer.parseInt(st.nextToken());
        }

        HashMap<Integer, Integer> map = new HashMap<>();
        for (int i = 1; i <= 7; i++) {
            map.put(i, 0);
        }

        map.put(1, C);

        int one = heights[0];
        int two = heights[1];

        if (one == two) {
            map.put(2, map.get(2)+ 1);
            map.put(6, map.get(6)+ 1);
            map.put(7, map.get(7)+ 1);
        }
        if (one - two == 1) {
            map.put(3, map.get(3)+ 1);
            map.put(5, map.get(5)+ 1);
        }
        if (one - two == -1) {
            map.put(4, map.get(4)+ 1);
            map.put(5, map.get(5)+ 1);
        }
        if (one - two == 2) {
            map.put(6, map.get(6)+ 1);
        }
        if (one - two == -2) {
            map.put(7, map.get(7)+ 1);
        }

        for (int i = 2; i < C; i++) {
            int ch = heights[i];
            if (one == two && two == ch) {
                map.put(5, map.get(5)+ 1);
                map.put(6, map.get(6)+ 1);
                map.put(7, map.get(7)+ 1);
            }
            if (two == ch) {
                map.put(2, map.get(2)+ 1);
                map.put(6, map.get(6)+ 1);
                map.put(7, map.get(7)+ 1);
            }

            if (two - ch == 1) {
                map.put(3, map.get(3)+ 1);
                map.put(5, map.get(5)+ 1);
            }
            if (two - ch == -1) {
                map.put(4, map.get(4)+ 1);
                map.put(5, map.get(5)+ 1);
            }
            if (two - ch == 2) {
                map.put(6, map.get(6)+ 1);
            }
            if (two - ch == -2) {
                map.put(7, map.get(7)+ 1);
            }

            if (one == two && ch - two == 1) {
                map.put(3, map.get(3)+ 1);
            }
            if (one - two == 1 && ch == two) {
                map.put(4, map.get(4)+ 1);
            }
            if (one - two == -1 && two == ch) {
                map.put(6, map.get(6)+ 1);
            }
            if (one == two && two - ch == 1) {
                map.put(7, map.get(7)+ 1);
            }

            if (one == ch && one - two == 1) {
                map.put(5, map.get(5)+ 1);
            }

            if (i < C-1) {
                if (one == two && one == ch && one == heights[i + 1]) {
                    map.put(1, map.get(1) + 1);
                }
            }

            one = two;
            two = ch;
        }
    System.out.println(map.get(P));

    }
}

2차 코드

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

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        int C = Integer.parseInt(st.nextToken());  // C개 열
        int P = Integer.parseInt(st.nextToken());  // 블록 번호 P
        int[] heights = new int[C];  // 각 칸의 높이를 담을 배열
        st = new StringTokenizer(br.readLine());
        for (int i = 0; i < C; i++) {
            heights[i] = Integer.parseInt(st.nextToken());
        }

        int[][][] blockPatterns = {
                {{0}, {0, 0, 0, 0}},
                {{0, 0}},
                {{0, 0, 1}, {0, -1}},
                {{0, -1, -1}, {0, 1}},
                {{0, 0, 0}, {0, 1}, {0, -1}, {0, -1, 0}},
                {{0, 0, 0}, {0, -2}, {0, 1, 1}, {0, 0}},
                {{0, 0, 0}, {0, 0}, {0, 0, -1}, {0, 2}}
        };

        int count = 0;

        for (int i = 0; i < blockPatterns[P - 1].length; i++) {
            int[] pattern = blockPatterns[P - 1][i];  // 현재 블록의 i번째 회전 패턴
            for (int j = 0; j <= C - pattern.length; j++) {
                boolean match = true;
                int base = heights[j];
                for (int k = 0; k < pattern.length; k++) {
                    if (heights[j + k] - base != pattern[k]) {
                        match = false;
                        break;
                    }
                }

                if (match) count++;
            }
        }

        System.out.println(count);
    }
}