[Baekjoon] 14426_접두사 찾기

Silver III

문제(출처: https://www.acmicpc.net/problem/14426)

---

<접두사 찾기>

문제

 

문자열 S의 접두사란 S의 가장 앞에서부터 부분 문자열을 의미한다. 예를 들어, S = "codeplus"의 접두사는 "code", "co", "codepl", "codeplus"가 있고, "plus", "s", "cude", "crud"는 접두사가 아니다.
총 N개의 문자열로 이루어진 집합 S가 주어진다.
입력으로 주어지는 M개의 문자열 중에서 집합 S에 포함되어 있는 문자열 중 적어도 하나의 접두사인 것의 개수를 구하는 프로그램을 작성하시오.

 

입력

 

첫째 줄에 문자열의 개수 N과 M (1 ≤ N ≤ 10,000, 1 ≤ M ≤ 10,000)이 주어진다.
다음 N개의 줄에는 집합 S에 포함되어 있는 문자열이 주어진다.
다음 M개의 줄에는 검사해야 하는 문자열이 주어진다.
입력으로 주어지는 문자열은 알파벳 소문자로만 이루어져 있으며, 길이는 500을 넘지 않는다. 집합 S에 같은 문자열이 여러 번 주어지는 경우는 없다.

 

출력

 

첫째 줄에 M개의 문자열 중에 총 몇 개가 포함되어 있는 문자열 중 적어도 하나의 접두사인지 출력한다.

 

문제 풀이
- 시간 초과 코드

import sys

if __name__=='__main__':
    N,M=map(int, sys.stdin.readline().split())

    temp=[]
    for i in range(N):
        temp.append(sys.stdin.readline().strip())

    cnt=0
    for i in range(M):
        find=sys.stdin.readline().strip()
        for j in temp:
            pre=j[:len(find)]
            if find==pre:
                cnt+=1
                break

    print(cnt)

 

• 이중 반복문을 활용- 찾을 접두사 입력받고 그 길이만큼 문자열 잘라서 비교하여 같으면 cnt +1

이 문제에 대해서 찾아보니 글이 3개 정도밖에 없었다. 어떤 분도 이렇게 풀어 PyPy3로 통과하였다고 해서 일단 돌려보았더니
통과... ㅎ_ㅎ 하지만 난 Python3로 풀기에.. 다시 찾아서 도전!


- 통과한 코드

import sys

if __name__=='__main__':
    N,M=map(int, sys.stdin.readline().strip().split())

    temp=[]
    for i in range(N):
        temp.append(sys.stdin.readline().strip()) # 문자열 입력

    find=[]
    for i in range(M):
        find.append(sys.stdin.readline().strip()) # 접두사 입력

    #정렬
    find.sort()
    temp.sort()

    cnt=0
    s=0
    for i in find:
        for j in range(s,N):
            if i==temp[j][:len(i)]:
                cnt+=1
                s=j # 정렬 했기 때문에 index 교체
                break

    print(cnt)

 

• 문자열, 접두사 입력받은 후 정렬
• 특정 문자로 시작하는지 여부 확인
• 접두사 맞다면 cnt 증가 및 for문 시작하는 index 수정

---

생각🤔

시간 초과를 해결하는데 핵심 포인트는 정렬과 for문의 index 수정인 것 같다.
조금만 생각 더 해보면 정렬을 생각해낼 수 있었을 텐데 바로 찾아본 나의... 손가락에 후회한다 ^_^

---