洛谷 P1928外星密码

外星密码

题目描述

有了防护伞，并不能完全避免 2012 的灾难。地球防卫小队决定去求助外星种族的帮助。经过很长时间的努力，小队终于收到了外星生命的回信。但是外星人发过来的却是一串密码。只有解开密码，才能知道外星人给的准确回复。解开密码的第一道工序就是解压缩密码，外星人对于连续的若干个相同的子串 $\text{X}$ 会压缩为 $\text{[DX]}$ 的形式（$D$ 是一个整数且 $1\le D\le 99$），比如说字符串 $\text{CBCBCBCB}$ 就压缩为 $\text{[4CB]}$ 或者$\text{[2[2CB]]}$，类似于后面这种压缩之后再压缩的称为二重压缩。如果是 $\text{[2[2[2CB]]]}$ 则是三重的。现在我们给你外星人发送的密码，请你对其进行解压缩。

输入格式

输入一行，一个字符串，表示外星人发送的密码。

输出格式

输出一行，一个字符串，表示解压缩后的结果。

样例 #1

样例输入 #1

AC[3FUN]

样例输出 #1

ACFUNFUNFUN

提示

【数据范围】

对于 $50\%$ 的数据：解压后的字符串长度在 $1000$ 以内，最多只有三重压缩。

对于 $100\%$ 的数据：解压后的字符串长度在 $20000$ 以内，最多只有十重压缩。保证只包含数字、大写字母、[ 和 ]。

#include <bits/stdc++.h>
using namespace std;
//每次读到数字就入栈，
//读到字符暂时存储，遇到】就先检查数字栈是否为空，并拼接一个字符；数字就字符、数字都入栈 ；遇到字符继续拼接； 
stack<int> num;
stack<string> str;
int main(){
	ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
	string ans;
	int n=0;
	char ch;
	while(cin>>ch)
	{
		
		if(ch == '\n') break;
		if((ch>='0')&&(ch<='9'))
		{
			
			n = n*10 + (ch-'0');
		}
		else if(ch == ']')
		{
			int time = num.top();
			num.pop();
			string t = ans;
			for(int i=0;i<time-1;i++) ans+=t;
			if(!str.empty())
			{
			t = str.top();
			str.pop(); 
			ans = t+ans;}
		}
		else if(ch == '[')
		{
			if(ans.size()!=0) 
			{
			str.push(ans);
			ans.clear();
			} 
			if(n!=0)
			{
			num.push(n);
			n = 0;} 
		}
		else 
		{
			if(n!=0)
			{
			num.push(n);
			n = 0;	
			}
			ans+=ch;
		}
	}
	printf("%s",ans.c_str());
	return 0; 
}