Codeforces 567C Geometric Progression（思路）

题目大概说给一个整数序列，问里面有几个包含三个数字的子序列ai,aj,ak，满足a_i*k*k=a_j*k=a_k。

感觉很多种做法的样子，我想到这么一种：

• 枚举中间的a_j，看它左边有多少个a_j/k右边有多少个a_j*k，两边个数的乘积就是答案的一部分贡献。
• 而左边各个数字的个数和右边各个数字可以用两个map维护，一边枚举一边删除或插入。

 #include<cstdio>
 #include<map>
 #include<algorithm>
 using namespace std;
 int a[];
 int main(){
     map<long long,int> amap,bmap;
     int n;
     long long k;
     scanf("%d%lld",&n,&k);
     for(int i=; i<n; ++i){
         scanf("%d",a+i);
         ++bmap[a[i]];
     }
     long long ans=;
     for(int i=; i<n; ++i){
         --bmap[a[i]];
         if(a[i]%k==){
             ans+=(long long)amap[a[i]/k]*bmap[a[i]*k];
         }
         ++amap[a[i]];
     }
     printf("%lld",ans);
     return ;
 }