题目如下:
思路:
从后往前走一遍,记录下a和b的个数。注意因为找的是“NBA”,a必须在b的后面。所以要将a和b的记录分为多块存储。个人选用了结构体数组的方式。这里需要注意每一块记录数据的a,都应继承前一块记录的a。
每次取到n时,只要计算在此之后每一块记录数据的a,b的组合就行了,也就是a*b。
最后别忘了总数取余。
#include<stdio.h>
#include<string.h>
const long tot=1e5+10;
char a[tot];
struct num
{
long num_a;
long num_b;
};
int main()
{
int n,i;
gets(a);
n=strlen(a);
struct num note[50000];
memset(note,0,sizeof(note));
long long sum=0;
int b[n],f[n],e=0,j;
for(i=n-1;i>=0;--i)
{
if(a[i]=='A')
{
if(note[e].num_b!=0)//判断是否需要切换下一个块记录
{
e++;
note[e].num_a+=note[e-1].num_a;//继承上一次的数据a
}
note[e].num_a++;
}
else if(a[i]=='B')
note[e].num_b++;
else if(a[i]=='N')
for(j=0;j<=e;++j)
sum+=note[j].num_a*note[j].num_b;
if(sum>=1000000007)
sum%=1000000007;
}
printf("%lld",sum);
return 0;
}