UESTC_How many good substrings CDOJ 1026

Icerain likes strings very much. Especially the strings only consist of 0 and 1,she call them easy strings. One day, she is so boring that she want to find how many good substrings in an easy string?

A good substring is a substring which can be a palindrome string after you change any two characters' positions(you can do this operation any times).For example,100 is a good substring of 1001,beacuse you can change it to be 010,which is a palindrome string.

Input

The first line is the number of test cases. (no more than 100)

Each test case has one line containing one string only consist of 0 and 1. The length of the string is no greater than 100000.

Output

For each test case, output one line contains the number of good substrings.

Sample input and output

Sample Input	Sample Output
2 01 01101	2 11

 

解题报告

前缀dp...感谢卿神指导..

 

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 1e5 + 5;
int dp[4];
char buffer[maxn];

int main(int argc,char *argv[])
{
  int Case;
  scanf("%d",&Case);
  while(Case--)
   {
         scanf("%s",buffer);
         int len = strlen(buffer);
         long long ans = 0;
         memset(dp,0,sizeof(dp));
         for(int i = 1 ; i <= len ; ++ i)
          {
                int flag = 0;
                if (buffer[i-1] == '1')
                 {
                       swap(dp[0],dp[1]);
                       swap(dp[2],dp[3]);
                       flag = 1;
           }
          else 
           {
                 swap(dp[0],dp[3]);
                 swap(dp[1],dp[2]);
           }
          if (flag)
           dp[2] ++ ;
          else
           dp[0] ++ ;
          ans += (dp[0] + dp[2] + dp[3]);
       }
      cout << ans << endl;
   }
  return 0;
}