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[222222];
int main(){
    map<long long,int> amap,bmap;
    int n;
    long long k;
    scanf("%d%lld",&n,&k);
    for(int i=0; i<n; ++i){
        scanf("%d",a+i);
        ++bmap[a[i]];
    }
    long long ans=0;
    for(int i=0; i<n; ++i){
        --bmap[a[i]];
        if(a[i]%k==0){
            ans+=(long long)amap[a[i]/k]*bmap[a[i]*k];
        }
        ++amap[a[i]];
    }
    printf("%lld",ans);
    return 0;
}