#1039 : 字符消除


from 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

基本思路是模拟消除,先写一个reduce函数出来进行字符串消除;然后在每一个字符直接分别插入 A B C,进行消除。

使用一个指针对字符串从左到右进行处理,每一次只处理当前字符:根据上一次的标记以及与后一个字符是否相等,并对后一个字符保留一个标记。 如果当前字符i和后一个字符i+1相同,那么这两个字符都不要了,标记keep=false;如果不同则复制当前字符i并标识keep=true,移动指针。 Note: 初始化keep=true ,并再最后结束时需要对最后一个字符根据标记进行处理。

int reduce(string &arr){
    if(arr.empty())return 0;
    int pos=0;
    bool keep=true;//flag
    for(int i=0;i<arr.size()-1;i++){
        if(arr[i]==arr[i+1]){
            keep=false;
        }
        else{
            if(keep)arr[pos++]=arr[i];
            keep=true;
        }
    }
    if(last)arr[pos++]=arr[arr.size()-1];
    if(arr.size()==pos)return pos;
    arr.resize(pos);
    return reduce(arr);
}

然后对输入的字符进行插入,模拟缩减即可;这里有一个疑惑:开始想着要优化一下代码,先对字符串进行reduce,然后对剩余的进行插入reduce,这样可以减少不少计算,可是为什么不对呢?还没有想到一个反例,可以hihoCoder上的测试用例跑不过, 下面注释掉的代码就是先进行reduce然后对剩余依次插入。

int main(){
    int nums=0;
    cin>>nums;
    while(nums--){
        string str;
        cin>>str;
        int maximum=1;
        //int before=str.size()-reduce(str);//remove duplicate first
        //cout<<before<<" ";
        char arr[]="ABC";
        for(int i=0;i<str.size();i++){
        	for(int j=0;j<3;j++){
        		string strdup(str);
	            strdup.insert(i,1,arr[j]);
	            int num=strdup.size()-reduce(strdup);
	            maximum=max(maximum,num);
        	}

        }
        //cout<<maximum+before<<endl;
        cout<<maximum<<endl;
    }
    
} 

完整代码Gist https://gist.github.com/86fe534fc3baaa26470d.git

Written with StackEdit.


Haiyang Xu 23 March 2015