16932번 모양 만들기

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

저번에 푼 코드 : https://hsdevelopment.tistory.com/254

공부하기 좋고, 복습하기 좋은 DFS 문제여서 다시 풀어봤다.

다시 또 풀어볼 생각이다.

저번에 푼 코드에서는 필요 없는 함수, 변수들이 있었다.

또한, 다시 풀었음에도 불구하고,

group 배열 크기를 1010으로 했었다.

연결 요소의 개수는 1010 * 1010 크기의 배열에서 1010 이상일 수 있기 때문이다.

#include <iostream>
#include <queue>
#include <stack>
#include <cstdio>
#include <vector>
#include <cstring>
#include <string>
#include <math.h>
#include <algorithm>
#include <map>
#include <set>
#include <sstream>
#include <tuple>
 
#pragma warning(disable:4996)  
#pragma comment(linker, "/STACK:336777216")
 
using namespace std;
 
typedef struct info
{
    int value;
    int group;
}info;
 
int N = 0, M = 0, g[1010 * 1010] = { 0, }; // 그룹 배열 크기를 1010으로 했었다.
int dy[4] = { 0, 1, 0, -1 };
int dx[4] = { 1, 0, -1, 0 };
 
info Map[1010][1010], temp;
 
bool visit[1010][1010] = { false, };
 
int DFS(int r, int c, int group)
{
    int ret = 1;
 
    visit[r][c] = true;
    Map[r][c].group = group;
 
    for (int i = 0; i < 4; i++)
    {
        int y = r + dy[i];
        int x = c + dx[i];
 
        if (y < 0 || y >= N || x < 0 || x >= M || visit[y][x] || Map[y][x].value == 0) continue;
 
        ret += DFS(y, x, group);
    }
 
    return ret;
}
 
int main(void)
{
    int n = 0, group = 1, ans = 0;
 
    scanf("%d %d", &N, &M);
 
    for (int i = 0; i < N; i++)
        for (int j = 0; j < M; j++)
        {
            scanf("%d", &n);
 
            temp.value = n;
            temp.group = 0;
 
            Map[i][j] = temp;
        }
 
    for (int i = 0; i < N; i++)
        for (int j = 0; j < M; j++)
        {
            if (!visit[i][j] && Map[i][j].value == 1)
            {
                g[group++] = DFS(i, j, group); 
                /*DFS(i, j, group++); 쓰면 안 된다. 
                DFS 함수가 먼저 실행되기 때문에 g 배열에 저장할 때 이미 group 변수는 1 증가.*/
            }
        }
    
    for (int i = 0; i < N; i++)
        for (int j = 0; j < M; j++)
        {
            if (Map[i][j].value == 1) continue;
 
            int sum = 0;
 
            vector<int> v;
 
            for (int k = 0; k < 4; k++)
            {
                int y = i + dy[k];
                int x = j + dx[k];
 
                bool stop = false;
 
                if (y < 0 || y >= N || x < 0 || x >= M || Map[y][x].value == 0) continue;
 
                for (auto a : v)
                    if (a == Map[y][x].group)
                    {
                        stop = true;
                        break;
                    }            
 
                if (stop) continue;
 
                v.push_back(Map[y][x].group);
            }
 
            for (auto a : v) sum += g[a];
 
            ans = max(ans, sum + 1);
        }
    
    printf("%d\n", ans);    
 
    return 0;
}