PS/Python

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

yoo.o 2020. 8. 26. 17:58
반응형

문제

 

5052번: 전화번호 목록

문제 전화번호 목록이 주어진다. 이때, 이 목록이 일관성이 있는지 없는지를 구하는 프로그램을 작성하시오. 전화번호 목록이 일관성을 유지하려면, 한 번호가 다른 번호의 접두어인 경우가 없�

www.acmicpc.net

풀이

프로그래머스에서 풀었던것(https://yuuj.tistory.com/24)과 출력 유형만 다르고 동일한 문제다. 

스트링으로 받아와서 정렬 후, 바로 뒤에 애랑만 startswith()를 사용해서 비교를 해줬다.

import sys

TC = int(sys.stdin.readline())
for _ in range(TC):

    li = []
    flag = 0

    n = int(sys.stdin.readline())
    for _ in range(n):
        li.append(sys.stdin.readline().strip())

    li.sort()
    flag = 0
    
    for i in range(len(li)-1):
        if li[i+1].startswith(li[i]):
            print("NO")
            flag = 1
            break

    if flag == 0:
        print("YES")

 

반응형