【Codeforces 805】D. Minimum number of steps

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

思路：从左到右进行遍历，当遇到b的时候，计算前面有多少个a，
然后求出转换从str[0]到这个位置的子字符串变化需要次数，是2^numA-1次，
写出下面代码，结果TLE。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
char str[1000002];
const int mod= 1e9+7;
__int64 findNum(int n){
    __int64 num=1;
    while(n--){
        num=num*2;
        num=num%mod;
    }
    return num-1;
}
int main()
{
    int n;
    scanf("%s",str);
    int numA=0,numB=0;
    __int64 sum=0;
    for(int i=0;i<strlen(str);i++){
        if(str[i]=='a'){
            numA++;
        }else{
            sum+=findNum(numA);
            sum=sum%mod;
        }
    }
    printf("%I64d\n",sum);
    return 0;
}

可以发现findNum有很多重复计算，时间复杂度变成O(N^2)了， 优化后的代码为。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

const int mod= 1e9+7;
int main()
{
    int n;
    string s;
    cin>>s;
    int numA=0,numB=0,t=1;
    __int64 sum=0;
    for(int i=0;i<s.size();i++){
        if(s[i]=='a'){
            numA++;
            t=(t<<1)%mod;
        }else{

            sum+=t-1;
            sum=sum%mod;
        }
    }
    printf("%I64d\n",sum);
    return 0;
}