문제
상근이는 어렸을 적에 "봄보니 (Bomboni)" 게임을 즐겨했다.
가장 처음에 N×N크기에 사탕을 채워 놓는다. 사탕의 색은 모두 같지 않을 수도 있다. 상근이는 사탕의 색이 다른 인접한 두 칸을 고른다. 그 다음 고른 칸에 들어있는 사탕을 서로 교환한다. 이제, 모두 같은 색으로 이루어져 있는 가장 긴 연속 부분(행 또는 열)을 고른 다음 그 사탕을 모두 먹는다.
사탕이 채워진 상태가 주어졌을 때, 상근이가 먹을 수 있는 사탕의 최대 개수를 구하는 프로그램을 작성하시오.
입력
첫째 줄에 보드의 크기 N이 주어진다. (3 ≤ N ≤ 50)
다음 N개 줄에는 보드에 채워져 있는 사탕의 색상이 주어진다. 빨간색은 C, 파란색은 P, 초록색은 Z, 노란색은 Y로 주어진다.
사탕의 색이 다른 인접한 두 칸이 존재하는 입력만 주어진다.
출력
첫째 줄에 상근이가 먹을 수 있는 사탕의 최대 개수를 출력한다.
코드
def change(N):
# 가로 바꾸기
for i in range(N):
for j in range(1,N):
BRD[i][j-1], BRD[i][j] = BRD[i][j], BRD[i][j-1]
check(N)
# 원상태로 복구
BRD[i][j], BRD[i][j-1] = BRD[i][j-1], BRD[i][j]
# 세로 바꾸기
for i in range(N):
for j in range(1,N):
BRD[j-1][i], BRD[j][i] = BRD[j][i], BRD[j-1][i]
check(N)
BRD[j][i], BRD[j-1][i] = BRD[j-1][i], BRD[j][i]
def check(N):
global max_cnt
cnt_list = []
# garo
for i in range(N):
cnt = 1
for j in range(1, N):
if BRD[i][j] == BRD[i][j-1]:
cnt +=1
else:
if max_cnt < cnt:
max_cnt = cnt
cnt = 1
if max_cnt < cnt:
max_cnt = cnt
# sero
for i in range(N):
cnt2 = 1
for j in range(1,N):
if BRD[j][i] == BRD[j-1][i]:
cnt2 += 1
else:
if max_cnt < cnt2:
max_cnt = cnt2
cnt2 = 1
if max_cnt < cnt2:
max_cnt = cnt2
N = int(input())
BRD = [list(input()) for _ in range(N)]
max_cnt = 0
change(N)
print(max_cnt)
풀이
change() 함수 안 반복문에서 가로로, 세로로 색을 바꾸고, check() 함수로 제일 긴 문자열 길이를 구해서 글로벌 변수인 max_cnt에 저장하였습니다. 색을 한 번 바꾼 뒤 다시 원상태로 복구해야 주어야 하고, 인덱스의 범위를 주의하여야 합니다
'Algorithm' 카테고리의 다른 글
[PROGRAMMERS] 추석 트래픽 (python, js) (0) | 2021.07.26 |
---|---|
[PROGRAMMERS] 야근 지수 - heap (0) | 2021.07.26 |
[PROGRAMMERS] 입국심사 (0) | 2021.07.22 |
[Programmers] 가장 먼 노드 (0) | 2021.07.15 |