백준 14500 테트로미노 - 모든게 다 dfs로 되는건 아니다 c++

문제 링크

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

14500번: 테트로미노

풀이 생각 및 아이디어

한 점에서 블록 3개(이미 있는 블록 한개 포함 4개) 가 뻗어나가는 모든 경우의 수를 구하는 것이기 때문에, dfs를 구현해서 풀어야 하는 것은 어렵지 않게 생각할 수 있었다.

 

하지만 문제가 2가지 있었다.

1. ㅗ,ㅏ,ㅜ,ㅓ 모양의 블록은, dfs로 구현할 수 없는 블록의 종류이다. ㅗ의 정 중앙 블록을 탐색하게 되면, 나머지 두 개의 블록을 동시에 탐색을 할 수가 없기 때문이다.

2. 중복해서 탐색하는 경우의 수가 마음에 걸렸다. ㅁ 모양의 블록을 예를 들어서 생각해보면, 1,1 1,2 2,1 2,2 에 위치한 ㅁ 모양의 블록은 총 4번이나 중복되서 계산을 하게 된다. 나는 이렇게 중복되서 계산하게 되는 경우가 과연 채점을 통과할 것인지 의문이 들었다.(dfs는 재귀라서 함부로 시간복잡도를 계산할 수가 없겠더라고,,)

--> 나중에 AC받은 코드를 참고하니까, 이런거 신경 안쓰고 짠 일반적인 dfs코드가 AC를 받은걸 확인하고, 생각한 대로 코딩을 시작했다.

 

결국 결론을 정리하면 블록은 5종류가 있는데, 4종류는 dfs로 블록의 모형을 구현할수 있고, 1 종류(ㅗ) 는 우리가 직접 4가지의 경우를 모두 (ㅗ ㅏ ㅜ ㅓ) 돌려보면서 제일 큰 값을 update하면서 코드를 진행 시켜야 한다.

 

dfs 구현하는 것은 일반적인 수준과 크게 다르지 않다. 

 

코드

#include <iostream>
using namespace std;
int n,m;
int map[501][501];
int answer=0;
bool visited[501][501];
 
int dy[4] = {1,-1,0,0};
int dx[4] = {0,0,1,-1};
 
int dfs(int y, int x, int blocknum){
    if(blocknum==4)
        return map[y][x];
    int res=0;
    for(int i=0; i<4; i++){
        int ny = y + dy[i];
        int nx = x + dx[i];
        if(nx>=0 && nx<=m-1 && ny>=0 && ny<=n-1){
            if(!visited[ny][nx]){
                visited[ny][nx]=true;
                res = max(res,map[y][x] + dfs(ny,nx,blocknum+1));
                visited[ny][nx]=false;
            }
        }
    }
    return res;
}
 
int brute(int y, int x){
    int res=0;
    //ㅗ
    if(y>=1 && x>=1 && x<m-1)
        res = max(res,map[y][x] + map[y][x-1] + map[y][x+1] + map[y-1][x]);
    //ㅏ
    if(y>=1 && x<m-1 && y<n-1)
        res = max(res,map[y][x] + map[y-1][x] + map[y+1][x] + map[y][x+1]);
    //ㅜ
    if(y<n-1 && x>=1 && x<m-1)
        res = max(res, map[y][x] + map[y][x-1] + map[y][x+1] + map[y+1][x]);
    //ㅓ
    if(y>=1 && x>=1 && y<n-1)
        res = max(res,map[y][x] + map[y-1][x] + map[y+1][x] + map[y][x-1]);
    return res;
}
 
int main(void){
    cin>>n>>m;
    for(int i=0;i<n;i++)
        for(int j=0; j<m; j++)
            cin>>map[i][j];
 
    for(int i=0; i<n;i++){
        for(int j=0; j<m;j++){
            visited[i][j]=true;
            answer = max(answer,dfs(i,j,1));
            answer = max(answer,brute(i,j));
            visited[i][j]=false;
        }
    }
    cout<<answer;