[BC#89A]Fxx and string

最新推荐文章于 2019-07-14 10:33:05 发布

原创最新推荐文章于 2019-07-14 10:33:05 发布 · 797 阅读

0 ·

CC 4.0 BY-SA版权

暴力专栏收录该内容

57 篇文章

订阅专栏

本文介绍一种算法，用于解决给定字符串中特定字符形成的等比数列三元组计数问题。通过枚举i和j来确定k，实现nlogn复杂度的解决方案。

题目大意

给定一个字符串求有多少三元组(i,j,k)满足i、j、k形成等比数列，且s[i]=’y’，s[j]=’r’，s[k]=’x’

瞎做

枚举i和j即可求k，这个枚举是n log n的。
注意不一定i

#include<cstdio>
#include<algorithm>
#include<cstring>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const int maxn=10000+10;
char s[maxn];
int i,j,k,l,t,n,m,ca;
ll ans;
int main(){
    scanf("%d",&ca);
    while (ca--){
        scanf("%s",s+1);
        n=strlen(s+1);
        ans=0;
        fo(i,1,n)
            if (s[i]=='y')
                fo(j,2,n/i){
                    if ((ll)i*j*j>n) break;
                    if (s[i*j]!='r') continue;
                    if (s[i*j*j]=='x') ans++;
                }
        fo(i,1,n)
            if (s[i]=='x')
                fo(j,2,n/i){
                    if ((ll)i*j*j>n) break;
                    if (s[i*j]!='r') continue;
                    if (s[i*j*j]=='y') ans++;
                }
        printf("%lld\n",ans);
    }
}