【Gym-101972H】H.Beautiful Substrings 思维

Gym-101972H

Beautiful Substrings

题目大意

两个字符串a、b
在a中长度为k的子串，在b中有多少任意长的子串，满足首尾字符和a的子串相同

Time : 1000 ms
Memory: 262144 kB

解题思路及分析

遍历字符串a，对于长度为k的子串，用一个二维bool数组记录以i开头以j结尾的字符是否存在
然后遍历字符串b，记录对于长度为i的前缀子串每个字符出现的次数
然后再次遍历字符串b作为结尾，枚举每个字符j为首，判断刚刚a的子串数组中是否出现过以j开头以b[i]结尾的子串，若为真则在结果加上cnt[i][j]，即以b[i]结尾的字符串中，前i个字符中j字符的数量，也即前i个字符中以j开头以b[i]结尾的子串的个数。

AC代码

#include <bits/stdc++.h>
typedef long long llong;
const int N = 1e5 + 5;
using namespace std;

int n, m, k;
string a, b;
bool exist['z' + 10]['z' + 10];
int cnt[N]['z' + 10];

int main() {
    #ifdef LOCAL
        freopen("E:/code/ACM/in.in", "r", stdin);
        freopen("E:/code/ACM/out.out", "w", stdout);
    #endif
    int T;	cin >> T;
    while (T--) {
        memset(exist, false, sizeof(exist));
        memset(cnt[0], 0, sizeof(cnt[0]));
        cin >> n >> m >> k;
        cin >> a >> b;
        for (int i = 0; i < a.size() - k + 1; i++) {
            exist[a[i]][a[i + k - 1]] = true;
        }
        cnt[0][b[0]]++;
        for (int i = 1; i < b.size(); i++) {
            for (int j = 'a'; j <= 'z'; j++) {
                cnt[i][j] = cnt[i - 1][j];
            }
            cnt[i][b[i]]++;
        }
        llong sum = 0;
        for (int i = 0; i < b.size(); i++) {
            for (int j = 'a'; j <= 'z'; j++) {
                if (exist[j][b[i]]) {
                    sum += cnt[i][j];
                }
            }
        }
        cout << sum << endl;
    }

    return 0;
}