Best Compression Algorithms（网易游戏2015笔试题）

Question

时间限制:10000ms
单点时限:1000ms
内存限制:256MB

描述

易信是由网易和电信联合开发的一款即时通讯软件。除了语音聊天，免费电话等新功能以外，传统的文字信息聊天功能也得以保留，因此每天都有大量的文字信息需要在服务器中存储，中转。
小Y是一名负责处理文字信息的易信工程师，每天他都要和字符串打交道。为了提高存储和传输效率，小Y在课余时间经常会去研究字符串的存储方法。通过内部使用的一种统一的加密算法，所有的文字信息首先都会被转化成只包含大写字母['A'...'Z']的字符串。为了压缩这个加密后的字符串，小Y最近想出了两个存储的规则：
(1)如果字符串中有连续相同的大写字母，它们可以选择用"字符+出现次数"的方式替代。如字符串'AABCCCCDD'，可以用'A2BC4D2'表示，也可以用'A2BC2C2DD'表示。
(2)如果字符串中有连续出现的模式串(模式串长度大于1)，它们可以选择用"(模式)+出现次数"的方式替代。如字符串'FABCABCABCE'，可以用'F(ABC)3E'表示，也可以用'F(ABC)2ABCE'表示。
上述规则中的"连续"均指出现次数大于1，规则(2)中的括号后一定是一个大于1的数值，代表出现次数。
综合上述两个规则，字符串'AABAABFAABAABFG'可以用'((A2B)2F)2G'表示。小Y保证输出的压缩串符合上述的两个规则，以下类型的非法字符串不会出现：
'(A)5': 括号冗余
'A1A4': 数字1冗余
'A((AA))2': 括号冗余
(ABC)1: 括号和数字1冗余
对于给定的一个用上述规则压缩后的字符串，对应的原串是唯一的。小Y想知道这个字符串原来的长度是多少，以此计算压缩倍率。你能帮助他吗？

输入

第一行是整数T(T <= 100)，表示下面有T组数据。之后T行，每行为一组数据，每组数据为一个字符串。
每个字符串只会包含26个大写字母['A'...'Z']，数字['0'...'9']和左右括号['(', ')']。
保证输入数据一定合法，括号内的模式串长度一定大于1，表示出现次数的数字一定大于1，且字符串长度L <= 100。

输出

输出T行，每行对应一个数据的输出结果，表示字符串的实际长度。
保证每个字符串展开后的长度不超过109。

样例输入

4
(AA)2A
((A2B)2)2G
WANGYI
A2BC4D2

样例输出

5
13
6
9

My Solution

#include <iostream>
#include <vector>
#include <string>

using namespace std;

/*得到数字，并让指针指向最后一个数*/
int getDig(const string& str, int& idx)
{
int dig = 0;
for (; idx < str.size(); ++idx)
{
if (str[idx] >= '0' && str[idx] <= '9')	//数字
{
dig = dig * 10 + str[idx] - '0';
}
else    // 不是数字
{
--idx;
break;
}
}
return dig;
}

int countLength(const string& str, int& idx)
{
int score = 0;
int sSub = 0;
for (; idx < str.size(); ++idx)
{
char c = str[idx];
if (c >= 'A' && c <= 'Z')	//字符
{
++score;
}
else if (c == '(')
{
++idx;
sSub = countLength(str, idx);
}
else if (c == ')')
{
return score;
}
else    // 数字
{
if (str[idx - 1] == ')')	//跟在模式串后面
{
int dig = getDig(str, idx);
score += dig * sSub;
}
else
{
int dig = getDig(str, idx);
score += dig - 1;
}
}
}
return score;
}

int main()
{
/*读取字符串*/
int t;
vector<string> lines;
cin >> t;
for (int i = 0; i < t; ++i)
{
string line;
cin >> line;
lines.push_back(line);
}

for (auto str : lines)
{
int start = 0;
cout << countLength(str, start) << endl;
}

return 0;
}