[백준 - 18808] 스티커 붙이기 - C++

분류: 구현, 시뮬레이션 /

 

문제

문제 링크

문제 설명

혜윤이는 최근에 다양한 대회를 참여하면서 노트북에 붙일 수 있는 스티커들을 많이 받았다. 스티커는 아래와 같이 사각 모눈종이 위에 인쇄되어 있으며, 스티커의 각 칸은 상하좌우로 모두 연결되어 있다. 또한 모눈종이의 크기는 스티커의 크기에 꼭 맞아서, 상하좌우에 스티커가 포함되지 않는 불필요한 행이나 열이 존재하지 않는다.

아래는 올바른 모눈종이의 예시이다. 주황색 칸은 스티커가 붙은 칸을, 하얀색 칸은 스티커가 붙지 않은 칸을 나타낸다.

반면 아래는 올바르지 않은 모눈종이의 예시이다. 첫 번째는 윗쪽에 불필요한 행이 있고, 두 번째는 왼쪽에 불필요한 열이 있다. 그리고 세 번째는 스티커의 각 칸이 상하좌우로 모두 연결되어 있지 않다.

혜윤이는 자신의 노트북에 이 스티커들을 붙이기로 했다. 혜윤이의 노트북은 마침 직사각형 모양이고, 스티커가 인쇄된 모눈종이와 같은 간격으로 격자가 그려져 있다. 혜윤이는 스티커들을 먼저 받았던 것부터 차례대로 격자에 맞춰서 붙이려고 한다.

혜윤이가 스티커를 붙이는 방법은 다음과 같다.

1. 스티커를 회전시키지 않고 모눈종이에서 떼어낸다.
2. 다른 스티커와 겹치거나 노트북을 벗어나지 않으면서 스티커를 붙일 수 있는 위치를 찾는다. 혜윤이는 노트북의 위쪽부터 스티커를 채워 나가려고 해서, 스티커를 붙일 수 있는 위치가 여러 곳 있다면 가장 위쪽의 위치를 선택한다. 가장 위쪽에 해당하는 위치도 여러 곳이 있다면 그중에서 가장 왼쪽의 위치를 선택한다.
3. 선택한 위치에 스티커를 붙인다. 만약 스티커를 붙일 수 있는 위치가 전혀 없어서 스티커를 붙이지 못했다면, 스티커를 시계 방향으로 90도 회전한 뒤 2번 과정을 반복한다.
4. 위의 과정을 네 번 반복해서 스티커를 0도, 90도, 180도, 270도 회전시켜 봤음에도 스티커를 붙이지 못했다면 해당 스티커를 붙이지 않고 버린다.

아래의 예시를 통해 스티커를 붙이는 과정을 이해해보자. 노트북은 세로 5칸, 가로 4칸 크기이고, 혜윤이가 가지고 있는 스티커들은 아래와 같다. 왼쪽에서 오른쪽 순으로 스티커를 붙일 것이다.

1. 첫 번째 스티커는 회전 없이 온전히 붙일 수 있는 공간이 아래와 같이 6곳이 있다.

이 중에서 가장 위쪽의 위치, 가능한 가장 위쪽의 위치가 여러 개이면 그 중에서 가장 왼쪽의 위치는 첫 번째이다. 스티커를 붙인 후 노트북의 모양은 아래와 같다.

1. 두 번째 스티커는 회전 없이 온전히 붙일 수 있는 공간이 없다. 그러나 시계 방향으로 90도 회전한 후에는 붙일 수 있는 공간이 1개 생긴다. 해당 공간에 스티커를 붙인 후 노트북의 모양은 아래와 같다.
2. 
3. 세 번째 스티커는 스티커를 시계방향으로 0, 90, 180, 270도 회전시킨 모양에 대해 전부 확인을 해도 스티커를 붙일 수 있는 공간이 없다. 그러므로 해당 스티커를 붙이지 않고 버린다.
4. 네 번째 스티커는 스티커를 시계방향으로 0, 90, 180도 회전 시킨 모양에 대해 온전히 붙일 수 있는 공간이 없다. 그러나 시계 방향으로 270도 회전한 후에는 공간이 1개 생긴다. 스티커를 붙인 후 노트북의 모양은 아래와 같다. 최종적으로 노트북의 18칸이 스티커로 채워졌다.
5. 

