一、概述
给出一段字符串,输出按顺序能组成的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;
}