728x90
반응형
문제 바로가기
문제:
전화번호 목록이 주어진다. 이때, 이 목록이 일관성이 있는지 없는지를 구하는 프로그램을 작성하시오. 전화번호 목록이 일관성을 유지하려면, 한 번호가 다른 번호의 접두어인 경우가 없어야 한다. 예를 들어, 전화번호 목록이 아래와 같은 경우를 생각해보자
- 긴급전화: 911
- 상근: 97 625 999
- 선영: 91 12 54 26
이 경우에 선영이에게 전화를 걸 수 있는 방법이 없다. 전화기를 들고 선영이 번호의 처음 세 자리를 누르는 순간 바로 긴급전화가 걸리기 때문이다. 따라서, 이 목록은 일관성이 없는 목록이다.
입력:
첫째 줄에 테스트 케이스의 개수 t가 주어진다. (1 ≤ t ≤ 50) 각 테스트 케이스의 첫째 줄에는 전화번호의 수 n이 주어진다. (1 ≤ n ≤ 10000) 다음 n개의 줄에는 목록에 포함되어 있는 전화번호가 하나씩 주어진다. 전화번호의 길이는 길어야 10자리이며, 목록에 있는 두 전화번호가 같은 경우는 없다.
출력:
각 테스트 케이스에 대해서, 일관성 있는 목록인 경우에는 YES, 아닌 경우에는 NO를 출력한다.
풀이
시도1. 조합으로 2개 뽑아서 비교하기
import sys
from itertools import combinations
t = int(input())
for _ in range(t):
n = int(input())
phone = sorted([input().split() for _ in range(n)], key = len)
flag = True
lst = list(combinations(phone, 2))
for x,y in lst:
if y[0].startswith(x[0]):
flag = False
print("YES" if flag else "NO")
문자열 길이로 정렬하고 2개 뽑는 조합을 만든후, startswith로 특정문자로 시작하는지 비교했다.
특정문자로 시작하면 911로 전화되는것처럼 falg = false로 된다고 생각했기 때문이다.
>> 메모리초과
정답.
import sys
sys.stdin = open('/Users/song/Desktop/Python/Python/h.txt', 'r')
t = int(sys.stdin.readline())
for _ in range(t):
n = int(sys.stdin.readline())
phone = sorted([sys.stdin.readline().split() for _ in range(n)])
flag = True
for i in range(n-1): # 바로 다음꺼랑 비교
if phone[i+1][0].startswith(phone[i][0]):
flag = False
break
print("YES" if flag else "NO")
문자열 기준으로 정렬을 하면
바로 다음 문자열 비슷할 것이므로 바로 다음것이랑만 비교하면 된다고 생각했다.
나머지는 동일하다.
반응형
댓글