✏️ 문제
마야 문자를 해독하는 일은 예상 외로 어려운 일이다. 현재에도 뜻이 완전히 밝혀진 마야 문자는 거의 없는 실정이며, 그나마 해독에 진척이 시작된 지는 30여 년도 되지 않았다.
마야 문자는 소리를 나타내는 여러 종류의 그림글자로 구성되는데, 이 글자들이 여러 위치에서 결합함으로써 단어를 형성한다.
마야 문자 해독을 어렵게 하는 요인 중 하나는 바로 단어를 읽는 순서이다. 마야 문자를 쓰는 고대인들은 단어를 기록할 때 특정한 규칙 대신, 그들이 보기에 좋게 보이도록 단어를 이루는 글자들을 아무렇게나 배열했다. 그렇기 때문에 고고학자들이 마야 기록에서 단어를 이루는 각 그림글자들의 발음을 알아내더라도 그 단어를 실제로 발음하는 방법은 정확히 알 수 없는 셈이다.
고고학자들은 W라는 특정 단어를 발굴 기록으로부터 찾고 있다. 그 단어를 구성하는 각 글자들은 무엇인지 알고 있지만, 이것이 고대 기록에 어떤 형태로 숨어 있는지는 다 알지 못한다.
W를 이루고 있는 g개의 그림문자와, 연구 대상인 벽화에 기록된 마야 문자열 S가 주어졌을 때, 단어 W가 마야 문자열 S에 들어있을 수 있는 모든 가짓수를 계산하는 프로그램을 작성하시오. 즉, 문자열 S안에서 문자열 W의 순열 중 하나가 부분 문자열로 들어있는 모든 경우의 수를 계산하라는 뜻이다.
🎈 입력
첫째 줄에 고고학자들이 찾고자 하는 단어 W의 길이 g와 발굴된 벽화에서 추출한 문자열 S의 길이 |S|가 빈 칸을 사이에 두고 주어진다. (1≤g≤3000, g≤|S|≤3,000,000) 둘째 줄에 W, 셋째 줄에 S의 실제 내용이 들어있다. 모든 문자열은 알파벳으로 이루어지며, 대소문자를 구분한다.
🎈 출력
첫째 줄에 W의 순열이 S 안에 있을 수 있는 형태의 개수를 출력한다.
🎈 입출력 예
입력
4 11
cAda
AbrAcadAbRa
출력
2
👩💻 내 코드
엄청 어렵지 않은 문제였던 것 같은데 시간복잡도에 약했던 나는 꽤나 고생했다ㅠㅠㅠㅠ
이 문제는 슬라이딩 윈도우 알고리즘을 사용해서 풀어야 하는 문제이다.
원래 C++로 풀려고 했으나 문자열 같은 경우는 파이썬으로 푸는 것이 훨씬 쉽고 효율적일 것 같아서 파이썬으로 풀었다.
처음 시간복잡도를 고려하지 않고 생각했던 아이디어는 여러가지가 있었는데 아래와 같다.
- 찾으려는 문자열과 슬라이딩 윈도우를 사용해 뽑아낸 문자열을 정렬하여 비교하고, 같으면 카운트
- flag 변수를 사용 (찾으려는 문자열과 슬라이딩 윈도우를 사용해 뽑아낸 문자열이 모두 같으면 flag = 0, 하나라도 다른 문자가 있으면 flag = 1)
// 2번 아이디어를 이용해서 구현한 코드
wLen, sLen = map(int, input().split())
w, s = str(input()), str(input())
left, right = 0, wLen-1
cnt = 0
tmp = ""
for i in range(left, right):
tmp += s[i]
while right < sLen:
tmp += s[right]
flag = 0
for t in tmp:
if t not in w:
flag = 1
if flag == 0:
cnt += 1
tmp = tmp[1:]
left += 1
right += 1
print(cnt)
시간초과가 뜨자 마자 딱 알 수 있었다. 반복해서 정렬을 하면 시간이 오래걸리는 것은 당연하고, while문 안에 for문을 돌리는 것 또한 시간초과가 날 수 밖에 없다는 것을...
결과적으로 DAT를 이용해서 문제를 풀었고, 시간초과를 면할 수 있었다.
import sys
wLen, sLen = map(int, input().split())
w = sys.stdin.readline().strip()
s = sys.stdin.readline().strip()
left, right = 0, wLen-1
wDAT = [0]*58
tmpDAT = [0]*58
cnt = 0
for i in w:
wDAT[ord(i)-65] += 1
for i in range(left, right):
tmpDAT[ord(s[i])-65] += 1
while right < sLen:
tmpDAT[ord(s[right])-65] += 1
if wDAT == tmpDAT:
cnt += 1
tmpDAT[ord(s[left])-65] -= 1
left += 1
right += 1
print(cnt)
