Codeforces Round #Pi (Div. 2) C. Geometric Progression

C. Geometric Progression

题意：

给出一个序列找出多少个长度为三子序列的公差为k（a，a*k，a*k*k）

从后向前更新，分别更新当前每个数a出现的次数、i和i以后所有i*k数量的和、累i后i*k*k的所有数量和的和

例如：(k=3,更新到红3情况)

1 3 9 3 9

<----------

                     前      后         表达式
num[3][2]     0       0           num[3][2]+=num[9][1];
num[3][1]     1       3           num[3][1]+=num[9][0];
num[3][0]     1       2           num[3][0]++;

</pre><p></p><p></p><p><span style="color:#0000ee"><u></u></span></p><p></p><pre name="code" class="cpp">#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<vector>
#include<queue>
#define ll __int64
#define INF 1e9
#define maxn 1000005
using namespace std;
ll a[maxn];
ll b[maxn];
ll ser[maxn];
ll num[maxn][3];
/**
num[i][0]存储i的数量
num[i][1]累加当前i，后面i*k的数量
num[i][2]累加i*k后i*k*k的数量
从后往前更新
最后累加num[i][2]即为所求结果
*/
int A;
int find(ll num){
    return lower_bound(ser,ser+A,num)-ser;
}
int main(){
    ll n,k;
    while(~scanf("%lld%lld",&n,&k)){
        A=0;
        for(int i=0;i<n;i++){
            scanf("%lld",&a[i]);
            b[i]=a[i];
        }
        sort(b,b+n);
        ser[A++]=b[0];
        for(int i=1;i<n;i++)
            if(b[i]!=ser[A-1]) ser[A++]=b[i];
        int p1,p2,p3;
        memset(num,0,sizeof(num));
        for(int i=n-1;i>=0;i--){
            p1=find(a[i]);
            p2=find(a[i]*k);
            p3=find(a[i]*k*k);
            if(p2<A&&ser[p2]==a[i]*k){
                num[p1][2]+=num[p2][1];
                num[p1][1]+=num[p2][0];
            }
            num[p1][0]++;
        }
        ll ans=0;
        for(int i=0;i<A;i++)
            ans+=num[i][2];
        printf("%lld\n",ans);
    }
    return 0;
}