HDU 5944 Fxx and string(数学)

问题描述

青年理论计算机科学家Fxx得到了一个只包含小写字母的字符串。

字符串的长度为$n$，下标从1开始，第$i$位的字母为\:s_is​i​​，现在Fxx想知道有多少三元组$(i,j,k)$满足下列条件

1、$i,j,k$三个数成等比数列

2、s_i=s​i​​='$y$',s_j=,s​j​​='$r$',s_k=,s​k​​='$x$'

3.$i/j$和$k/j$中必须有整数

输入描述

第一行一个整数$T(1\le T\le 100)$表示数据组数。

接下来$T$行，每行一个仅包含小写字母的字符串（字符串长度不超过$10000$）

输出描述

输出共$T$行。

每行一个整数表示答案。

输入样例

2
xyyrxx
yyrrxxxxx

输出样例

0
2

注意：公比可能小于一

代码如下：

#include<cstdio>
#include<cstring>
#include<algorithm> 
using namespace std;
char a[100010];
int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%s",a);
		int len=strlen(a);
	    int q,i,j,k;
	    int num=0;
		for(int p=0;p<len;p++)
		{
           int i=p+1;
		   for(q=2;;q++)
		   {
	         j=i*q;
			 k=j*q;
			 if(k>len)
			 break;
			 if(a[i-1]=='y'&&a[j-1]=='r'&&a[k-1]=='x')
			 {
			 	num++;
			 }
			 if(a[i-1]=='x'&&a[j-1]=='r'&&a[k-1]=='y')//公比小于1的情况 
			 {
			 	num++;
			 }
		   }		
		   
		}
		printf("%d\n",num);	
	} 
}