[프로그래머스] 전화번호 목록 - java

출처 : https://school.programmers.co.kr/learn/courses/30/lessons/42577

프로그래머스

import java.util.*;
class Solution {
    public boolean solution(String[] phone_book) {
        boolean answer = true;
        Arrays.sort(phone_book, new Comparator<String>(){
            public int compare(String o1,String o2){
                return o1.length()-o2.length();
            }
        });
        for(int i=0;i<phone_book.length-1;i++){
            String num=phone_book[i];
            for(int j=i+1;j<phone_book.length;j++){
                if(num.equals(phone_book[j].substring(0,num.length()))){
                    answer=false;
                    break;
                }
            }
        }
        return answer;
    }
}

역시나 phone_book 의 length가 최대 1,000,000이므로 이중 for문에서는 시간 초과가 났습니다.

 

startsWith함수를 이용하여 더 쉽게 하였습니다.

import java.util.*;
class Solution {
    public boolean solution(String[] phone_book) {
          Arrays.sort(phone_book);
        for (int i = 0; i < phone_book.length - 1; i++){        
            if (phone_book[i + 1].startsWith(phone_book[i])){
                return false;
            }
        }
        return true;
    }
}

startsWith 함수를 알고 싶다면 

https://signaldogfeet.tistory.com/31

startsWith 와 endsWith - java

를 참고해주세요.