传送门
题目大意:求满足倍数是k,长度为3的序列有多少个。
解题思路:对于一个a[i],我们只要知道他的前面有多少个a[i]/k,后面有多少个a[i]*k就行了呀,所有我们用两个map维护,某个数字总共出现了多少次,和已经出现了多少次就行了,然后就是要注意一下,k为1和a[i]为0的情况,这地方我还wa了几发。
#include<bits/stdc++.h>
#define il inline
#define pb push_back
#define fi first
#define se second
#define ms(_data,v) memset(_data,v,sizeof(_data))
#define sc(n) scanf("%d",&n)
#define SC(n,m) scanf("%d %d",&n,&m)
#define sz(a) int((a).size())
#define rep(i,a,b) for(int i=a;i<=b;++i)
using namespace std;
typedef long long ll;
const ll inf=0x3f3f3f3f;
const double PI=acos(-1.0);
const double eps=1e-9;
const int maxn=2e5+5;
ll n,k,a[maxn];
map<ll,ll> all,front; //所有的,前面的
int main() {
std::ios::sync_with_stdio(0);
cin>>n>>k;
rep(i,1,n) cin>>a[i],all[a[i]]++;
ll ans=0;
if(k==1) {
rep(i,1,n){
ans+=front[a[i]]*(all[a[i]]-1-front[a[i]]);
front[a[i]]++;
}
} else {
rep(i,1,n) {
if(a[i]%k==0) {
if(a[i]!=0) ans+=(front[a[i]/k]*(all[a[i]*k]-front[a[i]*k]));
else ans+=(front[0]*(all[0]-front[0]-1));
}
front[a[i]]++;
}
}
cout<<ans<<endl;
return 0;
}