[1543] 문서 검색 (Python)

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

1543번: 문서 검색

분류

• 그리디 알고리즘
• 완전탐색 (Brute Force) 알고리즘 - 구현

문제

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

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

 

예제입력1

ababababa

aba

예제출력1

2

 

로직 정리

• 첫째 줄을 doc, 둘째 줄을 search로 하고 중복되지 않게 등장한 횟수를 cnt라 한다.
• search의 길이만큼 doc문자열을 자르고 같은지 비교했을 때
  • 같으면 doc의 인덱스를 search의 길이만큼 증가시키고(중복되면 안되므로), cnt를 1만큼 증가한다.
  • 다르면 doc의 인덱스를 1만큼 증가시킨다.

---

내가 처음 접근했던 코드는 다음과 같다.

n = input()
search = input()
i=0
cnt = 0
bool = True
while bool:
    for j in range(0,len(search)):
        if i < len(n):
            if n[i] != search[j]:
                i+=1
                if j == len(search) - 1:
                    i-=1
                break            # n에서 다음 알파벳으로 넘어가서 다시 처음부터 비교. 
            else:
                i += 1
                if j == len(search)-1: # search단어의 마지막 길이까지 같으면
                    cnt += 1
                    break
        else:
            bool= False
            break
print(cnt)

코드를 실행시켜보면 결과는 잘 나오지만

백준에서 돌려봤을 때 결과는 시 간 초 과  (예상은 했지만..) 

 

코드

doc = input()
search = input()
cnt = 0
i = 0

while i <= len(doc) - len(search):  # search와 비교할 수 있을 때 까지
    if doc[i:i+len(search)] == search: # i부터 문자열길이까지의 문자와 같은지 비교
    	cnt+=1
        i+=len(search)    # 중복되지 않아야하므로 검색단어길이 뒤에서부터 다시 찾는다.
    else:
    	i+=1   # 다르면 인덱스 1만 증가시키고 다시 비교
print(cnt)