[백준][Python] 5052번 전화번호 목록

5052번: 전화번호 목록

 

 

 

코드

import sys
t = int(input())
for _ in range(t):
	n_list = []
	flag = 0
	n = int(input())
	for _ in range(n):
		n_list.append(sys.stdin.readline().rstrip())
	n_list.sort()
	for i in range(n-1):
		if n_list[i+1][:len(n_list[i])] == n_list[i]:
			flag = -1
			break
	if flag == -1:
		print("NO")
	else:
		print("YES")

 

설명

[911, 97625999, 91125426]의 경우 정렬을 시키면 [911, 91125426, 97625999]가 된다.

따라서 입력받은 문자열을 정렬시킨 후 앞,뒤의 문자열끼리만 비교를 해주면 된다.