PAT (Advanced Level) Practice 1093 Count PAT's 遍历一次以保存从起点到某点的数据

一、概述

给出一段字符串，输出按顺序能组成的PAT的个数。

例如APPAPT，按顺序有第2，4，6三位和3，4，6三位可以组成PAT。

按照通常的直接做法就是遍历，看该位是不是P，是P往下遍历找A，是A往下遍历找T。

这得累死。按照最夸张的，PAAAAAAAAAA一大堆AT，查这个PAT得遍历整个字符串。不可接受。

观察可以得到能够组成的PAT的数量是A左边的P的数量乘以A右边的T的数量。

因此，计算出每一位左边的P和每一位右边的T就可以了。只需要遍历三次就可以计算出结果。

二、分析

这道题，乃至这类题都有一个共同的特点：从左往右遍历，第二个的值与第一个有关系，第三个与第二个有关系。。。因此遍历一次可以求出所有待求值。求出之后就可以根据这个来解决问题了。这种思想很有用。关键是求出每一位的某个值，不是找特殊，如果只求A左侧的，只对值为A的位赋值就落了下乘了，失去了递归的性质。

找P的代码如下：

for(it=s.begin();it!=s.end();it++)
  {
    if(*it=='P')
    {
      leftp[num+1]=leftp[num]+1;
    }
    else
    {
      leftp[num+1]=leftp[num];
    }
    num++;
  }

找T的类似。

然后再次遍历并输出即可。

三、总结

这种问题，一般直观的做，都需要每求出一个结果遍历多次，但是时间复杂度不可接受，一般用遍历加递归来做。

PS：代码如下：

#include<cstdio>
#include<stdio.h>
#include<iostream>
#include<string>
#include<cstring>
#include<algorithm>
using namespace std;
string s;
int leftp[100010]={0};
int rightt[100010]={0};
int main()
{
  cin>>s;
  string::iterator it;
  int num=0;
  for(it=s.begin();it!=s.end();it++)
  {
    if(*it=='P')
    {
      leftp[num+1]=leftp[num]+1;
    }
    else
    {
      leftp[num+1]=leftp[num];
    }
    num++;
  }
  num--;
  int MAX=num;
  string::reverse_iterator rit;
  for(rit=s.rbegin();rit!=s.rend();rit++)
  {
    if(*rit=='T')
    {
      rightt[num-1]=rightt[num]+1;
    }
    else
    {
      rightt[num-1]=rightt[num];
    }
    num--;
  }
  long long int ans=0;
  for(int i=1;i<MAX;i++)
  {
  	if(s[i]=='A')
  	{
  		ans+=leftp[i]*rightt[i];
    ans=ans%1000000007;
	  } 
  }
  cout<<ans;
}