CodeForces 567C Geometric Progression 类似dp的递推统计方案数

input

n,k　　 1<=n,k<=200000

a1 a2 ... an

1<=ai<=1e9

output

数组中选三个数,且三个数的下标严格递增，凑成形如b,b*k,b*k*k的种数

做法：先将可以作为第三个数的数放到map中，然后再扫一遍依次统计map中的数作为第三个数的种数，第二个数的种数，第三个数的种数

 #include<cstdio>
 #include<map>
 struct node
 {
     int b;//a[i]作为i1的种数
     long long c;//a[i]作为i2的种数
 };
 using namespace std;
 int a[], n, k;
 long long k2, sum;
 map<int,node>q;
 map<int,node>::iterator j;
 int main()
 {
     while (scanf("%d%d", &n, &k) == )
     {
         q.clear();
         k2 = (long long)k*k;
         node u;
         sum = u.b = u.c = ;
         for (int i = ; i < n; i++)
         {
             scanf("%d", &a[i]);
             if (!(a[i] % k2)) q.insert(make_pair(a[i] / k2,u));//统计出所有的i3
         }
         if (q.empty()) { puts("");continue; }
         for (int i = ; i < n; i++)
         {
             if (a[i] % k2 == )//a[i]作为i3
             {
                 j = q.find(a[i]/k2);//找i2
                 if(j!=q.end()) sum += j->second.c;//累加到总方法数
             }
             if (a[i] % k == )//a[i]作为i2
             {
                 j = q.find(a[i] / k);//找i1
                 if (j != q.end()) j->second.c += j->second.b;//累加到i2的方法数
             }
             j = q.find(a[i]);//a[i]作为i1找i1
             if (j != q.end()) j->second.b++;//累加
         }
         printf("%lld\n", sum);
     }
     return ;
 }