CCUT应用OJ——小龙的字符串函数

题目简介

• 题源：1073 - 小龙的字符串函数 | CCUT OJ
• 题意：给定 \(n\) 个等长字符串，定义函数 \(f(s_i,s_j)\) 表示字符串 \(s_i\) 与 \(s_j\) 中位置和字符相同的总数。输出 \(\sum f(s_i,s_j)\) ( 其中 \(i<j\) )。
• 数据范围：\(1\le n\le 2000,1\le |s_i|\le 2000\)
• 注：若无特殊说明，博主的代码模板如下，通过 solve 函数处理多组测试用例。本文后续代码仅给出 solve 函数。

#include <bits/stdc++.h>
using namespace std;
using i64 = long long;
#define ln '\n'

int solve(){

}

int main(){
	ios::sync_with_stdio(0),cin.tie(0);
	int T;cin>>T;
	while(T--){
		cout<<solve()<<ln;
	}
	return 0;
}

朴素想法

朴素想法极其简单，双循环两两遍历字符串，逐位检查是否相同即可。复杂度 \(O(n^2 \cdot |s_i|)\)，超时。

int solve(){
	int n;cin >> n;
	vector<string> strs(n);
	for (auto &i:strs) cin >> i;
	int L = strs[0].size();
	i64 ans = 0;
	for (int i = 0; i < n; i++) 
	    for (int j = i + 1; j < n; j++) 
	        for (int k = 0; k < L; k++) 
	            if (strs[i][k] == strs[j][k]) 
	                ans++;
	return ans;
}

题解

定义权值数组 cnt[i][26]，表示这些字符串第 \(i\) 位上各字母出现的频率。根据排列组合知识，假设字符 \(j\) 在第 \(i\) 位上出现了 cnt[i][j] 次，则其在该位上可两两配对的次数为\(C_{cnt[i][j]}^2=\dfrac{cnt[i][j]\cdot (cnt[i][j]-1)}{2}\) 次。时间复杂度 \(O(n \cdot|s_i|\cdot 26)\)。

int solve() {
	int n; cin >> n;
	vector<string> strs(n);
	for (auto &i:strs) cin >> i;
	int L = strs[0].size();
	int cnt[2000][26] = {0};
	for (int k = 0; k < n; k++)
	    for (int i = 0; i < L; i++)
	        cnt[i][strs[k][i]-'a']++;
	i64 ans = 0;
	for (int i = 0; i < L; i++)
	    for (int j = 0; j < 26; j++)
	        ans += 1LL * cnt[i][j] * (cnt[i][j]-1) / 2;
	return ans;
}