辞树的QAQ水题(字符串统计,思维)

思路:统计一串字符有多少个‘A’,并分别统计出每个‘A’前后有多少‘Q’。然后让每个‘A’前后的‘Q’相乘并相加就能得出结果了。

注意:数据的类型,卡了int,要用long long。

还有就是在pc^2上面用gets()读入字符串会WA?

AC代码:

#include<bits/stdc++.h>
#define ll long long
#define ms(a) memset(a,0,sizeof(a))
using namespace std;
const int maxn=1e6+10;
char ch[maxn];
char a[maxn];
ll F[maxn],L[maxn];
int main()
{
//	freopen("in.txt","r",stdin);
//	freopen("11out.txt","w",stdout);
	int t;
	cin>>t;
	while(t--)
	{
		ms(ch);
		ms(a);
		ms(F);
		ms(L);
		scanf("%s",ch);
		int l=strlen(ch);
		int k=0;
		for(int i=0;i<l;i++)
		{
			if(ch[i]=='Q'||ch[i]=='A')
			{
				a[k]=ch[i];
				k+=1;
			}
		}
		int j=0;
		int x=0;
		for(int i=0;i<k;i++)
		{
			if(a[i]=='Q')
			{
				x++;
			}
			else
			{
				F[++j]=x;
			}
		}
		int y=0;
		int cnt=j;
		for(int i=k-1;i>=0;i--)
		{
			if(a[i]=='Q')
				y++;
			else
			{
				L[cnt--]=y;
			}	
		}
		ll ans=0;
		for(int i=1;i<=k;i++)
		{
			ans=ans+F[i]*L[i];
		}
		printf("%lld\n",ans);
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/wang_123_zy/article/details/81191723