Codeforces Round 578 (Div. 2) E题 Compress Words(扩展KMP)

题目链接

https://codeforces.com/problemset/problem/1200/E

思路

取当前串 s i s_{i} si的长度与当前答案 a n s ans ans的长度的最小值。

令 s i z = m i n ( ∣ s i ∣ , ∣ a n s ∣ ) siz = min(|s_{i}|,|ans|) siz=min(∣si∣,∣ans∣)。

将 s i s_{i} si的前 s i z siz siz部分和 a n s ans ans的后 s i z siz siz部分进行拼接,然后使用扩展KMP算法求出 z z z数组,根据 z z z数组的含义可以直接计算答案。

代码

cpp 复制代码
#include <bits/stdc++.h>
using namespace std;
#define int long long
const int N = 1e5 + 5;
int n;
string s[N];
// z[i]表示 s 和 s[i:] 匹配的最大前缀长度
vector<int> zfunc(const string& s) {
	int n = s.size();
	vector<int> z(n);
	for (int i = 1, l = 0, r = 0; i < n; i++) {
		if (i <= r) z[i] = min(z[i - l], r - i + 1);
		while (i + z[i] < n && s[i + z[i]] == s[z[i]]) z[i]++;
		if (i + z[i] - 1 > r) l = i, r = i + z[i] - 1;
	}
	z[0] = n;
	return z;
}
void solve()
{
	cin >> n;
	for (int i = 1; i <= n; i++)
	{
		cin >> s[i];
	}
	string ans = s[1];
	for (int i = 2; i <= n; i++)
	{
		string str = s[i];
		for (int j = max((int)((int)(ans.size()) - (int)(s[i].size())), 0ll); j < ans.size(); j++)
		{
			str.push_back(ans[j]);
		}
		vector<int>z = zfunc(str);
		int res = 0;
		for (int j = s[i].size(); j < str.size(); j++)
		{
			if (z[j] + j == str.size())
			{
				res = z[j];
				break;
			}
		}
		for (int j = res; j < s[i].size(); j++)
		{
			ans.push_back(s[i][j]);
		}
	}
	cout << ans << endl;
}
signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(0), cout.tie(0);
	int test = 1;
	// cin >> test;
	for (int i = 1; i <= test; i++)
	{
		solve();
	}
	return 0;
}
相关推荐
我是谁??19 分钟前
C/C++使用AddressSanitizer检测内存错误
c语言·c++
小码农<^_^>21 分钟前
优选算法精品课--滑动窗口算法(一)
算法
羊小猪~~23 分钟前
神经网络基础--什么是正向传播??什么是方向传播??
人工智能·pytorch·python·深度学习·神经网络·算法·机器学习
软工菜鸡1 小时前
预训练语言模型BERT——PaddleNLP中的预训练模型
大数据·人工智能·深度学习·算法·语言模型·自然语言处理·bert
南宫生1 小时前
贪心算法习题其三【力扣】【算法学习day.20】
java·数据结构·学习·算法·leetcode·贪心算法
发霉的闲鱼1 小时前
MFC 重写了listControl类(类名为A),并把双击事件的处理函数定义在A中,主窗口如何接收表格是否被双击
c++·mfc
小c君tt1 小时前
MFC中Excel的导入以及使用步骤
c++·excel·mfc
xiaoxiao涛1 小时前
协程6 --- HOOK
c++·协程
AI视觉网奇1 小时前
sklearn 安装使用笔记
人工智能·算法·sklearn
JingHongB2 小时前
代码随想录算法训练营Day55 | 图论理论基础、深度优先搜索理论基础、卡玛网 98.所有可达路径、797. 所有可能的路径、广度优先搜索理论基础
算法·深度优先·图论