字符串 (子串的匹配) (智算之道初赛高校组第一场)

题目列表

共 10 个测试点 每个测试点 10 分
每个测试点限时 1 秒 运行内存上限 512MB

给定两个字符串 S 和 T，它们都只由小写字母组成。现在请计算出字符串 SS的本质不同的排列中有多少种是字符串 T 的子串。
本质不同：就是看起来不同，例如 aab 有 3 种本质不同的排列 aab, aba, baa。

输入格式
第一行有一个字符串 S
第二行有一个字符串 T
输出格式
输出一个整数表示字符串 S 的本质不同的排列中有多少种是字符串 T 的子串

数据规模与约定
对于30%的数据，1 <= |S| <= 5，1 ≤ ∣T∣ ≤ 200
对于60%的数据，1 ≤ ∣S∣ ≤ 2000，1 ≤ ∣T∣ ≤ 2000
对于100%的数据，1 ≤ ∣S∣ ≤ 200000，1 ≤ ∣T∣ ≤ 200000
样例输入
aab
abacabaa
样例输出
2

思路： 利用一个窗口滑动截取t中s.size()长的子串，并判断该子串和s的各种字母数量是否对应一致，并且用一个set记录这些能够匹配的子串；最后直接输出set的大小就好。

代码实现：

#include<bits/stdc++.h>
#define endl '\n'
#define null NULL
#define ll long long
#define int long long
#define pii pair<int, int>
#define lowbit(x) (x &(-x))
#define ls(x) x<<1
#define rs(x) (x<<1+1)
#define me(ar) memset(ar, 0, sizeof ar)
#define mem(ar,num) memset(ar, num, sizeof ar)
#define rp(i, n) for(int i = 0, i < n; i ++)
#define rep(i, a, n) for(int i = a; i <= n; i ++)
#define pre(i, n, a) for(int i = n; i >= a; i --)
#define IOS ios::sync_with_stdio(0); cin.tie(0);cout.tie(0);
const int way[4][2] = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
using namespace std;
const int  inf = 0x7fffffff;
const double PI = acos(-1.0);
const double eps = 1e-6;
const ll   mod = 1e9 + 7;
const int  N = 2e5 + 5;

int x, y;
string s, t;
set<string> ss;
int a[30], b[30];

void solve()
{
    for(int i = 0; i < x; i ++)
		a[s[i]-'a'] ++;
	for(int i = 0; i < y; i ++){
		b[t[i]-'a'] ++;
		if(i >= x - 1){
			if(i != x - 1) b[t[i - x] - 'a'] --;
			if(b[t[i] - 'a'] == a[t[i] - 'a']){
				int ok = 0;
				for(char j = 'a'; j <= 'z'; j ++){
					if(b[j - 'a'] != a[j - 'a']){
						ok = 1;
						break;
					}
				}
				if(!ok)
					ss.insert(t.substr(i - x + 1, x));
			}
		}
	}
	cout << ss.size() << endl;
}

signed main()
{
	IOS;

	cin >> s >> t;
	x = s.length();
	y = t.length();
	if(x > y){
		cout << 0 << endl;
		return 0;
	}
	solve();
	return 0;
}

后来队友还分享了前缀和的代码：

#include<bits/stdc++.h>
#define endl '\n'
#define null NULL
#define ll long long
#define int long long
#define pii pair<int, int>
#define lowbit(x) (x &(-x))
#define ls(x) x<<1
#define rs(x) (x<<1+1)
#define me(ar) memset(ar, 0, sizeof ar)
#define mem(ar,num) memset(ar, num, sizeof ar)
#define rp(i, n) for(int i = 0, i < n; i ++)
#define rep(i, a, n) for(int i = a; i <= n; i ++)
#define pre(i, n, a) for(int i = n; i >= a; i --)
#define IOS ios::sync_with_stdio(0); cin.tie(0);cout.tie(0);
const int way[4][2] = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
using namespace std;
const int  inf = 0x7fffffff;
const double PI = acos(-1.0);
const double eps = 1e-6;
const ll   mod = 1e9 + 7;
const int  N = 2e5 + 5;

int n, m;
string s,t;
set<string> st;
char s[N], t[N];
int a[110], b[110], tt[110];
int ss[N][30], d[N][30];

signed main()
{
	cin >> s >> t;
	for(int i = 0; i < s.size(); i ++)
		a[s[i]-'a'+1]++;
	for(int i = 0; i < t.size(); i ++)
		d[i+1][t[i]-'a'+1]++;
		
	for(int j = 1; j <= 26; j ++){
		for(int i = 1; i <= t.size(); i ++)
			ss[i][j]=ss[i-1][j]+d[i][j];
			
	int k = s.size();
	for(int i = 1; i <= t.size() - k + 1; i ++){
		int flag = 1;
		for(int j = 1; j <= 26; j ++){
			tt[j] = ss[i + k - 1][j] - ss[i - 1][j];
			if(tt[j] != a[j])
				flag = 0;
		}
		if(flag){
			string temp;
			temp = t.substr(i - 1,k);
			st.insert(temp);
		}
	}
	int cnt = st.size();
	cout << cnt << endl;;
	return 0;
}