개발 블로그

문제 링크 : www.acmicpc.net/problem/3085

문제

상근이는 어렸을 적에 "봄보니 (Bomboni)" 게임을 즐겨했다.

가장 처음에 N×N크기에 사탕을 채워 놓는다. 사탕의 색은 모두 같지 않을 수도 있다. 상근이는 사탕의 색이 다른 인접한 두 칸을 고른다. 그 다음 고른 칸에 들어있는 사탕을 서로 교환한다. 이제, 모두 같은 색으로 이루어져 있는 가장 긴 연속 부분(행 또는 열)을 고른 다음 그 사탕을 모두 먹는다.

사탕이 채워진 상태가 주어졌을 때, 상근이가 먹을 수 있는 사탕의 최대 개수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 보드의 크기 N이 주어진다. (3 ≤ N ≤ 50)

다음 N개 줄에는 보드에 채워져 있는 사탕의 색상이 주어진다. 빨간색은 C, 파란색은 P, 초록색은 Z, 노란색은 Y로 주어진다.

사탕의 색이 다른 인접한 두 칸이 존재하는 입력만 주어진다.

출력

첫째 줄에 상근이가 먹을 수 있는 사탕의 최대 개수를 출력한다.

접근 방법

처음에는 DFS 처럼 상하좌우로 swap 하고 사탕 최대 개수를 행, 열에 대해 각각 count 하는 것인줄 알았다.

그러나 DFS가 아니라 가로에 대해 한 번 swap 하고 행, 열 각각 사탕 최대 개수를 세고 이를 반복한다.

그 다음 세로에 대해 한 번 swap 하고 행, 열에 대해 사탕 최대 개수를 센다.

즉,

1) 가로

map[0][0] 와 map[0][1] swap -> countRow() -> countCol() -> map[0][1] 과 map[0][2] swap -> countRow() -> countCol()......

2) 세로

map[0][0] 와 map[1][0] swap -> countRow() -> countCol() -> map[1][0] 과 map[2][0] swap -> countRow() -> countCol()......

과 같은 순서로 탐색하면 된다.

주의할 점은 cnt를 초기화해주는 곳이었다.

사탕은 같은 행 또는 열에 있어야 먹을 수 있으므로 행 또는 열이 바뀔 때마다 cnt를 1로 초기화 해줘야 한다.

#include <iostream>
#include <algorithm>

using namespace std;

char map[50][50];


int N;
int resMax = 0;

void countRow(){

    for(int i = 0; i<N; i++){
        int cnt=1;

        for(int j = 0; j<N-1; j++){
            if(map[i][j]==map[i][j+1]){
                cnt++;
            }
            else{
                resMax=max(resMax,cnt);
                cnt=1;
            }
        }
        resMax=max(resMax,cnt);
    }
}

void countCol(){
    
    for(int j = 0; j<N; j++){
        int cnt=1;

        for(int i = 0; i<N-1; i++){
            if(map[i][j]==map[i+1][j]){
                cnt++;
            }
            else{
            resMax=max(resMax,cnt);
            cnt=1;
            }
        }
        resMax=max(resMax,cnt);
    }
}

void swapFunc(int r,int c,int nr,int nc){
    int tmp = map[r][c];
    map[r][c] = map[nr][nc];
    map[nr][nc] = tmp;
}


void solve(int r, int c){
    countRow();	    //행 검사

    countCol();		    //열검사
}



int main(){
    
    cin >> N;
    
    for(int i = 0; i<N; i++){
        for(int j = 0; j<N; j++){
            cin >> map[i][j];
        }
    }
    
    
    for(int i = 0; i<N; i++){
        for(int j = 0; j<N-1; j++){
            swapFunc(i,j,i,j+1);
            solve(i,j);             //가로 스왑
            swapFunc(i,j,i,j+1);
        }
    }
    
    
   for(int j = 0; j<N; j++){
       for(int i = 0; i<N-1; i++){
           swapFunc(i,j,i+1,j);
           solve(i,j);               //세로 스왑
           swapFunc(i,j,i+1,j);
       }
   }

    cout<<resMax<<"\n";
    
    return 0;
}