题解:优秀的暴力。。。
首先,很容易想到一种写法每次枚举四个数,计算是否符合,时间复杂度 N^4。
再者,在最后一维可以用一个数组保存每个数出现的次数,优化一维,同时用离散化减少数组大小,N^3。
在思考的过程中,可以发现既然需要满足 abc=d ,那么只有当两数乘积小于最大的数才可能对答案产生贡献。且可以用数组维护一个数,就可以用数组维护两个数的乘积出现的次数。但问题就出现了,如何保证两数的区间是递增——可以线性维护的呢?
由此我们得出正解:
枚举一个 i ,用数组维护i之前两数的乘积,同时再枚举一个 j ,i<j<=n,作为d来查找数组cnt[ a[j] / a[i] ]更新答案,N^2。
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
#define N 1000005
using namespace std;
long long n,a[N],cnt[N],maxn,ans;
int main()
{
cin>>n;
for(int i=1;i<=n;i++){
cin>>a[i];
maxn=max(maxn,a[i]);
}
for(int i=1;i<=n;i++){
for(int j=i+1;j<=n;j++){
if(a[j]>=a[i]&&a[j]%a[i]==0){
ans+=cnt[a[j]/a[i]];
}
}
for(int j=1;j<i;j++){
if(a[i]*a[j]<=maxn){
cnt[a[i]*a[j]]+=1;
}
}
}
cout<<ans<<endl;
return 0;
}