본문 바로가기

algorithm/프로그래머스

프로그래머스_전화번호 목록(C++, JAVA)

 

programmers.co.kr/learn/courses/30/lessons/42577

 

코딩테스트 연습 - 전화번호 목록

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

programmers.co.kr

[문제 풀이]

set을 이용하여 문제를 풀었습니다.

이 문제는 전화번호 목록 중에 접두어로써 존재하는 번호가 있는지를 찾는 문제입니다. 짧은 시간 안에 번호의 존재를 찾기 위해 set을 이용하여 전화번호 목록을 저장하였고 전화번호 리스트를 순회하며 길이가 N인 번호에 대해 0 ~N까지의 번호 중 set에 존재하는 번호가 있는지 찾아주었습니다.  

[C++]

#include <string>
#include <vector>
#include<set>

using namespace std;

bool solution(vector<string> phone_book) {
    
    set<string> pre_phone;
    
      for(auto item : phone_book)
        pre_phone.insert(item);
    
    for(auto item : phone_book)
        for(int i = 0; i < item.size(); i++)
            if(pre_phone.find(item.substr(0, i)) !=  pre_phone.end()){
                return false;
            }
    
    return true;
    
}

 

[JAVA]

import java.util.HashSet;
import java.util.Set;


class Solution {
    public boolean solution(String[] phone_book) {
        boolean answer = true;
        
        Set<String> pre_phone = new HashSet<>();
        
        for(int i = 0; i < phone_book.length; i++) {
        	pre_phone.add(phone_book[i]);
        }
        
        for(int i = 0; i < phone_book.length; i++) {
        	for(int j = 0; j < phone_book[i].length(); j++) {
        		if(pre_phone.contains(phone_book[i].substring(0, j)))
        			return false;
        	}
        }
        
        return answer;
    }
}