[프로그래머스] 전화번호 목록 / 파이썬

전화번호 목록

문제 설명

전화번호부에 적힌 전화번호 중, 한 번호가 다른 번호의 접두어인 경우가 있는지 확인하려 합니다.
전화번호가 다음과 같을 경우, 구조대 전화번호는 영석이의 전화번호의 접두사입니다.

  • 구조대 : 119
  • 박준영 : 97 674 223
  • 지영석 : 11 9552 4421

전화번호부에 적힌 전화번호를 담은 배열 phone_book 이 solution 함수의 매개변수로 주어질 때, 어떤 번호가 다른 번호의 접두어인 경우가 있으면 false를 그렇지 않으면 true를 return 하도록 solution 함수를 작성해주세요.

제한 사항

  • phone_book의 길이는 1 이상 1,000,000 이하입니다.
  • 각 전화번호의 길이는 1 이상 20 이하입니다.

입출력 예제

phone_book return
[119, 97674223, 1195524421] false
[123,456,789] true
[12,123,1235,567,88] false

입출력 예 설명

입출력 예 #1
앞에서 설명한 예와 같습니다.

입출력 예 #2
한 번호가 다른 번호의 접두사인 경우가 없으므로, 답은 true입니다.

입출력 예 #3
첫 번째 전화번호, “12”가 두 번째 전화번호 “123”의 접두사입니다. 따라서 답은 false입니다.

풀이

  • sorted를 이용해서 앞자리가 같은 문자열을 정렬
  • (phone_book[start][0] == A[0])에서 앞자리가 같은 값들만 비교
    • 중간에 똑같은 값을 찾으면 탈출 (A == phone_book[start][0:len(A)])
def solution(phone_book):
answer = True
phone_book = sorted(phone_book)
while len(phone_book) > 1: 
  start = 0
  A = phone_book.pop(0)
  while (phone_book[start][0] == A[0]) & (start < len(phone_book)):
      if A == phone_book[start][0:len(A)]:
          answer = False
          break
      start += 1
  if answer == False: break
return answer

다른사람 풀이

  • 풀이는 똑같음

  • startswith는 괄호안의 값이 같은지 확인하는 함수

    • A.startswith(B, 0, 10) : A의 0부터 10이 B와 같은 지 확인
    • 아무값도 없으면 0부터 B의 길이까지 확인
  • 근데, 이 방법이 되는 이유가 sorted()에 의해서 됨

    • 위의 내 코드도 sorted때문에 앞자리 같은 값만 비교하지 않아도 되었음 !!!

def solution(phoneBook):
    phoneBook = sorted(phoneBook)

    for p1, p2 in zip(phoneBook, phoneBook[1:]):
        if p2.startswith(p1):
            return False
	return True

출처: 프로그래머스 코딩 테스트 연습,https://programmers.co.kr/learn/challenges

댓글(0)

Designed by JB FACTORY