问题 C: 字符串匹配

发布于:2025-07-28 ⋅ 阅读:(15) ⋅ 点赞:(0)
题目描述

这天,喜欢学习的小 B 学到了跟字符串有关的算法,也就是字符串匹配。但是这个字符串匹配的算法跟常见的匹配有所不同,它回传的讯息是两个字符串当中有几个位置的字符是相同的,我们称为「同位置字符串匹配」。举例来说,"NPSC"跟"NTU"这两个字符串(不含双引号)的同位置字符串匹配是 1,因为两者的相同之处只有第⼀个字符 N是相同的。而" abcdabc "跟" abcdcba"(不含双引号)的同位置字符串匹配是 5,因为这两个字符串在第 1,2,3,4,6 这五个位置的字符是⼀样的。

现在,刚学完这个新潮字符串算法的小 B 兴致勃勃地研究出了同位置字符串匹配的进阶版本:给定 n 个字符串 s1,s2,··· ,sn,求出这 n 个字符串当中两两字符串的「同位置字符串匹配」结果的加总。换句话说,假设两个字符串 a,b 的同位置字符串匹配是 f(a,b),那么小 B 的算法可以求出

 (也就是 (f(s1,s2) + f(s1,s3) + ··· + f(s1,sn)) + (f(s2,s3) + f(s2,s4) +··· + f(s2,sn)) + ··· + (f(sn−1,sn)) )是多少。

由于小 B 无法验证他写的这个新算法的程序是否正确,于是小 B 前来寻求你的帮助。请帮助他算出每⼀笔问题的答案,让小 B 知道他的程序是否正确。


 

输入

输⼊的第⼀行是⼀个整数 n,代表字符串的数量。接下来的 n 行,每⼀行有⼀个字符串 si。

输出

请输出小 B 新研究出的算法对于这 n 个字符串作为输入的输出

样例输入 Copy
【输入样例1】
2
NTU
NPSC

【输入样例2】
5
meow
Meow
MEOW
MeoW
meoW
样例输出 Copy
【输出样例1】
1

【输出样例2】
20
提示

【数据规模】

  • 1 ≤ n ≤ 106
  •  每个字符串的长度均介于 1 到 106 之间
  •  每个字符串均只包含大写英文字母、小写英文字母或是数字
  •  所有字符串的长度加总不会超过 5 × 106

只要检查每个位置相同的字符各自有多少个,然后累加计算

(再也不随便乱看T-T。。。。

代码 

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int n;
string s[1000009];
ll ans=0;
ll jab(ll x){
	return (x-1)*x/2;
}
int main(){
	cin>>n;
	for(int i=0;i<n;++i)cin>>s[i];
	int l=1;
	while(l){
		int f=0,c1[30]{0},c2[30]{0},c3[30]{0};
		for(int i=0;i<n;++i){
			if(s[i].length()<l)continue;
			f=1;
			char x=s[i][l-1];
			if(x>='a'&&x<='z')c1[x-'a']++;
			else if(x>='A'&&x<='Z')c2[x-'A']++;
			else c3[x-'0']++;
		}
		for(int i=0;i<26;++i){
			ans+=jab((ll)c1[i]);
			ans+=jab((ll)c2[i]);
			ans+=jab((ll)c3[i]);
		}
		if(!f)break;
		l++;
	}
    cout<<ans;
	return 0;
}


网站公告

今日签到

点亮在社区的每一天
去签到