分析:
题目中的格式有点像韦达定理,就是对于一元二次方程ax^2 + bx + c = 0有
所以可以推出要找的就是两个点,可以直接二分查找存不存在,这题有很多边界问题,有b^2 - 4ac小于0或者等于0,或者求出来的根在数组中找不到,都要处理。
代码:
#include <bits/stdc++.h>
using namespace std;
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
int T;
cin >> T;
while(T --) {
int n;
cin >> n;
map<long long, long long> cnt;
for(int i = 0; i < n; i ++) {
long long x;
cin >> x;
cnt[x] ++;
}
int q;
cin >> q;
while(q --) {
long long x, y;
cin >> x >> y;
long long d = x * x - 4 * y;
if(d < 0) {
cout << 0 << ' ';
continue;
}
long long x1 = (x - sqrt(d)) / 2;
long long x2 = (x + sqrt(d)) / 2;
if(x1 + x2 != x || x1 * x2 != y) {
cout << 0 << ' ';
continue;
}
if(x1 == x2) cout << cnt[x1] * (cnt[x1] - 1) / 2 << ' ';
else cout << cnt[x1] * cnt[x2] << ' ';
}
cout << '\n';
}
}