티스토리 뷰

백준

백준 소스코드 [C++] 1305 광고

Hani_Levenshtein 2020. 11. 9. 08:28

www.acmicpc.net/problem/1305

 

1305번: 광고

첫째 줄에 광고판의 크기 L이 주어지고, 둘째 줄에 현재 광고판에 보이는 문자열이 주어진다. L은 백만보다 작거나 같은 자연수이다.

www.acmicpc.net

백준 소스코드 [C++] 1305 광고

#include <iostream>
#include <algorithm>
#include <queue>
#include <string.h>
#include <limits.h>
#include <vector>
#include <math.h>
#include <stack>
#include <bitset>
#include <string>
typedef long long ll;
using namespace std;
vector<int> getPi(string sub) {
	vector<int> pi(sub.size(), 0);
	for (int sur = 1, pre = 0; sur < sub.size(); sur++) {
		while (pre > 0 && sub[sur] != sub[pre]) pre = pi[--pre];
		if (sub[sur] == sub[pre]) pi[sur] = ++pre;
	}
	return pi;
}

int main() {
	ios_base::sync_with_stdio(false), cin.tie(nullptr);
	string sub;
	int n;
	cin >> n;
	cin >> sub;
	vector<int> res =getPi(sub);
	cout << n - res[res.size() - 1];
	return 0;
}
댓글