字符串
APPAPT
中包含了两个单词PAT
,其中第一个PAT
是第 2 位(P
),第 4 位(A
),第 6 位(T
);第二个PAT
是第 3 位(P
),第 4 位(A
),第 6 位(T
)。现给定字符串,问一共可以形成多少个
PAT
?输入格式:
输入只有一行,包含一个字符串,长度不超过105,只包含
P
、A
、T
三种字母。输出格式:
在一行中输出给定字符串中包含多少个
PAT
。由于结果可能比较大,只输出对 1000000007 取余数的结果。
思路分析:有两种比较简单的方法
1、扫描整个字符串,分别计算PA和AT的次数,两者相加即可,每次遇见‘P'则P++,遇见’A'则PA=(PA+P)即每遇见以此'A'那么‘PA’的组合次数就增加P那么多次,同理可得PAT;
2、扫描整个字符串统计'T'的出现次数countT,再次扫描整个字符串遇见‘P'则countP++,遇见’T‘就执行countT--,每遇见A计算一次countT*countP并累加,核心思路:每遇见一次'A',计算A的左右两边的'P'的次数和‘T’的次数,两者相乘累加到result上
当然题目有提示,注意取余
思路1代码:
#include<bits/stdc++.h>
#define MAX 1000000007
using namespace std;
main(int argc, char *argv[])
{
string a;
cin>>a;
int lena=a.length(),P=0,PA=0,PAT=0;
for(int i=0;i<lena;i++)
{
if(a[i]=='P')
P++;
else if(a[i]=='A')
PA=(PA+P)%MAX;
else if(a[i]=='T')
PAT=(PAT+PA)%MAX;
}
cout<<PAT;
return 0;
}
思路2代码:
#include<bits/stdc++.h>
#define MAX 1000000007
using namespace std;
main(int argc, char *argv[])
{
string a;
cin>>a;
int lena=a.length(),countT=0,countP=0,countA=0,result=0;
for(int i=0;i<lena;i++)
{
if(a[i]=='T')
countT++;
}
for(int i=0;i<lena;i++)
{
if(a[i]=='P')
countP++;
else if(a[i]=='T')
countT--;
else if(a[i]=='A')
result=(result+countT*countP)%MAX;
}
cout<<result;
return 0;
}