CF567C

题目链接

    题目大意: 给出一个序列, 要求统计所有这样的子序列(长度为3, 是以k为公比的等比级数)
    这样来定义状态 dp[i]   := {i与i/k所有组合情况}
                 nmap[i] := {i的个数}

    如果当前元素i(i % k == 0)
        就直接查看dp[i/k], 获得与i相关的所有情况;
        更新dp[i];
    map当成数组用~~

代码

#include<cstdio>
#include<map>
using namespace std;
long long n, k, t;
long long a;
long long ans = 0;
map<long long, long long> nmap;
map<long long, long long> dp;
int main()
{
    scanf("%I64d%I64d", &n, &k);
    for ( int i = 0; i < n; i ++ )
    {
        scanf("%I64d", &a);
        if ( a % k == 0 )
        {
            t      = a / k;
            ans   += dp[t];
            dp[a] += nmap[t];
        }
        nmap[a] ++;
    }
    printf("%I64d\n", ans);
    return 0;
}