【hihoCoder】1039 : 字符消除

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

给定一个字符串s，只包含'A', 'B', 'C'三种字符

1. 向 s 的任意位置 (包括头和尾) 中插入'A', 'B', 'C'中任意一个字符

2. 递归删除 s 中的相同字符，要求第1步的插入位置使得消除的字符最多

递归删除: "ABCCBCCCAA"->"ABB"->"A'

思路

1. 枚举：向 s 中插入一个字符获得新字符串s', 计算s' 可以删除的字符

2. 递归删除

• 用stack<Segment>存储每次可以删除的位置。利用先进后出，在删除的时候不用重新计算位置

源码 

#include <string>
#include <stack>
#include <iostream>
using namespace std;

struct Segment
{
    Segment(int _s, int _e){ start = _s; end = _e; }
    int start;
    int end;
};

int remove(string& str)
{
    stack<Segment> segments;
    int i, j, strLen, score;
    i = 0, j = 1, score = 0;
    strLen = str.length();
    while (j < strLen)
    {
        if (str[i] == str[j])
            ++j;
        if (j >= strLen || str[i] != str[j])
        {
            if (j - i > 1)
                segments.push(Segment(i, j));
            i = j;
            j = i + 1;

        }
        if (j >= strLen)
        {
            if (segments.empty())
                break;
            while (!segments.empty())
            {
                Segment tmp = segments.top();
                segments.pop();
                str.erase(str.begin() + tmp.start, str.begin() + tmp.end);
                score = score + tmp.end - tmp.start;
            }
            i = 0;
            j = 1;
            strLen = str.length();
        }
    }
    return score;
}
int main()
{
    int cnt, strLen, i, j, score;
    string str;
    
    cin >> cnt;
    while (cnt-- > 0)
    {
        cin >> str;
        score = 0;
        strLen = str.length();
        for (i = 0; i < strLen; i++)
        {    
            for (char cha = 'A'; cha <= 'C'; cha++){
                string tmp = str;
                tmp.insert(i, &cha);
                j = remove(tmp);
                if (j > score)
                    score = j;
            }
        }
        cout << score<< endl;
    }
    return 0;
}