[Java] 백준 3085 사탕 게임

문제

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

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

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

 

입력

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

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

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

 

출력

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

 

문제 링크 : 사탕 게임

---

설명

문제를 읽고나서 예제를 봤는데 이해가 잘 안돼서 질문 게시판에 가봤더니 나 같은 사람이 많았다.

일단 위 문제에서 사탕 교환은 한번밖에 할 수 없다.

한번하고 나서 최대값 봐주고 이런 식으로 나는 문제를 풀었다.

 

요즘 문제를 읽는데 이해가 잘 안될 때가 많은 것 같다. ㅠㅠ 책을 꽤 좋아하는 편인데 독해력이 많이 안좋나보다

 

내가 구현한 방식은 candy라는 배열에 처음 사탕을 저장해 놓고, copy라는 배열에 위치를 바꾼 사탕을 넣어줬다.

근데 다른 사람들이 푼 방식을 보니 배열 하나로 충분히 가능했고, 다시 생각해보니 충분히 가능한 방식이였다.

 

swap이라는 메서드에서 행, 열 별로 연속 먹을 수 있는 같은 사탕 개수를 카운트 해줬다. (이 부분에서 처음에 이해를 잘못했기에 그냥 해당 행, 열 별 같은 색상 사탕 개수를 카운트한... 근데 예제는 다 맞아서 더 슬퍼...)

 

그리고 중간 값이 제일 최대일 수도 있으니 사탕 색이 달라지는 부분에서 현재 연속 먹을 수 있는 사탕 개수와 ans 중 최댓값 저장

 

탐색을 다 끝냈다면 마지막에서 다시 원본 배열로 돌려놓기.

이 부분도 사실 반복 다 돌거 없이 해당 두 부분만 해주면 좋다^^

(나는 다 돌아서 하는 말)

 

코드

import java.io.*;
import java.util.*;

public class Main {
    static String[][] candy, copy;
    static int ans, n;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        n = Integer.parseInt(br.readLine());
        candy = new String[n][n];
        copy = new String[n][n];
        ans = 0;
        for (int i = 0; i < n; i++) {
            String[] s = br.readLine().split("");
            for (int j = 0; j < n; j++) {
                candy[i][j] = s[j];
                copy[i][j] = candy[i][j];   // 그대로 복사
            }
        }

        swap();	// 교환 전 최대 사탕수 확인
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n - 1; j++) {
                copy[i][j] = candy[i][j + 1];	// 현재 열과 다음 열 교환
                copy[i][j + 1] = candy[i][j];
                swap();
                copy[j][i] = candy[j + 1][i];	// 현재 행과 다음 행 교환
                copy[j + 1][i] = candy[j][i];
                swap();
            }
        }
        System.out.println(ans);
    }

    static void swap() {
        int count = 1;	// 무조건 하나의 사탕 존재

        for (int i = 0; i < n; i++) {     // 현재 행에서 먹을 수 있는 연속된 같은 사탕의 최대 개수
            count = 1;
            for (int j = 0; j < n - 1; j++) {
                if (copy[i][j].equals(copy[i][j + 1])) {
                    count++;
                } else {
                    ans = Math.max(ans, count);
                    count = 1;
                }
            }
            ans = Math.max(ans, count);
        }
        for (int i = 0; i < n; i++) {     // 현재 열에서 먹을 수 있는 연속된 같은 사탕의 최대 개수
            count = 1;
            for (int j = 0; j < n - 1; j++) {
                if (copy[j][i].equals(copy[j + 1][i])) {
                    count++;
                } else {
                    ans = Math.max(ans, count);
                    count = 1;
                }
            }
            ans = Math.max(ans, count);	// 마지막까지 저장된 개수 비교
        }
        // 다시 되돌리기
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                copy[i][j] = candy[i][j];
            }
        }
    }
}