“21 天好习惯”第一期－13

H xay loves count

题意

在数组 $a$ 中找满足 $a_i\ast a_j=a_k$ 的三元组 $(i,j,k)$

暴力即可（看似是 $O(n^2)$​ 但不是）

#include <bits/stdc++.h>
using namespace std;
#define rep(i, j, k) for (ll(i) = (j); (i) <= (k); (++i))
typedef long long ll;
const int N = 1e6 + 7;
ll a[N];
signed main() {
    ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
    int n, x, mx = 0;
    cin >> n;
    rep(i, 1, n) {
        cin >> x;
        a[x]++;
        mx = max(mx, x);
    }
    ll ans = 0;
    rep(i, 1, mx) {
        rep(j, i, mx) {
            ll x = i * j;
            if (x > mx)
                break;
            ans += a[i] * a[j] * a[x] * (j == i ? 1 : 2);
        }
    }
    cout << ans << '\n';
    return 0;
}