수열의 수 n과 확인할 수열의 길이 k를 입력받은 후, 각 수열의 길이, 수열을 입력받는다. k 길이만큼의 반복되는 부분 수열과 해당 부분 수열을 거꾸로한 것이 모든 수열에 존재 시 YES를, 아닐 시 NO를 출력하는 문제. 제일 짧은 수열을 기준으로 k 길이만큼 슬라이싱 후 KMP 알고리즘을 통해 수열에 확인하는 지 확인하여 풀었다. 자세한 설명은 주석을 참고.
\nimport sys\ninput = sys.stdin.readline\n\ndef KMP(text, suspect):\n lt = len(text)\n ls = len(suspect)\n\n i = 0\n j = 0\n while i < lt:\n if text[i] == suspect[j]:\n i += 1\n j += 1\n elif text[i] != suspect[j]:\n if j != 0:\n j = lps[j-1]\n else:\n i += 1\n\n if j == ls:\n # 동일한 부분이 존재 시 return\n return True\n # 없을 시\n return False\n\ndef getLPS(p, lps):\n leng = 0\n i = 1\n while i < len(p):\n if p[i] == p[leng]:\n leng += 1\n lps[i] = leng\n i += 1\n else:\n if leng != 0:\n leng = lps[leng-1]\n else:\n lps[i] = 0\n i += 1\n\n\nn, k = map(int, input().split())\ntext_list = []\nmin_n, min_index = 0, 0\n\nfor i in range(n):\n t_len = int(input())\n # 제일 짧은 수열의 길이와 인덱스를 저장\n if min_n < t_len:\n min_n = t_len\n min_index = i\n\n text_list.append(input().split())\n\n# 0부터 제일 짧은 수열의 길이 - k + 1만큼 반복\nfor i in range(0, len(text_list[min_index])-k+1):\n # i부터 k 길이를 갖도록 슬라이싱\n suspect = text_list[min_index][i:i+k]\n # LPS 계산하여 저장\n lps = [0 for _ in range(k)]\n getLPS(suspect, lps)\n\n # 정답 여부를 확인 할 변수 ans\n ans = 0\n\n # 다른 문자열들을 기준으로 반복\n for text_index in range(n):\n # 제일 짧은 문자열과 동일 시 확인하지 않음\n if text_index == min_index:\n continue\n # suspect 수열과 동일한 부분이 존재 시\n if KMP(text_list[text_index], suspect):\n ans += 1\n # suspect 수열을 반대로 한 것과 동일한 부분이 존재 시\n elif KMP(text_list[text_index], list(reversed(suspect))):\n ans += 1\n\n # 다른 문자열들 모두 동일한 수열이 존재할 시\n if ans == n-1:\n print(\"YES\")\n sys.exit(0)\nprint(\"NO\")