혜윤이는 스티커를 다 붙인 후의 노트북의 모습이 궁금해졌다. 노트북의 크기와 스티커들이 주어졌을 때 스티커들을 차례대로 붙이고 난 후 노트북에서 몇 개의 칸이 채워졌는지 구해보자.

입력

첫째 줄에 노트북의 세로와 가로 길이를 나타내는 N(1 ≤ N ≤ 40)과 M(1 ≤ M ≤ 40), 그리고 스티커의 개수 K(1 ≤ K ≤ 100)이 한 칸의 빈칸을 사이에 두고 주어진다.

그 다음 줄부터는 K개의 스티커들에 대한 정보가 주어진다. 각 스티커는 아래와 같은 형식으로 주어진다.

먼저 i번째 스티커가 인쇄된 모눈종이의 행의 개수와 열의 개수를 나타내는 R_i(1 ≤ R_i ≤ 10)와 C_i(1 ≤ C_i ≤ 10)가 한 칸의 빈칸을 사이에 두고 주어진다.

다음 R_i개의 줄에는 각 줄마다 모눈종이의 각 행을 나타내는 C_i개의 정수가 한 개의 빈칸을 사이에 두고 주어진다. 각 칸에 들어가는 값은 0, 1이다. 0은 스티커가 붙지 않은 칸을, 1은 스티커가 붙은 칸을 의미한다.

문제에서 설명한 것과 같이 스티커는 모두 올바른 모눈종이에 인쇄되어 있다. 구체적으로 스티커의 각 칸은 상하좌우로 모두 연결되어 있고, 모눈종이의 크기는 스티커의 크기에 꼭 맞아서 상하좌우에 스티커에 전혀 포함되지 않는 불필요한 행이나 열이 존재하지 않는다.

출력

첫째 줄에 주어진 스티커들을 차례대로 붙였을 때 노트북에서 스티커가 붙은 칸의 수를 출력한다.

 

풀이

#include <iostream>
#include <vector>
using namespace std;

struct xy {int x, y;};
int N, M, K ,ans;
bool laptop[40][40];

void swap(int &a, int &b) { int tmp = a; a = b; b = tmp; }
void rotate90(vector<xy> &sticker, int &R, int &C) {
    swap(R, C);
    for(auto &[x, y] : sticker) {
        swap(x, y);
        y = C - 1 - y;
    }
}
bool stick(int locX, int locY, vector<xy> &sticker) {
    for(const auto &[x, y] : sticker)
        if(laptop[locX+x][locY+y]) return false;
    for(const auto &[x, y] : sticker) 
        laptop[locX+x][locY+y] = true;
    return true;
}
int main() {
    ios::sync_with_stdio(0); cin.tie(0);
    cin >> N >> M >> K;
    while(K--) {
        int R, C; cin >> R >> C;
        vector<xy> sticker;
        for(int r=0; r<R; r++) {
            for(int c=0; c<C; c++) {
                bool isSticker; cin >> isSticker;
                if(isSticker) sticker.push_back({r, c});
            }
        }

        bool sticked = false;
        for(int rot = 4; not sticked and rot--;) {
            for(int i = 0; not sticked and i <= N-R; i++)
                for(int j = 0; not sticked and j <= M-C; j++)
                    sticked = stick(i, j, sticker);
            if(not sticked) rotate90(sticker, R, C);
        }
        if(sticked) ans += sticker.size();
    }
    cout << ans;
}

스티커는 상대좌표로, 랩탑은 절대좌표로 생각하고 스티커의 상대좌표를 저장

`main`의 3중 for문이 핵심

• 스티커가 안 붙었다면 0, 90, 180, 270도에 대해 다음을 수행
  • 스티커가 안 붙었다면 상 -> 하, 좌 -> 우의 순서로 스티커의 원점을 절대좌표`(i, j)`로 설정
    • `(i, j)`에 스티커를 붙일 수 있다면 붙인다.
  • 스티커가 안 붙었다면 스티커 90도 회전

`stick`: 스티커를 붙일 수 있는지 확인하고 붙이는 함수