leetcode394.字符串解码

题目大意

给定一个经过编码的字符串，返回它解码后的字符串。

编码规则为: k[encoded_string]，表示其中方括号内部的 encoded_string 正好重复 k 次。注意 k 保证为正整数。

你可以认为输入字符串总是有效的；输入字符串中没有额外的空格，且输入的方括号总是符合格式要求的。

此外，你可以认为原始数据不包含数字，所有的数字只表示重复的次数 k ，例如不会出现像 3a 或 2[4] 的输入。

示例:

s = "3[a]2[bc]", 返回 "aaabcbc".
s = "3[a2[c]]", 返回 "accaccacc".
s = "2[abc]3[cd]ef", 返回 "abcabccdcdcdef".

解题思路

方法一：

准备一个字符串栈和整型栈，用来存放遍历过程中的字符串和整数。
遍历给定字符串的每个位置：

• 如果是0~9，则不断获取数字，记为int；
• 如果是字符，则不断获取字符，记为str；
• 如果是’[’，则将之前获取到的int和str分别压入对应的栈中，并重新初始化int和str；
• 如果是’]’，表示"[xxxx]“中的字符串str已经被记录，弹出整型栈顶元素k，将str重复k次，并重新压入字符串栈中（注意，只有在碰到”[“时才重新初始化int和str，因此”]“时，str就表示”[xxxx]“中的"xxxx”）；

class Solution {
public:
    string decodeString(string s) {
    	if (s.empty()){
    		return "";
    	}

    	stack<int> numStack;
    	stack<string> strStack;
    	int num = 0;
    	string cur = "";

    	for (int i = 0; i < s.size(); ++i){
    		if (s[i] <= '9' && s[i] >= '0'){
    			num = num * 10 + (s[i] - '0');
    		}
    		else if (s[i] == '['){
    			strStack.push(cur);
    			numStack.push(num);
    			num = 0;
    			cur.clear();
    		}
    		else if ((s[i] >='a' && s[i] <= 'z') || (s[i] >='A' && s[i] <= 'Z')){
    			cur += s[i];
    		}
    		else if (s[i] == ']'){
    			int count = numStack.top(); numStack.pop();
    			for (int j = 0; j < count; ++j){
    				strStack.top() += cur;
    			}
    			cur = strStack.top(); strStack.pop();
    		}
    	}
    }
    return cur;
};

方法二：

碰到非"]“元素时，直接压入栈中；
碰到”["元素时：

• 不断弹出栈顶元素，直到栈顶元素是"["，将这些元素拼成字符串；
• 将栈顶的"["弹出后，不断弹出栈顶元素，直到栈为空或者元素不是数字为止，将这些数字拼成整数（表示字符串重复的次数）。
• 将上述字符串重复K次，并重新放回栈中；

给定的原始字符串遍历完毕后，将栈中元素一一弹出，并组合成字符串返回即可。

class Solution {
public:
	string decodeString(string s) {
		if (s.size() == 0) return{};
		stack<string> mystack;
		for (int i = 0; i < s.size(); i++)
		{
			if (s[i] != ']')
			{
				string t{ s[i] };
				mystack.push(t);  // 如果不是右括号，则将当前字符放入栈中
			}

			else  //是右括号，则将左右括号中的字符弹出，并跟前面的数字组成新的字符串
			{
				string curInStr = "";
				while (mystack.top() != "[")  //将中括号中的字符拼成字符串，放入curInStr中
				{
					curInStr = mystack.top() + curInStr;
					mystack.pop();
				}
				mystack.pop();  //将左括号弹出

				string numStr = "";
				while (mystack.size() && mystack.top()[0] >= '0' && mystack.top()[0] <= '9')  //将栈中的数字依次弹出
				{
					numStr = mystack.top() + numStr;
					mystack.pop();
				}

				int count = stoi(numStr);  //表明当前括号中的字符串重复的次数

				string tmpRes = "";  //记录新字符串
				for (int j = 0; j < count; j++)
					tmpRes += curInStr;

				mystack.push(tmpRes);  //将新字符串压入栈中
			}
		}
		string res = "";
		while (mystack.size())
		{
			res = mystack.top() + res;
			mystack.pop();
		}
		return res;
	}
};