给定n个字符串s[1...n], 求有多少个数对(i, j), 满足i < j 且 s[i] + s[j] == s[j] + s[i]?

题目

思路:

对于字符串a,b, (a.size() < b.size()), 考虑对字符串b满足什么条件:

由1、3可知a是b的前后缀,由2知b有一个周期是3,即a.size(),所以b是用多个a拼接而成的,有因为a是b的前后缀,所以a和b的循环节相同,且a,b均恰好由整数个循环节组成。循环节长度 = 字符串长度 - 最大公共前后缀长度。

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
#define int long long
const int maxn = 1e6 + 5;
string s[maxn];
int nxt[maxn];
map<string, int> mp;
void getnext(string s){
	int j = -1;
	nxt[0] = -1;
	int n = s.size();
	s = ' ' + s;
	for(int i = 1; i <= n; i++){
		while(j >= 0 && s[i] != s[j + 1]){
			j = nxt[j];
		}
		nxt[i] = j + 1;
		j++;
	}
}
// void f(vector<int> &a){
//     a.push_back(6);
// }
signed main(){
    int n;
    while(cin >> n){
//         cin >> n;
        mp.clear();
        int res = 0;
        for(int i = 1; i <= n; i++){
            cin >> s[i];
            getnext(s[i]);
            int m = s[i].size();
            int T = m - nxt[m];
            if(m % T != 0){
                T = m;//整个字符串也是一个循环节
            }
            string t = s[i].substr(0, T);
            res += mp[t];
            mp[t]++;
        }
         cout << res << '\n';
    }
    
//     string s = "666";
//     cout << s.size() << '\n';
//     getnext(s);
//     cout << s.size() << '\n';
//     vector<int> a;
//     cout << a.size() << '\n';
//     f(a);
//     cout << a.size() << '\n';
    return 0;
}
相关推荐
写代码的小球35 分钟前
求模运算符c
算法
大千AI助手4 小时前
DTW模版匹配:弹性对齐的时间序列相似度度量算法
人工智能·算法·机器学习·数据挖掘·模版匹配·dtw模版匹配
YuTaoShao6 小时前
【LeetCode 热题 100】48. 旋转图像——转置+水平翻转
java·算法·leetcode·职场和发展
生态遥感监测笔记6 小时前
GEE利用已有土地利用数据选取样本点并进行分类
人工智能·算法·机器学习·分类·数据挖掘
Tony沈哲6 小时前
macOS 上为 Compose Desktop 构建跨架构图像处理 dylib:OpenCV + libraw + libheif 实践指南
opencv·算法
刘海东刘海东7 小时前
结构型智能科技的关键可行性——信息型智能向结构型智能的转变(修改提纲)
人工智能·算法·机器学习
pumpkin845147 小时前
Rust 调用 C 函数的 FFI
c语言·算法·rust
挺菜的7 小时前
【算法刷题记录(简单题)003】统计大写字母个数(java代码实现)
java·数据结构·算法
mit6.8247 小时前
7.6 优先队列| dijkstra | hash | rust
算法