본문 바로가기

코딩테스트 연습/programmers

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

String.startsWith 라는 메소드를 배웠다. "접두사"라는 말을 잘 못보고 contains 메소드를 써버려서 계속 오답이 나왔던.. ㅠㅠ 

문제를 잘 읽자!!!

 

2중 for문 -> 시간초과로 통과하지 못했던 코드 

import java.util.*;
import java.io.*;

class Solution {
    public boolean solution(String[] phone_book) {
        boolean answer = true;
        
        HashMap<String, Integer> map = new HashMap<>();
        
        for(int i=0;i<phone_book.length;i++){
            for(int j=0;j<phone_book.length;j++){
                if(i == j) continue;
                if(phone_book[i].startsWith(phone_book[j])){
                    answer = false;
                    break;
                }
            }
        }
        
        return answer;
    }
}

2중 포문은 안되는건가?!

 

 

오름차순 정렬 후, 더 긴 string안에 짧은 string이 포함되어있는지 확인하는 코드로 변경! -> 통과

import java.util.*;
import java.io.*;

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

 

 

구글 검색을 참고해보니 해시로도 풀 수 있었다! 

import java.util.HashMap;
import java.util.Map;

class Solution {
    public boolean solution(String[] phoneBook) {
        // 1. HashMap을 선언한다.
        Map<String, Integer> map = new HashMap<>();

        // 2. 모든 전화번호를 HashMap에 넣는다.
        for (int i = 0; i < phoneBook.length; i++) 
            map.put(phoneBook[i], i);
        
        // 3. 모든 전화번호의 substring이 HashMap에 존재하는지 확인한다.
        for (int i = 0; i < phoneBook.length; i++)
            for (int j = 0; j < phoneBook[i].length(); j++)
                if (map.containsKey(phoneBook[i].substring(0, j)))
                    return false;

        return true;
    }
}