글 읽기 - Python 질문입니다

shpark120614 1년 전 1

브루트 포스를 했습니다. 로직은 다음과 같습니다

사탕을 2차원 배열로 입력받고 Transpose 한 것도 생성하여 둘을 같이 for문을 써서 위 아래로 사탕이 같으면 넘어가고 아니면 바꾼 뒤에 사탕을 최대로 먹어주는 것을 계산하여 최대인 것을 m에 넣어주고 마지막에 m을 출력하도록 하였습니다.

질문의 반례를 전부 넣어봐도 찾지 못하겠습니다. 도와주세요..

from sys import stdin

input = stdin.readline

def cnt(s) :
    tmp = 1
    m = 0
    for i in range(1, n) :
        if s[i-1] == s[i] :
            tmp += 1
        else :
            m = max(tmp, m)
            tmp = 1
    m = max(tmp, m)
    return m
        

def solution() :
    m = 0

    for i in range(n-1) :
        for j in range(n) :
            chk = lst[i][j] != lst[i+1][j]
            r_chk = r_lst[i][j] != r_lst[i+1][j]
            
            if chk :
                lst[i][j], lst[i+1][j] = lst[i+1][j], lst[i][j]
                for k in lst :
                    m = max(cnt(k), m)
                lst[i][j], lst[i+1][j] = lst[i+1][j], lst[i][j]
            
            if r_chk :
                r_lst[i][j], r_lst[i+1][j] = r_lst[i+1][j], r_lst[i][j]
                for k in r_lst :
                    m = max(cnt(k), m)
                r_lst[i][j], r_lst[i+1][j] = r_lst[i+1][j], r_lst[i][j]

    print(m)

n = int(input())
lst = [list(input().rstrip()) for _ in range(n) ]
r_lst = []

for i in range(n) :
    tmp = []
    for j in range(n) :
        tmp += lst[j][i]
    r_lst.append(tmp)

solution()

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

Python 질문입니다

shpark120614 1년 전 1

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말