#568825 (C++14) No.430 文字列検索

• 問題: No.430 文字列検索
• ユーザー: CELICA
• 提出日時: 2020-10-18 22:45:32
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 44 ms / 2,000 ms
• コード長: 942 bytes
• コンパイル時間: 1,730 ms
• コンパイル使用メモリ: 174,904 KB
• 実行使用メモリ: 7,708 KB
• 最終ジャッジ日時: 2024-11-10 00:48:15
• 合計ジャッジ時間: 2,830 ms
• ジャッジサーバーID （参考情報）: judge2 / judge4

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	44 ms; 7,704 KB
testcase_02	AC	20 ms; 7,704 KB
testcase_03	AC	20 ms; 7,704 KB
testcase_04	AC	2 ms; 5,248 KB
testcase_05	AC	1 ms; 5,248 KB
testcase_06	AC	2 ms; 5,248 KB
testcase_07	AC	2 ms; 5,248 KB
testcase_08	AC	43 ms; 7,700 KB
testcase_09	AC	2 ms; 5,248 KB
testcase_10	AC	5 ms; 5,248 KB
testcase_11	AC	37 ms; 7,696 KB
testcase_12	AC	37 ms; 7,704 KB
testcase_13	AC	38 ms; 7,708 KB
testcase_14	AC	36 ms; 7,700 KB
testcase_15	AC	33 ms; 7,704 KB
testcase_16	AC	23 ms; 7,696 KB
testcase_17	AC	20 ms; 7,700 KB

ソースコード

#include <bits/stdc++.h>

using namespace std;

using ll = long long;
using ul = unsigned long;
using ull = unsigned long long;

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(nullptr);

	string s;
	cin >> s;

	ull ls = s.size();

	vector<vector<ull> > hs(11);
	for (ull i = 1; i <= min(ls, 10ull); ++i)
	{
		ull h{ 0 };
		for (ull j = 0; j < i; ++j)
			h = h * 26 + s[j] - 'A';

		hs[i].push_back(h);
		ull d = (ull)pow(26, i - 1);
		for (ull j = 1; j <= ls - i; ++j)
		{
			h %= d;
			h *= 26;
			h += s[j + i - 1] - 'A';
			hs[i].push_back(h);
		}

		sort(hs[i].begin(), hs[i].end());
	}

	int m;
	cin >> m;

	int res{ 0 };
	for (int i = 0; i < m; ++i)
	{
		string c;
		cin >> c;

		ull h{ 0 };
		for (ull i = 0; i < c.size(); ++i)
			h = h * 26 + c[i] - 'A';

		auto itb = hs[c.size()].begin();
		auto ite = hs[c.size()].end();

		res += upper_bound(itb, ite, h)
			- lower_bound(itb, ite, h);
	}

	cout << res << "\n";

	return 0;
}