CF 1200 E Compress Words(扩展kmp)

CF 1200 E. Compress Words(扩展kmp)

Problem - E - Codeforces

大意:给出 n 个字符串 , 有一个结果串 , 依次合并第 i 个串和结果串 , 合并的过程中去除结果串的后缀和第i个串前缀的最长公共部分 , 求结果串。

思路:一个串的前缀和另一个串的后缀 , 很容易让人想到 扩展 kmp 算法。

对于 结果串 和 当前要合并的串 , 不难想出可以用扩展 kmp求出结果串后缀对于与合并串的LCP = 结果串后缀的最大长度 , 这就是我们要的答案。

但是这样显然会 T 飞 , 因为随着不断合并 , 结果串越来越长 ,时间复杂度会上升到 n方级别 。 考虑优化 , 我们要求的最长公共部分显然不会超过 min(结果串 , 当前串) , 所以当结果串很长的时候 , 我们只需要用结果串的有效后缀长度去做 扩展 kmp 即可 ,由于所有串总长不超过 1e6 , 所以这样优化之后跑得飞快。

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
#define fi first
#define se second
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
#define int long long
const int N = 1e5 + 10;
const int M = 1e6 + 10;
const int mod = 1e9 + 7;
typedef pair<int,int>PII;

int n;
string s[N] , ans , now;
int z[M] , p[M];

void get_z(string s , string t){
	
	int n = s.size();int m = t.size();
	s = ' ' + s;t = ' ' + t;
	z[1] = n;
	for(int i = 2 , l = 1 , r = 0 ; i <= n ; i ++){
		if(i <= r) z[i] = min(z[i - l + 1] , r - i + 1);
		else z[i] = 0;
		while(s[z[i] + 1] == s[i + z[i]]) z[i] += 1;
		if(i + z[i] - 1 > r) l = i , r = i + z[i] - 1;
	}
	
	for(int i = 1 , l = 1 , r = 0 ; i <= m ; i ++){
		if(i <= r) p[i] = min(z[i - l + 1] , r - i + 1);
		else p[i] = 0;
		while(p[i] + 1 <= n && i + p[i] - 1 <= m && s[p[i] + 1] == t[p[i] + i]) p[i] += 1;
		if(i + p[i] - 1 > r) l = i , r = i + p[i] - 1;  
	}
}

signed main(){

	IOS
	cin >> n;
	for(int i = 1 ; i <= n ; i ++) cin >> s[i];
	ans += s[1];
	for(int i = 1 ; i <= n - 1 ; i ++){
		int x = ans.size();
		int y = s[i + 1].size();
		if(x > y) now = ans.substr(x - y);
		else now = ans;
		get_z(s[i + 1] , now);
		int m = now.size();
		int mx = 0;
		for(int j = 1 ; j <= m ; j ++){
			if(m - j + 1 == p[j]) mx = max(mx , p[j]); 
		}
		ans += s[i + 1].substr(mx);
	}
	
	cout << ans << "\n";

	return 0;
}
//freopen("文件名.in","r",stdin);
//freopen("文件名.out","w",stdout);
相关推荐
索迪迈科技1 小时前
Flink Task线程处理模型:Mailbox
java·大数据·开发语言·数据结构·算法·flink
元亓亓亓2 小时前
LeetCode热题100--230. 二叉搜索树中第 K 小的元素--中等
算法·leetcode·职场和发展
草莓熊Lotso2 小时前
《算法闯关指南:优选算法-双指针》--01移动零,02复写零
c语言·c++·经验分享·算法·leetcode
焜昱错眩..3 小时前
代码随想录算法训练营第三十九天|62.不同路径 63.不同路径ll
算法
焦耳加热7 小时前
阿德莱德大学Nat. Commun.:盐模板策略实现废弃塑料到单原子催化剂的高值转化,推动环境与能源催化应用
人工智能·算法·机器学习·能源·材料工程
wan5555cn7 小时前
多张图片生成视频模型技术深度解析
人工智能·笔记·深度学习·算法·音视频
u6067 小时前
常用排序算法核心知识点梳理
算法·排序
蒋星熠9 小时前
Flutter跨平台工程实践与原理透视:从渲染引擎到高质产物
开发语言·python·算法·flutter·设计模式·性能优化·硬件工程
小欣加油10 小时前
leetcode 面试题01.02判定是否互为字符重排
数据结构·c++·算法·leetcode·职场和发展
3Cloudream10 小时前
LeetCode 003. 无重复字符的最长子串 - 滑动窗口与哈希表详解
算法·leetcode·字符串·双指针·滑动窗口·哈希表·中等