美团的笔试题——字符串a和b

解题思路：

S为长串，T为短串

1、先计算T的长度为nt；

2、然后计算S中去掉最后nt-1个字符的字串中a、b的个数保存在na、nb中

3、开始循环

4、判断T的第一个字符，若该字符为a则加上nb，反之加na
继续判断字串结束的下一个字符，若为a，na++，反之nb++；

6、去第四步看T的下一个字符，直到T遍历结束

7、输出，程序结束

这是我在赛码网上写的解题思路，接下来是今天刚写的代码：

#include<iostream>
#include<string>

using namespace std;

int main()
{
    string S,T;
    cin>>S>>T;
    int ret=0;           //结果

    int nt=T.length();
    int na,nb;          //保存字符a、b的个数
    na=nb=0;

    int f,r;            //子串的首尾下标

    for(r=0; S[r+nt-1]!='\0'; r++)
    {
        if(S[r]=='a')   na++;
        else    nb++;
    }
    //此时r指向了子串之外的一个字符，集该字符不在当前子串中
    f=0;
    int i=0;    //这里i其实和f是相等的
    while(T[i]!='\0')
    {
        //与T串匹配
        if(T[i]=='a')       ret+=nb;
        else                ret+=na;
        //下一个子串与上一个子串相比，
        //第一位少了一个字符，最后多了一个字符，我们要把他计算下来
        if(S[f++]=='a')     na--;
        else                nb--;

        if(S[r++]=='a')     na++;
        else                nb++;

        i++;
    }
    cout<<ret;
    return 0;
}