[백준] 5052 - 전화번호 목록 / Python / 골드 4

🚩문제 주소

📄풀이

정렬

입력 받은 전화번호 배열을 정렬합니다.
정렬하면 현재 전화번호와 다음 전화번호를 비교 하기만 하면 됩니다.

두 전화번호를 비교하는 check 함수를 생성해서 앞의 전화번호가 뒤의 전화번호에 속한다면 False, 전부 통과한다면 True를 반환합니다.

check 함수가 True라면 YES, False라면 NO를 출력합니다.

👨‍💻코드

import sys

def check():
  for i in range(n-1):
    if phoneNumbers[i] in phoneNumbers[i+1][:len(phoneNumbers[i+1])-1]:
      return False
  return True

t = int(sys.stdin.readline())
for _ in range(t):
  n = int(sys.stdin.readline())
  phoneNumbers = [sys.stdin.readline().rstrip() for _ in range(n)]
  phoneNumbers.sort()

  if check():
    print('YES')
  else:
    print('NO')