题目链接
思路:考虑一下每个数对答案的贡献,对第i个数a【i】如果是k的倍数的话,那么a【i】的贡献就是前面a【i】/k的数个数,dp【i】代表【1,i-1】中a【i】/k的个数。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=2e5+1;
ll dp[maxn],a[maxn];
map<ll,ll>p;
int main()
{
int n,k;
scanf("%d %d",&n,&k);
for(int i=1;i<=n;++i)
{
scanf("%lld",&a[i]);
if(a[i]%k==0) dp[i]=p[a[i]/k];
else dp[i]=0;
p[a[i]]++;
}
p.clear();
ll ans=0;
for(int i=1;i<=n;++i)
{
if(a[i]%k==0) ans+=p[a[i]/k];
p[a[i]]+=dp[i];
}
printf("%lld\n",ans);
}