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

백준 5883 - 아이폰 9S

시간 제한 1초, 메모리 제한 128MB

# 조건

• 오늘은 애플의 아이폰 9S가 출시되는 날이다.
• N(1 ≤ N ≤ 1000)명의 사람들은 새 아이폰을 누구보다 먼저 구매하기 위해서 애플 스토어 앞에 한 줄로 서있다.
• 아이폰 9S는 구매자가 용량을 마음대로 정할 수 있다.
  • 즉, 지금까지 아이폰은 16/32/64GB와 같이 용량의 크기가 미리 정해져 있었다.
• 하지만, 9S는 자신이 원하는 용량의 크기 B(i)를 점원에게 말하면, 정확하게 그 용량을 가진 아이폰 9S를 그 자리에서 만들어 구매하는 방식이다.
• 애플 스토어의 점원 상근이는 같은 용량을 원하는 사람들이 연속되어 있으면, 더 보기 좋을 것이라고 생각한다.
• 따라서, 특정 사람을 고르고, 그 사람이 원하는 용량을 원하는 사람을 줄에서 모두 빼버리려고 한다.
• 어떤 용량을 원하는 사람을 줄에서 빼 버리면, 같은 용량을 원하는 사람들이 연속되어 있는 구간의 길이중 가장 긴 값이 최대가 되는지 구하는 프로그램을 작성하시오.

입력

• 첫째 줄에 N이 주어진다.
• 둘째 줄부터 N개 줄에는 줄을 서 있는 순서대로 각 사람이 원하는 용량 B(i)가 주어진다.
• B(i)는 0보다 크거나 같, 1,000,000보다 작거나 같은 정수이다.
• 항상 두 개 이상의 서로 다른 B(i)가 존재한다.

출력

• 상근이가 만들 수 있는 가장 길이가 긴 같은 용량을 원하는 사람의 연속 구간의 길이를 출력한다.

# 접근 방법

• 브루트포스를 활용하여 풀어준다.
• 우선 주어진 숫자들을 check라는 set 자료구조에 담아준다.
• check를 순회하며 하나의 숫자씩을 제외하는 반복문과 i 변수에 0을 할당해준 후 1부터 N까지 순회하는 반복문을 실행해준다.
  • 현재 c와 nums[j]가 같다면 cotinue
  • nums[i]와 nums[j]가 같다면 cnt += 1, result 갱신
  • 아니라면 cnt = 1
  • 이후, i = j를 통해 i를 변경해준다.

import sys  
sys.stdin = open('input.txt')  
input = sys.stdin.readline  

N = int(input())  
nums = []  
for _ in range(N):  
    nums.append(int(input()))  

check = set(nums)  
result = 1  
for c in check:  
    cnt = 1  
    i = 0  
    for j in range(1, N):  
        if nums[j] == c:  
            continue  
        if nums[i] == nums[j]:  
            cnt += 1  
            result = max(result, cnt)  
        else:  
            cnt = 1  

        i = j  
print(result)