[Python] 백준 25192번

25192. 인사성 밝은 곰곰이 (silver 4)

 

 

입력

첫 번째 줄에는 채팅방의 기록 수를 나타내는 정수 N 이 주어진다. (1≤N≤100000)

두 번째 줄부터 N개의 줄에 걸쳐 새로운 사람의 입장을 나타내는 ENTER, 혹은 채팅을 입력한 유저의 닉네임이 문자열로 주어진다.

(1≤문자열 길이≤20)

첫 번째 주어지는 문자열은 무조건 ENTER이다.

 

출력

채팅 기록 중 곰곰티콘이 사용된 횟수를 출력하시오.

 

예제 입력1

9
ENTER
pjshwa
chansol
chogahui05
lms0806
pichulia
r4pidstart
swoon
tony9402

예제 출력1

8

 

 

 

풀이)

gom이라는 집합을 선언해주었는데, 중복을 피하기위해 집합으로 선언해주었다.

그리고 곰곰티콘을 사용하는 횟수인 cnt를 0으로 저장해주었다.

for문을 사용해서 입력받은 채팅방의 기록 수인 n만큼 for문을 통해 반복해주었다.

record에 문자열을 입력해주었고, if문을 통해 record가 ENTER인 경우, ENTER이 아닌 경우로 나눠서 코드를 작성해주었다.

ENTER인 경우에는 clear를 사용해서 집합을 초기화해주었고, ENTER이 아닌 경우에는 처음 채팅을 입력하는 사람, 즉 gom 집합에 없는 사람이 있으면 cnt를 1 증가시키도록 해주었고, gom에 해당 record를 추가해주었다.

마지막으로 곰곰티콘의 사용 횟수인 cnt를 출력해주면 된다.

def welcome(n):
    gom = set()
    cnt = 0
    for i in range(n):
        record = input()
        # ENTER인 경우
        if record == 'ENTER':
            gom.clear()   # 초기화
        # ENTER이 아닌 경우
       	elif record != 'ENTER':
            if record not in gom:  # 처음 채팅 입력하는 사람
                cnt += 1           # 곰곰티콘 사용
                gom.add(record)  
    return cnt

n = int(input())
print(welcome(n))  # n 입력받음