Codeforces 451D 计数问题

题目链接：http://codeforces.com/contest/451/problem/D

题目思路：首先我是先尝试，然后因为状态比较差，竟然只发现了合并后a,b,aba,bab这四种情况的好串，然后提交显然是wrong answer，然后猛地发现ababa之类的也可以，后来就发现合并后长度是奇数的都可以，然后就开始暴力枚举长度为奇数，然后考虑这个串的两端，然后发现是T了，再仔细一看，好像只要合并后两端是同一个字母，那他就是好串，接着就想到了，考虑任意两个相同的字母，他都是一个好串，然后就可以通过左边来计数了

AC代码

#include<bits/stdc++.h>
using namespace std;


int main()
{
    string str;
    cin>>str;
    int cnta[2]={0};
    int cntb[2]={0};
    long long ans1=0;
    long long ans2=0;
    for(int i=0;i<str.size();++i)
    {
        if(str[i]=='a')
        {
            if(i&1)
            {
                ans2+=cnta[0];
                ans1+=cnta[1];
                cnta[1]++;
            }
            else
            {
                ans2+=cnta[1];
                ans1+=cnta[0];
                cnta[0]++;
            }
        }
        else
        {
            if(i&1)
            {
                ans2+=cntb[0];
                ans1+=cntb[1];
                cntb[1]++;
            }
            else
            {
                ans2+=cntb[1];
                ans1+=cntb[0];
                cntb[0]++;
            }
        }
    }
    ans1+=str.size();
    cout<<ans2<<" "<<ans1<<endl;
    return 0;
}