[校招] 压缩算法

压缩算法 - 腾讯2020校招

小Q想要给他的朋友发送一个神秘字符串，但是他发现字符串的过于长了，于是小Q发明了一种压缩算法对字符串中重复的部分进行了压缩

对于字符串中连续的m个相同字符串S将会压缩为 [m | S ] (m为一个整数且1<=m<=100)，例如字符串ABCABCABC将会被压缩为[3|ABC]，现在小Q的同学收到了小Q发送过来的字符串，你能帮助他进行解压缩么？

输入描述:

输入第一行包含一个字符串s，代表压缩后的字符串。
S的长度<=1000;
S仅包含大写字母、[、]、|;
解压后的字符串长度不超过100000;
压缩递归层数不超过10层;

输出描述:

输出一个字符串，代表解压后的字符串。

输入样例:

HG[3|B[2|CA]]F

输出样例:

HGBCACABCACABCACAF

思路：

由内向外替换，right 代表从左侧第一个开始的 ']' 的位置, left 代表与当前 ']' 对应的 '[' 位置。

k 记录 '|' 位置。替换结束后 left 回到 right 的位置开始新一轮替换。

优化：

经评论区dalao提醒，可以在遍历时将 ' [ ' 和 ' |' 的位置分别入栈

当读到 ' ] ' 时弹出，替换完毕后根据 left 的位置和串s2的长计算新的 right 的位置。省去了不必要的遍历。感谢dalao。

代码（优化后）：

#include <iostream>
#include <string>
#include <stack>
using namespace std;

int main() {
	string str;
	cin >> str;
	stack<int> s;
	for (int right = 0; right < str.length(); right++){
		//遇到 '[' 或 '|' 时入栈
		if (str[right] == '[' || str[right] == '|')
			s.push(right);
		if (str[right] == ']'){
			//出栈
			int k    = s.top(); s.pop();
			int left = s.top(); s.pop();
			int num = stoi(str.substr(left + 1, k- left));
			string s1 = str.substr(k + 1, right - k - 1);
			string s2;
			for (int i = 0; i < num; i++)
				s2 += s1;
			str = str.replace(left, right - left + 1, s2);
			//计算新的right的位置
			right = left+s2.size()-1;
		}
	}
	cout << str;
}

代码（优化前）：

#include <iostream>
#include <string>
using namespace std;

int main() {
	string str;
	cin >> str;
	for (int right = 0; right < str.length(); right++) 
	{
		if (str[right] == ']') 
		{
			int k = 0;
			int left = right;
			for (; str[left] != '['; left--) 
				if (str[left] == '|')
					k = left;
			int num = stoi(str.substr(left + 1, k - left));
			string s1 = str.substr(k + 1, right - k - 1);
			string s2;
			for (int i = 0; i < num; i++)
				s2 += s1;
			str = str.replace(left, right - left + 1, s2);
			right = left;
		}
	}
	cout << str;
}