[백준/1543] 문서 검색 (Python)

Problem : https://www.acmicpc.net/problem/1543

1543번: 문서 검색

Status : Solved

Time : 00:11:43

 

---

 

문제 설명

 

세준이는 영어로만 이루어진 어떤 문서를 검색하는 함수를 만들려고 한다. 이 함수는 어떤 단어가 총 몇 번 등장하는지 세려고 한다. 그러나, 세준이의 함수는 중복되어 세는 것은 빼고 세야 한다. 예를 들어, 문서가 abababa이고, 그리고 찾으려는 단어가 ababa라면, 세준이의 이 함수는 이 단어를 0번부터 찾을 수 있고, 2번부터도 찾을 수 있다. 그러나 동시에 셀 수는 없다.

세준이는 문서와 검색하려는 단어가 주어졌을 때, 그 단어가 최대 몇 번 중복되지 않게 등장하는지 구하는 프로그램을 작성하시오.

 

입력 및 출력

 

입력

첫째 줄에 문서가 주어진다. 문서의 길이는 최대 2500이다. 둘째 줄에 검색하고 싶은 단어가 주어진다. 이 길이는 최대 50이다. 문서와 단어는 알파벳 소문자와 공백으로 이루어져 있다.

 

 

출력

첫째 줄에 중복되지 않게 최대 몇 번 등장하는지 출력한다.

 

 

---

 

풀이

 중복을 제거한다는 조건이 있으므로, 순차적으로 문제를 탐색하되 다음과 같은 제약 조건을 가지고 풀이하면 된다.

• 어떤 인덱스에서 검색 단어와 문자열이 일치할 때, count를 1 올리고 검색 단어 길이만큼 인덱스를 넘긴다.
• count를 증가시키지 않고 인덱스를 1 증가시킨다.
• 만약 인덱스가 총 문자열 길이보다 길다면 탐색을 중단한다.

또한, 이러한 탐색 시 count와 인덱스가 중복인 경우가 매우 많이 나타날 수 있으므로, visited 배열을 통해 중복되는 경우를 제거하면 된다.

 

풀이 코드

from collections import deque

strings = input().strip()
length = len(strings)
word = input().strip()
result = 0
q = deque([(0, 0)])
visited = [[False]*length for _ in range(length)]

while q :
  idx, cnt = q.pop()
  if idx >= length :
    result = max(result, cnt)
    continue
  if visited[idx][cnt] : 
    continue
  visited[idx][cnt] = True

  if idx + len(word) <= length and strings[idx:idx+len(word)] == word :
    q.append((idx + len(word), cnt+1))
    
  q.append((idx+1, cnt))

print(result)

풀이 완료!