[백준] 3019. 테트리스 (파이썬)

문제

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

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

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

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

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

입력

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

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

출력

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

예제 입력 1

6 5
2 1 1 1 0 1

예제 출력 1 

5

예제 입력 2

5 1
0 0 0 0 0

예제 출력 2

7

예제 입력 3

9 4
4 3 5 4 6 5 7 6 6

예제 출력 3

1

 

 

 

문제 링크

---

풀이

블록 별로 바닥의 높이 조건을 확인하며 들어갈 수 있는지 확인한다.

1. 블록 1 부터 7 까지 들어갈 수 있는 모든 모양의 경우의 수를 구한다.
2. 블록이 들어갔을 때 블록과 블록, 블록과 바닥 사이에 빈틈이 생기지 않도록 주어진 높이 변수를 비교한다.
3. 왼쪽 열부터 오른쪽 열까지 탐색하며 블록이 들어갈 수 있는 경우를 센다.

def count_blocks(patterns):
    cnt = 0
    for length, condition in patterns:
        for i in range(c - length + 1):
            if condition(i):
                cnt += 1
    return cnt

def block1(): # 1번 블록
    return count_blocks([
        (1, lambda i: board[i] >= 0),
        (4, lambda i: board[i] == board[i+1] == board[i+2] == board[i+3])
    ])

def block2(): # 2번 블록
    return count_blocks([
        (2, lambda i: board[i] == board[i+1])
    ])

def block3(): # 3번 블록
    return count_blocks([
        (3, lambda i: board[i] == board[i+1] == board[i+2] - 1),
        (2, lambda i: board[i] - 1 == board[i+1])
    ])

def block4(): # 4번 블록
    return count_blocks([
        (3, lambda i: board[i] - 1 == board[i+1] == board[i+2]),
        (2, lambda i: board[i] == board[i+1] - 1)
    ])

def block5(): # 5번 블록
    return count_blocks([
        (3, lambda i: board[i] == board[i+1] == board[i+2]),
        (3, lambda i: board[i] - 1 == board[i+1] == board[i+2] - 1),
        (2, lambda i: board[i] - 1 == board[i+1]),
        (2, lambda i: board[i] == board[i+1] - 1)
    ])

def block6(): # 6번 블록
    return count_blocks([
        (3, lambda i: board[i] == board[i+1] == board[i+2]),
        (2, lambda i: board[i] - 2 == board[i+1]),
        (3, lambda i: board[i] == board[i+1] - 1 == board[i+2] - 1),
        (2, lambda i: board[i] == board[i+1])
    ])

def block7(): # 7번 블록
    return count_blocks([
        (3, lambda i: board[i] == board[i+1] == board[i+2]),
        (2, lambda i: board[i] == board[i+1]),
        (3, lambda i: board[i] - 1 == board[i+1] - 1 == board[i+2]),
        (2, lambda i: board[i] == board[i+1] - 2)
    ])

c, p = map(int, input().split())
board = list(map(int, input().split()))
blocks = {1: block1(), 2: block2(), 3: block3(), 4: block4(), 5: block5(), 6: block6(), 7: block7()}
print(blocks[p])