【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 = , j = , score = ;
     strLen = str.length();
     while (j < strLen)
     {
         if (str[i] == str[j])
             ++j;
         if (j >= strLen || str[i] != str[j])
         {
             if (j - i > )
                 segments.push(Segment(i, j));
             i = j;
             j = i + ;

         }
         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 = ;
             j = ;
             strLen = str.length();
         }
     }
     return score;
 }
 int main()
 {
     int cnt, strLen, i, j, score;
     string str;

     cin >> cnt;
     while (cnt-- > )
     {
         cin >> str;
         score = ;
         strLen = str.length();
         for (i = ; 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 ;
 }