백준 1786 찾기

Posted by : at

Category : Solution


Problem2 문제풀이

#include <iostream>
#include <string>
#include <vector>

using namespace std; 


vector<int> make_nested(string P);
vector<int> KMP(string S, string P);

int main()
{
	ios_base::sync_with_stdio(false);
	cin.tie(0);
	string S, P;
	getline(cin, S);
	getline(cin, P);;
	vector<int>ans = KMP(S, P);
	cout<<ans.size()<<'\n';
	for (int i = 0; i < ans.size(); i++)
	{
		cout << ans[i]+1 << ' ';
	}
	
	return 0;
}

vector<int> make_nested(string P)
{
	int len_P = P.length();
	int idx = 0;
	vector<int> nested(len_P);
	for (int i = 1; i < len_P; i++)
	{
		while (idx > 0)
		{
			if (P[i] == P[idx])
			{
				break;
			}
			idx = nested[idx - 1];
		}
		if (P[i] == P[idx])
		{
			nested[i] = ++idx;
		}
	}
	return nested;
}
vector<int> KMP(string S, string P)
{
	int idx = 0;
	int len_S = S.length();
	int len_P = P.length();
	vector<int>nested = make_nested(P);
	vector<int>ans;
	for (int i = 0; i < len_S; i++)
	{
		while (idx > 0)
		{
			if (S[i] == P[idx])
			{
				break;
			}
			idx = nested[idx - 1];
		}
		if(S[i]==P[idx])
		{
			if (idx == len_P - 1)
			{
				ans.push_back(i - (len_P - 1));
				idx = nested[idx];
			}
			else
			{
				idx++;
			}
		}
	}
	return ans;
}

About GJ

안녕하세요 방문해주셔서 감사합니다. 혹시 보시면서 궁금하신것 있으시면 https://open.kakao.com/o/sivaz71c로 연락주세용~

Star
Useful Links