백준 5052번 전화번호 목록 (Python)

문제

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

풀이

시간초과 주의!

input() 쓰지말자.

import sys
input = sys.stdin.readline
 
def solve(contacts) :  
 
    contacts.sort()
 
    st = contacts[0]
 
    for contact in contacts[1:] :
        if st in contact :
            return "NO" 
        else :
            st = contact
            
    return "YES"      
 
for i in range(int(input())) :
 
    n = int(input())
    contacts = []   
 
    for j in range(n) :        
        contacts.append(input().rstrip())
 
    ans = solve(contacts)    
    print(ans)