有几个PAT

今天见到的一道题,写了很久却一直运行超时,很佩服标准答案。

题目

字符串APPAPT中包含了两个单词“PAT”,其中第一个PAT是第2位(P),第4位(A),第6位(T);第二个PAT是第3位 (P),第4位(A),第6位(T)。

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

输入描述

输入只有一行,包含一个字符串,长度不超过105,只包含P、A、T三种字母。

输出描述

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

例子

APPAPT
2

思考
  • 我最初的想法是先寻找字符 ‘A’ ,用到了string类的find方法,再分别计算该字符前面 P 的个数和后面 T 的个数,二者相乘即为由这个 A 组成的所有PAT。应该是o(n2)的复杂度。
for(pos = s.find('A',pos); pos != -1; pos = s.find('A',pos)){
        num_of_P = find_num(s,'P',0,pos); //计算字符串s中,0到pos范围内P的个数
        num_of_T = find_num(s,'T',pos+1,s.length()+1);
        total += num_of_P * num_of_T;
        pos++;  //pos++是为了让下次find从下一个位置开始查找
}
  • 可是对于105长度的字符串而言,n2还是有点难以接受了,不妨反过来想。需要统计的是 PAT 的个数,那么每当遇到一个字符 T 时,我理解为一个有穷自动机的接受状态,即一次匹配的终点。此时需要统计的便是在字符 T 前面 PA 出现的个数。同理,可以同样认为 A 是 PA 匹配的终点,便需要统计在这个 A 前面 P 的个数。问题就这样在o(n)复杂度内得到了解决。看来我还得重新看看动态规划。
#include<iostream>
#include<string>
using namespace std;
int main()
{
	string s;
	int p,pa,pat,i;
	cin >> s;
	for(i=p=pa=pat=0; i<s.length(); i++){
		if(s[i] == 'P')
			p++;
		if(s[i] == 'A')
			pa += p;
		if(s[i] == 'T')
			pat += pa;
		pat = pat % 1000000007;
	}
	cout << pat;
}
  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值