Tarango

Untitled

Aug 5th, 2015
295
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 0.86 KB | None | 0 0
  1. //============================================================================
  2. // Name        : ACM
  3. // Author      : Tarango Khan
  4. // Team        : BRACU Byteheads
  5. //============================================================================
  6.  
  7. #include <bits/stdc++.h>
  8. using namespace std;
  9. #define MAX 200005
  10.  
  11. int A[MAX];
  12. int cnt[MAX];
  13. int divv[MAX];
  14. map<int,int> Map;
  15. int N,K;
  16.  
  17. long long calc(){
  18.     long long res = 0;
  19.     for(int i = 0;i<N;i++){
  20.         int val = A[i];
  21.         if(val % K == 0 && (val/K) % K == 0){
  22.             if(Map[val/K] != 0 && Map[val/K/K]  != 0){
  23.                 res = res  + Map[val/K];
  24.             }
  25.         }
  26.         if(val % K == 0)
  27.             Map[val] = Map[val] + Map[val/K];
  28.         else
  29.             Map[val]++;
  30.     }
  31.     return res;
  32. }
  33.  
  34. int main() {
  35.     scanf("%d %d",&N,&K);
  36.     for(int i = 0;i<N;i++){
  37.         scanf("%d",&A[i]);
  38.     }
  39.     memset(cnt,0,sizeof(cnt));
  40.     long long res = calc();
  41.     cout << res << endl;
  42. }
Advertisement
Add Comment
Please, Sign In to add comment