Cod sursa(job #3150589)

Utilizator moldovan_robert_lolMoldovan Robert moldovan_robert_lol Data 17 septembrie 2023 16:35:02
Problema Prefix Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.9 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#include <utility>
#include <algorithm>
#include <string>

std::ifstream fin("prefix.in");
std::ofstream fout("prefix.out");

std::string str;

std::vector<int> kmp(const std::string& str) {
	std::vector<int> ret(str.size(),0);
	for (int i = 1; i < str.size(); i++) {
		int j = ret[i-1];
		while (j&&str[i]!=str[j]) {
			j = ret[j-1];
		}
		j += str[i]==str[j];
		ret[i] = j;
	}
	return ret;
}

void test_case() {
	fin >> str;
	std::vector<int> pfx_func = kmp(str);

	int start = 0;
	int ret = 0;
	for (int i = 0; i < pfx_func.size(); i++) {
		if (!pfx_func[i]) {
			continue;
		}
		if (pfx_func[i]==1) {
			start = i;
		}
		if (pfx_func[i]) {
			if (pfx_func[i]%start==0&&(i+1)%start==0) {
				ret = std::max(ret,i+1);
			}
		}
	}

	fout << ret << "\n";
}

int main() {
	int test_cases;
	fin >> test_cases;
	while (test_cases--) {
		test_case();
	}
}