Untitled
#include<bits/stdc++.h> using namespace std; int n; long long a[2005]; vector<int> f; vector<pair<int, int>> P[(int)2e6 + 5]; int st[4 * 2005]; void update(int id, int l, int r, int i, int v){ if (i < l || i > r) return; if (l == r){ st[id] += v; if (v == 0) st[id] = 0; return; } int mid = (l + r) >> 1; update(2 * id, l, mid, i, v); update(2 * id + 1,mid + 1, r, i, v); st[id] = st[2 * id] + st[2 * id + 1]; } int get(int id, int l, int r, int u, int v){ if (u > v || r < u || l > v) return 0; if (l >= u && r <= v) return st[id]; int mid = (l + r) >> 1; return get(2 * id, l, mid, u, v) + get(2 * id + 1, mid + 1, r, u, v); } void input(){ cin >> n; for (int i = 1; i <= n; i++) cin >> a[i]; sort(a + 1, a + 1 + n); set<long long> stx; for (int i = 1; i <= n; i++){ for (int j = i + 1; j <= n; j++){ stx.insert(a[i] * a[j]); } } f.clear(); for (auto v : stx){ f.push_back(v); } for (int i = 1; i <= n; i++){ for (int j = i + 1; j <= n; j++){ long long x = a[i] * a[j]; int p = lower_bound(f.begin(), f.end(), x) - f.begin(); P[p].push_back({i, j}); } } } void solve(int testcase){ input(); long long ans = 0; int len = f.size(); for (int z = 0; z < len; z++){ auto u = P[z]; int sz = u.size(); if (sz == 0) continue; sort(u.begin(), u.end()); for (int i = 1; i < sz; i++){ if (u[i].first != u[i - 1].first){ int p = i - 1; while(p >= 0 && u[p].first == u[i - 1].first) update(1, 1, n, u[p].second, 1), p--; } ans += get(1, 1, n, u[i].second + 1, n); } for (int i = 0; i < sz; i++) update(1, 1, n, u[i].second, 0); P[z].clear(); } cout << "#" << testcase << ' ' << ans << '\n'; } int main(){ //freopen("input.txt", "r", stdin); int t = 1; cin >> t; for (int i = 1; i <= t; i++) solve(i); }
Leave a Comment