hihoCoder1039 - 字符消除

题目：http://hihocoder.com/problemset/problem/1039

描述

小Hi最近在玩一个字符消除游戏。给定一个只包含大写字母"ABC"的字符串s，消除过程是如下进行的：

1)如果s包含长度超过1的由相同字母组成的子串，那么这些子串会被同时消除，余下的子串拼成新的字符串。例如"ABCCBCCCAA"中"CC","CCC"和"AA"会被同时消除，余下"AB"和"B"拼成新的字符串"ABB"。

2)上述消除会反复一轮一轮进行，直到新的字符串不包含相邻的相同字符为止。例如”ABCCBCCCAA”经过一轮消除得到"ABB"，再经过一轮消除得到"A"

游戏中的每一关小Hi都会面对一个字符串s。在消除开始前小Hi有机会在s中任意位置(第一个字符之前、最后一个字符之后以及相邻两个字符之间)插入任意一个字符('A','B'或者'C')，得到字符串t。t经过一系列消除后，小Hi的得分是消除掉的字符的总数。

请帮助小Hi计算要如何插入字符，才能获得最高得分。

输入

输入第一行是一个整数T(1<=T<=100)，代表测试数据的数量。

之后T行每行一个由'A''B''C'组成的字符串s，长度不超过100。

输出

对于每一行输入的字符串，输出小Hi最高能得到的分数。

提示

第一组数据：在"ABCBCCCAA"的第2个字符后插入'C'得到"ABCCBCCCAA"，消除后得到"A"，总共消除9个字符(包括插入的'C')。

第二组数据："AAA"插入'A'得到"AAAA"，消除后得到""，总共消除4个字符。

第三组数据：无论是插入字符后得到"AABC","ABBC"还是"ABCC"都最多消除2个字符。

样例输入

3
ABCBCCCAA
AAA
ABC

样例输出

9
4
2

思路：

用一个函数计算最后的长度，在每个位置尝试，保存最终长度最小的。

注意，每个位置都要尝试ABC（如BBBAB是在第一个B后面插入A）。

代码：

#include 
#include 
#include 

int CharElimination(const std::string &s)
{
    std::string str = s;
    while(true)
    {
        if(str.size() == 0)
        {
            return 0;
        }
       
        std::vector v;
        char pre = str[0];
        int count = 1;
        for(int i = 1; i < str.size(); ++i)
        {
            if(str[i] == pre)
            {
                ++count;
            }
            else
            {
                if(count == 1)
                {
                    v.push_back(pre);
                }
                pre = str[i];
                count = 1;
            }
        }
        if(count == 1)  // handle last one
        {
            v.push_back(pre);
        }
        
        if(v.size() == str.size())  // end loop
        {
            break;
        }
        
        str = std::string(v.begin(), v.end());
    }
    
    return str.size();
}

int main()
{
    int T;
    std::cin >> T;
    std::string s;
    while(T--)
    {
        std::cin >> s;

        int min = s.size();
        for(int i = 0; i < s.size(); ++i)
        {
            for(char ch = 'A'; ch <= 'C'; ++ch)
            {
                std::string newStr = s.substr(0, i) + ch + s.substr(i);
                int len = CharElimination(newStr);
                if(len < min)
                {
                    min = len;
                }
            }
        }
        
        std::cout << s.size() + 1 - min << std::endl;
    }
    
    return 0;
}