hihoCoder #1039 : 字符消除

题目：

分类：字符串处理

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

描述

小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

思路

枚举所有可能，尝试每一种插入方法，寻得最优值。
注：通过c++自带string类的insert、erase、size等函数简化操作。

insert用法：

string str="123456789";
//在str[1]处插入字符串0，后面子串后移，第二个输入参数须为字符串，不接受字符输入
str.insert(1,"0");
cout << str;

运行结果：

1023456789

erase用法：

string str = "123456789";
str.erase(5, 3);//从str[5]开始删除3个字符
cout << str;

运行结果：

123459

AC代码

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

int main() {
    int T;
    cin >> T;
    while (T--) {
        int score, bestScore = 0;
        string str;
        cin >> str;
        for (int i = 0; i <= str.size(); i++) {
            for (int j = 1; j <= 3; j++) {
                string newStr = str;
                //插入新字母
                string tempStr;
                switch (j) {
                case 1:tempStr = "A"; break;
                case 2:tempStr = "B"; break;
                case 3:tempStr = "C";
                }
                if (i == str.size()) {
                    newStr += tempStr;
                }
                else {
                    newStr.insert(i, tempStr);
                }
                //删除重复字母
                bool change = true;
                while (change&& newStr.size()!=0) {
                    change = false;
                    for (int k = 0; k < newStr.size(); k++) {
                        int length;
                        if (newStr[k] == newStr[k + 1]) {
                            change = true;
                            length = 1;
                            while (newStr[k + length] == newStr[k + length + 1]) {
                                length++;
                            }
                            newStr.erase(k, length + 1);
                            k = k - 1;
                        }
                    }
                }
                //分数
                score = 1+ str.size() - newStr.size();
                if (score > bestScore) {
                    bestScore = score;
                }
            }
        }
        cout << bestScore << endl;
    }
    return 0;
}