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？

输入格式：

输入只有一行，包含一个字符串，长度不超过10⁵，只包含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;
}