【2021杭电多校第七场】Yiwen with Sqc

最新推荐文章于 2022-07-11 18:01:14 发布

原创最新推荐文章于 2022-07-11 18:01:14 发布 · 605 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

题解专栏收录该内容

16 篇文章

订阅专栏

博客主要讨论了一道ACM竞赛题目，该题目要求计算字符串中每个字符在所有连续子串内的出现次数平方和。通过分析，作者提出使用前缀和的方法优化算法，从最初的朴素算法降低到O(26n)的时间复杂度。在进一步思考后，作者发现可以利用字符出现位置和维持长度来进一步优化，将时间复杂度降至更低，实现了快速求解。

题目链接：

https://acm.hdu.edu.cn/showproblem.php?pid=7055

题目大意：

给一个字符串，对于每个小写字母ch，问所有区间内ch出现次数的平方和，其中|s|< $10^{5}$

$\sum_{c=97}^{122}\sum_{i=1}^{n}\sum_{j=1}^{n} sqc(l,r,ch)^{2}$

因为答案很大，所以对998244353取模（答案很大，忍一下

解题思路：

朴素算法，像公式那样，遍历每个区间，相加。但是因为长度是1e5肯定会TLE

对于这种区间问题，我们往往采用差分或者前缀和

这题赛场上就想到了用前缀和

对于一个字母ch，我们定义 s[i]为到i为止，ch出现的个数

那么每个区间[,r]的答案就是 $(s[r] - s[l - 1])^{2}$

$ans = \sum_{i=1}^{n}\sum_{j=i}^{n} (s[j]-s[i-1])^{2}$

我们将平方和拆开化简

就会得到 $ans= (n+1) * \sum_{i=1}^{n}s[i]^{2} - (\sum_{i=1}^{n}s[i])^{2}$

然后时间复杂度就会变成O（26n）

（可以统计字母出现的种类tot，时间复杂度优化到O（tot*n））赛场上卡常过了，赛后T了

代码如下：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

const int N = 1e5 + 3;
const int MOD = 998244353;
ll a[27][N];
int b[27], cc[27];
int main() {
    int T;
    scanf ("%d", &T);
    while (T--) {
        string s;
        cin >> s ;
        int len = s.length();
        memset (a, 0, sizeof (a) );
        memset (b, 0, sizeof (b) );
        for (int i = 0; i < len; i++) {
            b[s[i] - 'a' + 1] = 1;
            a[s[i] - 'a' + 1][i]++;
        }
        int tot = 0;
        for (int i = 1; i <= 26; i++)
            if (b[i])
                cc[++tot] = i;
        ll ans = 0;
        ll tt = 0, t = 0;
        for (int k = 1; k <= tot; k++) {
            tt = 0, t = 0;
            int c = cc[k];
            for (int i = 0; i < len; i++) {
                a[c][i] = a[c][i - 1] + a[c][i];
                t = (ll) (t + a[c][i] * a[c][i] % MOD) % MOD;
                tt =
                 (ll) (tt + a[c][i]) % MOD;
            }
            ans += ( (len + 1) * t % MOD - tt * tt % MOD + MOD) % MOD;
            ans %= MOD;
        }
        printf ("%lld\n", ans);
    }
    return 0;
}

我们可以想一想，怎么样改进

我们的思路是求前缀和的和，和前缀和平方和。

我们可以发现一段前缀和的值，在一个区间内是一样的，并且这个区间长度是可以知道的。而前缀和的平方和其实就是在区间内当前字母的个数*当前这一段的长度。

对于字a在字符串ababa举例：

i	0	1	2	3	4	5	6
字母	\	a	b	a	b	a	\
维持长度		2		2		1
s[i]	0	1	1	2	2	3

对于前缀和的和

我们可以用 1 * 2 + 2 * 2 + 3 * 1 = 9 算

对于前缀和的平方和

我们可以用 $1^{2} * 2 + 2^{2}* 2 + 3^{2}*1$ 来算

这上面的 2 2 1 其实是这一个ch对前缀和影响的长度

用a[i]来保存序列中第i个ch出现的位置

用len[i]表示第i个ch维持的长度

$len[i]=a[i + 1]-a[i]$

令区间内出现ch的次数为cnt

len[cnt]=n+1-a[cnt]

因此我们的答案就变成了

$ans=(n+1)*\sum_{i=1}^{cnt}i*i*len[i] -\sum_{i=1}^{cnt}*i*len[i]$

这样跑的就非常的快

代码实现：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

const int N = 1e5 + 3;
const int MOD = 998244353;
char s[N];
void work() {
    scanf ("%s", s);
    int len = strlen (s);
    vector<int>a[27];
    for (int i = 1; i <= 26; i++) a[i].push_back (0);
    for (int i = 0; i < len; i++) {
        a[s[i] - 'a' + 1].push_back (i + 1);
    }
    ll ans = 0;
    for (int i = 1; i <= 26; i++) {
        a[i].push_back (len + 1);
        ll t1 = 0, t2 = 0;
        for (int j = 1; j + 1 < a[i].size(); j++) {
            t1 += (ll)j * (a[i][j + 1] - a[i][j]) % MOD;
            t2 += (ll)j * j % MOD * (a[i][j + 1] - a[i][j]) % MOD;
            t1 %= MOD;
            t2 %= MOD;
        }
        ans += ( (ll) (len + 1) * t2 % MOD - t1 * t1 % MOD + MOD ) % MOD;
        ans %= MOD;
    }
    printf ("%lld\n", ans);

}
int main() {
    int T;
    scanf ("%d", &T);
    while (T--)
        work();
    return 0;
}