코딩복습장
[백준] 5052번(python 파이썬) - 전화번호 목록 본문
문제링크: https://www.acmicpc.net/problem/5052
시간 제한 | 메모리 제한 | 제출 | 정답 | 맞힌 사람 | 정답 비율 |
1 초 | 256 MB | 37304 | 11821 | 7261 | 29.904% |
문제
전화번호 목록이 주어진다. 이때, 이 목록이 일관성이 있는지 없는지를 구하는 프로그램을 작성하시오.
전화번호 목록이 일관성을 유지하려면, 한 번호가 다른 번호의 접두어인 경우가 없어야 한다.
예를 들어, 전화번호 목록이 아래와 같은 경우를 생각해보자
- 긴급전화: 911
- 상근: 97 625 999
- 선영: 91 12 54 26
이 경우에 선영이에게 전화를 걸 수 있는 방법이 없다. 전화기를 들고 선영이 번호의 처음 세 자리를 누르는 순간 바로 긴급전화가 걸리기 때문이다. 따라서, 이 목록은 일관성이 없는 목록이다.
입력
첫째 줄에 테스트 케이스의 개수 t가 주어진다. (1 ≤ t ≤ 50) 각 테스트 케이스의 첫째 줄에는 전화번호의 수 n이 주어진다. (1 ≤ n ≤ 10000) 다음 n개의 줄에는 목록에 포함되어 있는 전화번호가 하나씩 주어진다. 전화번호의 길이는 길어야 10자리이며, 목록에 있는 두 전화번호가 같은 경우는 없다.
출력
각 테스트 케이스에 대해서, 일관성 있는 목록인 경우에는 YES, 아닌 경우에는 NO를 출력한다.
풀이
이 문제는 정렬의 특성을 활용한 문제이다. 이 부분만 잘 생각해낸다면 빠르게 풀 수 있다.
문제는 다음과 같다.
111
111222
의 상황일 경우 어느 한 번호가 다른 번호에 포함되면 NO를 출력해야 된다.
이 조건에 해당되지 않는다면 YES를 출력하면 된다.
처음에는 모든 번호를 전부 탐색해야되는 문제인줄 알았다. 하지만 모두 탐색하기에는 번호의 개수가 최대 100000개 이고
모두 탐색하는 경우 시간복잡도가 O(n^2)이므로 불가능 했다. 그래서 for문을 한번 쓰는 방식을 생각해봤다.
정렬을 하는 경우를 잘 생각해봤을 때 문자열 정렬은 문자가 숫자일때 앞자리 숫자가 작은 수 부터 정렬이 된다( 오름차순 )
여기서 한 번호가 다른 번호에 포함될 수 있는 조건에 대해 생각해봤다.
한 번호가 다른 번호에 포함되려면 두 번호는 같은 번호거나 다른번호가 포함되는 번호보다 길어야한다.
같은번호는 있으면 안되기 때문에 후자의 조건에 해당되어야한다.
-> 앞자리 번호가 같은 상황에서 해당 번호를 포함하는 번호는 길이가 더 길기때문에 크기가 더 크다.
따라서 정렬을 할 경우 포함되는 숫자보다 뒤에 존재한다.
그리고 번호를 포함하는 번호가 하나라도 있을 경우 바로 NO라는 답이나오기 때문에 for문을 한번쓰면서 i번째 번호와
i+1번째 번호를 비교하면 답이 나온다.-> 시간복잡도 O(N)
i번째 번호와 i+1번째 번호를 비교하는 이유는 앞자리가 같은 번호가 있을 경우 포함하는 번호는 포함되는 번호의 바로 뒤에 존재하기 때문이다. ( 물론 포함하는 번호가 1개일 경우)
여러 예시를 들며 생각해보면 답이 나올것이다.
이상 오늘의 포스팅 끝!
구현 코드
t = int(input())
res = []
for _ in range(t):
n = int(input())
phonebook = []
for _ in range(n):
phonebook.append(input())
phonebook.sort() # 사전 순으로 정렬
for i in range(n - 1):
if phonebook[i] == phonebook[i + 1][:len(phonebook[i])]: #포합되는 수가 있는지 확인
res.append('NO')
break
else: res.append('YES')
for result in res: # 결과 출력
print(result)
'코딩 테스트 > python(파이썬)' 카테고리의 다른 글
[백준] 14499번(python 파이썬) - 주사위 굴리기 (0) | 2023.06.24 |
---|---|
[백준] 1010번(python 파이썬) - 다리 놓기 (0) | 2023.03.31 |
[백준] 1946번(python 파이썬) - 신입사원 (0) | 2023.03.23 |
[백준] 1018번(python 파이썬) - 체스판 다시 칠하기 (0) | 2023.03.16 |
[백준] 1005번(python 파이썬) - ACM Craft (3) | 2023.03.12 |