16932번 모양 만들기

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

지난 번 코드 : https://hsdevelopment.tistory.com/265

다시 풀어보았다.

이번에는 map으로 풀어보았다.

#include <bits/stdc++.h>
 
using namespace std;
 
bool visit[1001][1001] = { false, };
 
int N = 0, M = 0, g[1001 * 1001] = { 0, }, Map[1001][1001] = { 0, };
 
int dy[4] = { 0, 1, 0, -1 };
int dx[4] = { 1, 0, -1, 0 };
 
vector<pair<int, int> > v;
 
int dfs(int r, int c)
{
    int ret = 1;
 
    v.push_back({ r,c });
 
    visit[r][c] = true;
 
    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]) continue;
 
        ret += dfs(y, x);
    }
 
    return ret;
}
 
int main(void)
{
    int ans = 0;
 
    scanf("%d %d", &N, &M);
 
    for (int i = 0; i < N; i++)
        for (int j = 0; j < M; j++)
            scanf("%d", &Map[i][j]);
 
    int idx = 1;
 
    for (int i = 0; i < N; i++)
        for (int j = 0; j < M; j++)
            if (!visit[i][j] && Map[i][j])
            {
                v.clear();
                g[idx] = dfs(i, j);
                for (auto a : v) Map[a.first][a.second] = idx;
                ++idx;
            }
 
    for (int i = 0; i < N; i++)
        for (int j = 0; j < M; j++)    
            if (Map[i][j] == 0)
            {
                int sum = 0;
 
                map<int, int> m;
 
                for (int k = 0; k < 4; k++)
                {
                    int y = i + dy[k];
                    int x = j + dx[k];
 
                    if (y < 0 || y >= N || x < 0 || x >= M || !Map[y][x]) continue;
                    if (m.count(Map[y][x])) continue;
 
                    m[Map[y][x]] = 1;
        
                    sum += g[Map[y][x]];
                }
 
                ans = max(ans, sum + 1);
            }
 
    printf("%d\n", ans);
 
    return 0;
}