[Python/파이썬] 백준 5883번 아이폰 9S

5883번: 아이폰 9S

 

코드

n = int(input())
b = [int(input()) for _ in range(n)]


def ls(arr, r):
    prev = -1
    res = -1
    tmp = 0
    for i in range(n):
        if prev == arr[i]:
            tmp += 1
        elif arr[i] == r:
            continue
        else:
            prev = arr[i]
            res = max(res, tmp)
            tmp = 1
    res = max(res, tmp)
    return res


ans = -1
for r in set(b):
    ans = max(ans, ls(b, r))
print(ans)

 

ls() 함수는 파라미터로 받은 arr 배열에서 값이 r인 원소들을 제외하고 가장 긴 연속 구간의 길이를 반환하는 함수이다.

이 함수에 파라미터로 넣을 r은 b의 집합으로 b에 존재하는 용량의 종류들을 구하여 하나씩 넣어보면 된다. 하나씩 모두 해봐서 가장 긴 연속 구간을 구하면 된다.