본문 바로가기

알고리즘

파이썬) 듣보잡

https://www.acmicpc.net/problem/1764

 

1764번: 듣보잡

첫째 줄에 듣도 못한 사람의 수 N, 보도 못한 사람의 수 M이 주어진다. 이어서 둘째 줄부터 N개의 줄에 걸쳐 듣도 못한 사람의 이름과, N+2째 줄부터 보도 못한 사람의 이름이 순서대로 주어진다.

www.acmicpc.net

 

n, m = map(int, input().split())

dic = {}

arr = []
arr2 = []
for _ in range(n):
    arr.append(input())

for _ in range(m):
    arr2.append(input())

inter = list(set(arr) & set(arr2))

print(len(inter))
inter.sort()
for i in inter:
    print(i)

 

또는 

n, m = map(int, input().split())

dic = {}

arr = []

for i in range(n):
    s = input()

    dic[s] = i

for i in range(m):
    s = input()

    if s in dic:
        arr.append(s)


arr.sort()
print(len(arr))
for i in arr:
    print(i)

위 두 코드 유효한 코드이다.

개인적으로 셋트가 더 짧지만 아직 셋은 익숙하지가 않다.

'알고리즘' 카테고리의 다른 글

파이썬) 벽 부수고 이동하기 - BOJ  (0) 2023.12.10
파이썬) 2 X N 타일링 - BOJ  (0) 2023.12.07
파이썬) 차집합 - BOJ  (0) 2023.12.06
파이썬) 상자넣기 - BOJ  (0) 2023.12.05
파이썬) 파도반 수열 - BOJ  (0) 2023.12.04