[코드포스 CODEFORCES] 4C Registration system 풀이 코드 (Python)

코드포스 4C Registration system 풀이 코드 | Python (파이썬)

---

문제 해설

문자열들을 순서대로 기록하며 새로운 문자열일 땐 'OK'를, 중복된 문자열일 땐 문자열 끝에 오름차순으로 숫자를 붙여 출력하는 문제입니다.

풀이

문자열을 키로, 입력 횟수를 값으로 가지는 딕셔너리를 선언하고 새로 들어오는 문자열이 딕셔너리에 없을 경우 'OK'를 출력, 딕셔너리에 있을 경우 해당 문자열의 중복 횟수를 1회 늘린 후 중복 횟수와 함께 출력합니다.

코드

arr = {}

n = int(input())
for i in range(n):
    temp = input()
    if temp in arr:
        arr[temp] += 1;
        print(temp+str(arr[temp]))
    else:
        arr[temp] = 0;
        print("OK")
    

문제 출처

https://codeforces.com/contest/4/problem/C