F. Sum and Product - 思维

最新推荐文章于 2024-09-14 22:25:23 发布

.y.a.o.

最新推荐文章于 2024-09-14 22:25:23 发布

阅读量371

点赞数 1

文章标签：算法 c++ 思维二分

本文链接：https://blog.csdn.net/m0_75129175/article/details/132219547

版权

分析：

题目中的格式有点像韦达定理，就是对于一元二次方程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';
    }
}