题意:给定一个字符串s,求有多少个三元组(i,j,k)满足i,j,k是等比数列且s[i]=='y'&&s[j]=='r'&&s[k]=='x'。
注意点:公比可能小于1
#include <iostream>
#include <algorithm>
#include <list>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <queue>
#include <vector>
using namespace std;
typedef long long ll;
const ll mod=1e9+7;
const int N= 1e6+20;
char s[N];
int main()
{
int t;
cin>>t;
while(t--)
{
int ans=0;
scanf("%s",s+1);
int len=strlen(s+1);
for(int q=2;q*q<=len;q++)//枚举公比
{
for(int i=1;i*q*q<=len;i++)//起点
{
int x=i,y=x*q,z=y*q;
// Either j|i or j|k
if(s[x]=='y'&&s[y]=='r'&&s[z]=='x')
ans++;
if(s[x]=='x'&&s[y]=='r'&&s[z]=='y')//q'=1/q
ans++;
}
}
cout<<ans<<endl;
}
return 0;
}