1040 有几个PAT (25 分)

题意概述：
字符串 APPAPT 中包含了两个单词 PAT，其中第一个 PAT 是第 2 位§，第 4 位(A)，第 6 位(T)；第二个 PAT 是第 3 位( P)，第 4 位(A)，第 6 位(T)。

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

输入格式：
输入只有一行，包含一个字符串，长度不超过$10^5$ ，只包含 P、A、T 三种字母。

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

输入样例：
APPAPT
结尾无空行

输出样例：
2
结尾无空行

完全自己写的代码：（后3个测试点超时）

#include <stdio.h>
#define N 100001



//计PT间的A 

int main()
{
	char A[N]={'\0'};
	int i,j;
	long long n=0,count_A=0;
	gets(A);
	for (i=0;A[i]!='\0';i++)
	{
		count_A=0;
		if (A[i]!='P')
		    continue;
		else
		{
			for (j=i+1;A[j]!='\0';j++)
			{
				if (A[j]=='A')
					count_A++;
				else if(A[j]=='T')
					n+=count_A;
			}
		}
	}
	printf("%lld",n%1000000007);
	return 0;
}

两重循环，固定P，数每次到T时A的个数，n累加该数，然后P后移，时间复杂度O(n^2)；

改进后代码（参考柳神思路后）：

#include <stdio.h>
#define N 100001

//计A前的P与A后的T 

int main()
{
	char A[N]={'\0'};
	int i,j,n_A=0,number=0;
	long long n=0,count=0,B[N][2]={0};
	gets(A);
	for (i=0,j=0;A[i]!='\0';i++)
	{
		if (A[i]=='A')
		{
			B[j][0]=count;j++;
		}
		else if(A[i]=='P')
			count++;
	}
	count=0;/*清零，这次count计的是A后T的数，逆序数*/
	number=i;n_A=j;
	for (i=number-1,j=n_A-1;i>=0;i--)
	{
		if (A[i]=='A')
		{
			B[j][1]=count;j--;
		}
		else if(A[i]=='T')
			count++;
	}
	for (i=0;i<n_A;i++)
	    n+=B[i][0]*B[i][1];
	printf("%lld",n%1000000007);
	return 0;
}

时间复杂度O(n),顺序计一次P,逆序计一次T，然后把对应计数相乘累加；

Tips:
不要只关注首尾的固定，说不准中间某个关键字有奇效

运行结果：