[PAT] A1093は、PATのクラスをカウント

-15ポイント[バージョン]解決暴力

#include <bits/stdc++.h>
using namespace std;
char s[100010];
int p[100010];
int a[100010];
int t[10010];
int main(void)
{
	int res=0,np=0,na=0,nt=0;	
	scanf("%s",s);
	for(int i=0;i<strlen(s);i++)
	{
		if(s[i]=='P')
		{
			p[np]=i;np++;
		}
		else if(s[i]=='A')
		{
			a[na]=i;na++;
		}
		else if(s[i]=='T')
		{
			t[nt]=i;nt++;
		}
	}
	for(int i=0;i<np;i++)
	{
		for(int j=0;j<na;j++)
		{
			if(p[i]>a[j]) continue;
			for(int k=0;k<nt;k++)
			{
				if(a[j]>t[k]) continue;
				else
				{
					res = (res+nt-k)%1000000007;
					break;
				}
			}
		}
	}
	printf("%d",res);
} 

[O（N）時間複雑度バージョン] -25分
考え：PATの数彼の左を形成する位置Aを決定するため= P * Tの数は、彼の右の数です。したがって後者の計算はO（1）レベルを達成することができることはO（n ^ 2）の時間計算量は、まず、トラバース回文字列全体を回避するために、ことに留意されたいです。

#include <bits/stdc++.h>
using namespace std;
char s[100010];
int p[100010]; //p in left
int t[100010]; //t in left
int main(void)
{
	int res=0,len;	
	scanf("%s",s);
	len=strlen(s);
	p[0]=0;t[0]=0; //initially
	for(int i=0;i<len;i++)
	{
		if(s[i]=='P')
		{
			p[i+1]=p[i]+1;
			t[i+1]=t[i];
		}
		else
		{
			p[i+1]=p[i];
			if(s[i]=='T') t[i+1]=t[i]+1;
			else t[i+1]=t[i];	
		}
	}
	for(int i=0;i<len;i++)
	{
		if(s[i]=='A')
		{
			res = (res+p[i]*(t[len]-t[i]))%1000000007;
		}
	}
	printf("%d",res);
}