1040. 有几个PAT（25）

如题：http://www.patest.cn/contests/pat-b-practise/1040

字符串APPAPT中包含了两个单词“PAT”，其中第一个PAT是第2位(P),第4位(A),第6位(T)；第二个PAT是第3位(P),第4位(A),第6位(T)。

现给定字符串，问一共可以形成多少个PAT？

输入格式：

输入只有一行，包含一个字符串，长度不超过105，只包含P、A、T三种字母。

输出格式：

在一行中输出给定字符串中包含多少个PAT。由于结果可能比较大，只输出对1000000007取余数的结果。

输入样例：

APPAPT

输出样例：

2

#include<iostream>

#include<cstdio>

#include<cstring>

using namespace std;

int n;

int cp[100005];

int ct[100005];

int lowbit(int x)

{

return x&(x^(x-1));

}

void add(int c[],int pos,int x)

{

while(pos<=n)

{

c[pos]+=x;

pos+=lowbit(pos);

}

}

int sum(int c[],int pos)

{

int s=0;

while(pos>0)

{

s+=c[pos];

pos-=lowbit(pos);

}

return s;

}

int main()

{

// freopen("C:\\1.txt","r",stdin);

char str[100005];

cin>>str+1;

int len=strlen(str+1);

int i;

n=len;

for(i=1;i<=len;i++)

{

if(str[i]=='P')

add(cp,i,1);

else if(str[i]=='T')

add(ct,i,1);

}

int s=0;

for(i=1;i<=len;i++)

{

if(str[i]=='A')

{

int p=sum(cp,i-1);

int t=sum(ct,len)-sum(ct,i);

s+=p*t;

s%=1000000007;

}

}

cout<<s<<endl;

return 0;

}