字符串函数(strfun)[7.10]hl模拟赛

题目描述

这道题我上传到了题库中
给定两个等长的由大写英文字母构成的字符串 a 和 b, 从 a 中选出子串 x, 从 b 中选出 子串 y, 其中 x,y 等长。 定义函数 f(x, y)为满足条件 xi = yi(1 <= I <= |x|)的 i 的个数, 计算 f(x, y)的数学期望。
输入输出格式
输入格式：

第一行一个正整数: n, 表示 a 和 b 的长度。 第二行输入字符串 a。 第三行输入字符串 b。

输出格式：

输出一个实数表示 f(x, y)的期望, 答案保留 6 位小数。

输入输出样例
输入样例#1：

2
AB
BA

输出样例#1：

0.400000

【样例解释】
x,y 的选择有 5 种情况, 分别是(“A”, “B”), (“A”, “A”), (“B”, “B”), (“B”, “A”), (“AB”, “BA”)。
其中，第 2 对和第 3 对所对应的 f(x,y)等于 1, 其他都是 0, 由于选择每一对的概率都是
1/5，所以 f(x,y)的期望为 1/5 * 0 + 1/5 * 1 + 1/5 * 1 + 1/5 * 0 + 1/5 * 0 = 2/5 = 0.4。

说明

【数据约定】 对于 30%的数据, n <= 100。 对于 60%的数据, n <= 5000。 对于 100%的数据, n <= 2e5。

【时间限制】1s

【空间限制】 256MB

对于这道题，我们只要想暴力，然后优化暴力，优化暴力就行。
先看题面，我们先推出总方案数是${\sum }_{i=1}^n{i}^2$
我是直接乱推乱代出来的

然后我们考虑暴力怎么写。
最暴力的方法我忘了0.0…
那么我们就想高级一点的
对于每个串，有贡献的时候一定是s1[i] = = s2[j]且满足题面上要求的在子串中的下标相同，
那么我们考虑当s1[i] == s2[j]时如果在当前位置向左或向右同时增加长度，那么s1[i]和s2[j]在子串中的相对位置是不变的。
那么当s[1] = = s[2] 时，满足条件的子串数 = 向左能共同延伸的最大长度 * 向右能共同延伸的最大长度
于是我们就可以$n^2$枚举i，j然后算出长度，同时要注意，因为是共同延伸的最大长度，因此相乘时是min（i，j） * min（n-i+1， n-j+1）

这样我们就可以拿到60分了

如果想拿到100，我们就要考虑再优化我们的暴力。

我们先假设i $<$ j
那么上面的公式就可以吧min去掉，变成$i\ast (n-j+1)$
在这种情况下的答案就是$$\sum _{i=1}^{n}i\ast (n-j+1)(i<j)$$
写开就是$$\sum _{i=1}^{n}i\ast n-j\ast i+i$$
我们把i提出来
于是我们可以先用前缀和统计在i之前某个字符出现的位置的累加和
如果要保证$i<j$我们只需要在同一个数组中，统计操作比加到贡献的操作晚一步就行。
对于j>=i，我们反过来再做一遍就行了

这道题告诉我，变量名不要乱写，不然你都不知道自己是怎么错的

60分的代码

#include<bits/stdc++.h>

#define MAXN 200010
#define ll long long

using namespace std;

int n;

char s1[MAXN], s2[MAXN];

ll tot = 0;
ll su1[27], su2[27];
ll ans = 0;

int main()
{
    s1[0] = '-', s2[0] = '-';
    scanf("%d", &n);
    scanf("%s", s1 + 1); scanf("%s", s2 + 1);

    int q = n;
    while(q) tot += q * q, q--;
    
    for(int i = 1; i <= n; ++i){
        ans += (n - i + 1) * su1[s1[i] - 'A'];
        su1[s2[i]- 'A'] += i;
    }
    
    swap(s1, s2);
    
    for(int i = 1; i <= n; ++i){
        su2[s2[i] - 'A'] += i;
        ans += (n - i + 1) * su2[s1[i] - 'A'];
    }
    
    printf("%.6f\n", (double) ans/ tot/ 1.000000);
    
    return 0;
}

AC的代码

#include<bits/stdc++.h>

#define MAXN 200010
#define ll long long

using namespace std;

int n;

char s1[MAXN], s2[MAXN];

double tot = 0;
ll su1[27], su2[27];
double ans = 0;

int main()
{
	s1[0] = '-', s2[0] = '-';
	scanf("%d", &n);
	scanf("%s", s1 + 1); scanf("%s", s2 + 1);

	int q = n;
	while(q) tot += (double)q * q, q--;
	
	for(int i = 1; i <= n; ++i){
		ans += (double)(n - i + 1) * su1[s1[i] - 'A'];
		su1[s2[i]- 'A'] += i;
	}
	
	swap(s1, s2);
	
	for(int i = 1; i <= n; ++i){
		su2[s2[i] - 'A'] += i;
		ans += (double)(n - i + 1) * su2[s1[i] - 'A'];
	}
	
	printf("%.6f\n", (double) ans/ tot/ 1.000000);
	
	return 0;
}