【思维题】CodeForce 1016B Segment Occurrences

版权声明：Johnson https://blog.csdn.net/m0_38055352/article/details/91430441

这段时间要沉迷刷题一段时间了，就让CSDN陪我一起吧！

一、题目大意

题目的意思是给定一个长度为n的字符串s，和一个长度为m的字符串t，然后给定q个查询，对于每个查询，形如：l r，则取出s中第l个字符到第r个字符的字串，求解这个字串中所包含的字符串t的数量。

二、题目思路以及AC代码

在一开始我又惯性思维的去解题了， 也就是通常所说的暴力法。既然要求对子串匹配，那就直接给一个查询，把子串取出来，然后用子串去匹配字符串t。这样做显然是对的，但也显然会超时，因为每次查询都是O(nm)的复杂度，再加上是q次查询，也就是O(nm*q)的复杂度。

所以我们要考虑一种简单的办法，也就是不能惯性思维，那么我们拿到字符串s和t的时候，可以先对其进行匹配，扫描一遍s，找到匹配的下标范围，存起来，这一步的时间复杂度为O(nm)，然后在进行查询的时候，每次只需要遍历一遍存起来的部分就行了，而存起来的部分最多也就是n，那么这部分的时间复杂度也就是O(nq)，所以最终也就是O(max(q, m) * n)，不会超时。

下面给出AC代码：

#include <iostream>
#include <string>
#include <cstring>
#include <vector>
using namespace std;

int n, m, q;
vector<pair<int, int>> v;

int main()
{
	cin >> n >> m >> q;
	string s, t;

	cin >> s >> t;
	int slen = s.length();
	int tlen = t.length();

	for (int i = 0; i < slen; i++) {
		int a = i;
		int b = 0;
		while (s[a] == t[b]) {
			a++;
			b++;
			if (a >= slen || b >= tlen) break;
		}
		if (b >= tlen) v.push_back(make_pair(i + 1, a));
	}

	int size = v.size();
	while (q--) {
		int l, r;
		cin >> l >> r;

		int cnt = 0;
		for (int i = 0; i < size; i++) {
			if (v[i].first >= l && v[i].second <= r) cnt++;
		}

		cout << cnt << endl;
	}

    return 0;
}

如果有问题，欢迎大家指正！！！