20-dohyeondol1

[dohyeondol1]

🔗 문제 링크

접두사

✔️ 소요된 시간

20m

✨ 수도 코드

요즘 난이도가 있는 PR이 꽤 올라와서 난이도 완화...차원에서 조금 쉬운 문제를 가져왔습니다. 너무 쉬운가..?
제 풀이 말고도 다른 풀이가 많아보여서 여러분은 어떻게 접근하실지 궁금한 마음에 가져와 봤습니다.

수도 코드

main
    정수 N 선언 및 입력 받기
    반복 i = 0부터 N-1까지
        문자열 arr[i] 입력받기

    arr 배열을 정렬하기 (사전순)

    count = N으로 설정

    반복 i = 0부터 N-2까지
        prefix = arr[i+1]의 앞에서 arr[i]의 길이만큼 자른 부분 문자열
        만약 prefix와 arr[i]가 같다면
            count = count - 1

    count 출력

여러 단어가 주어졌을 때, 다른 단어의 접두어가 되지 않는 집합의 최대 크기를 구하는 문제입니다.
이는 정렬을 사용하면 간단하게 해결할 수 있습니다.

만약 한 단어의 접두어가 되는 단어가 있다면,
해당 단어는 사전순 정렬 시 항상 접두어를 포함하는 단어 앞에 위치하게 됩니다.

예제 입력 1을 예시로 들어보겠습니다.

(입력순)
hello
hi
h
run
rerun
running

(사전순)
h
hello
hi
rerun
run
running

사전순으로 정렬 시, hello의 접두어가 되는 h는 hello의 바로 앞에 위치하는 것을 확인할 수 있습니다.
run과 running 도 마찬가지죠.

이 경우, 반복문을 이용한 브루트포스 탐색을 통해 해결할 수 있습니다.

다음 인덱스의 단어에서 현재 인덱스의 단어 길이만큼 잘라낸 문자열이 현재 단어와 같다면,
전체 단어 개수에서 1을 빼는 방식으로 풀이할 수 있습니다.

이때 제외하는 단어는 접두어가 될 수 있는 단어가 됩니다.
앞서 예제 입력 1에서 확인한 h는, hello의 접두어가 될 수도 있지만 hi의 접두어가 될 수도 있기 때문이죵.

결과적으로 브루트포스를 이용해 전체 단어 개수에서 접두어가 될 수 있는 단어들만 제외하면 해결할 수 있는 문제입니다.

📚 새롭게 알게된 내용

[Fnhid]

저는 트라이 자료구조를 이용하여 문제를 해결하였습니다.

트리의 변형 버전인데 이런 식으로 생겨서 접두사를 쉽게 분석할 수 있습니다.

#include <iostream>
#include <string>
#include <memory>
#include <vector>

using namespace std;

struct Trie
{
    Trie* next[26];
    bool isEnd;
    Trie(){
        isEnd=false;
        fill(next, next+26, nullptr);
    }
    ~Trie()
    {
        for (int i=0;i<26;i++)
        {
            if (next[i]) delete next[i];
        }
    }
    void insert(const string& s)
    {
        Trie* node = this;
        for (char c : s)
        {
            int idx = c - 'a';
            if (!node->next[idx]) node->next[idx] = new Trie();
            node = node->next[idx];
        }
        node->isEnd = true;

    }
    int countLeaf()
    {
        int cnt = 0;
        bool hasChild = false;
        for (int i=0;i<26;i++)
        {
            if (next[i])
            {
                hasChild = true;
                cnt += next[i]->countLeaf();
            }
        }
        if (!hasChild && isEnd) return 1;
        return cnt;
    }
};


int main()
{
    int n;

    cin >> n;
    vector<string> words(n);
    for (int i = 0; i < n; i++) cin >> words[i];
    Trie* root = new Trie();
    for (int i=0;i<n;i++) root->insert(words[i]);

    cout << root->countLeaf() << "\n";
    delete root;
}

[dohyeondol1]

@Fnhid
트라이까지 사용한 풀이는 확실히 예상 못했습니다 ㅋㅋ..
트라이를 사용한 문제는 풀어본 적 없는데 한번 배워봐야겠어용

[froglike6]

저는 파이썬의 startswith를 이용해 이 문제를 해결해봤습니다. 정렬 후 for를 돌면서 startswith로 검사한 후 ans를 추가하는 방법으로 진행했습니다.

import sys

input = sys.stdin.readline
n = int(input())
words = [input().rstrip() for _ in range(n)]
words.sort()
ans = 0
for i, w in enumerate(words):
    if i == n - 1 or not words[i + 1].startswith(w):
        ans += 1
print(ans)

[dohyeondol1]

코드 길이 실화티비..? startswith는 처음들어보네요..!!

[hadongun]

파이썬 !! 에 있는 startswith 라는 함수가 내장되어 있더군요..!

if savedWord.startswith(currentWord):

savedWord가 currentWord로 시작하는지 검사해라는 뜻입니다!!

매우 편리하군요!

파이파이

word_count = int(input())
wordList = [input().strip() for _ in range(word_count)]

wordList.sort(key=lambda word: -len(word))

nonPrefixWords = []

for current_Word in wordList:
    is_prefix = False
    for savedWord in nonPrefixWords:
        if savedWord.startswith(current_Word):
            is_prefix = True
            break
    if not is_prefix:
        nonPrefixWords.append(current_Word)

print(len(nonPrefixWords))