codeforces 567C

mp2存放每个数当中间的数的次数，对每个数，如果有a[i]%k==0，那么ans就加上mp2[a[i]/k]，mp表示这个数当第一个数的次数，对每个数，如果有a[i]%k==0，那么mp2[a[i]]就加上mp[a[i]/k]。

#include<bitset>
#include<map>
#include<vector>
#include<cstdio>
#include<iostream>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#include<stack>
#include<queue>
#include<set>
#define inf 0x3f3f3f3f
#define mem(a,x) memset(a,x,sizeof(a))

using namespace std;

typedef long long ll;
typedef pair<int,int> pii;
inline int in()
{
int res=0;
char c;
while((c=getchar())<'0' || c>'9');
while(c>='0' && c<='9')res=res*10+c-'0',c=getchar();
return res;
}
int a[200020];
map<int,ll> mp;
map<int,ll>mp2;

int main()
{
int n=in(),k=in();

for(int i=0;i<n;i++)
{
scanf("%d",&a[i]);
}
ll ans=0;
for(int i=0;i<n;i++)
{

if(a[i]%k==0)
{
ans+=mp2[a[i]/k];
mp2[a[i]]+=mp[a[i]/k];
}
mp[a[i]]++;

}
cout<<ans<<endl;
return 0;
}