POJ-1840 Eqs---二分

题目链接：

https://vjudge.net/problem/POJ-1840

题目大意：

给出一个5元3次方程，输入其5个系数，求它的解的个数

其中系数 ai∈[-50,50]  自变量xi∈[-50,0）∪（0,50]

注意：xi不为0

解题思路：

五重循环肯定TLE，所以选择三重循环+两重循环，然后排序，二分找相同的数字即可

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<map>
#include<set>
#include<cmath>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long ll;
ll cnt[105];
const int maxn = 1e6 + 10;
ll sum1[maxn];
ll sum2[maxn];
int main()
{
    ll a, b, c, d, e;
    cin >> a >> b >> c >> d >> e;
    for(int i = -50; i <= 50; i++)cnt[i + 50] = i * i * i;
    int tot1 = 0, tot2 = 0;
    for(int i = -50; i <= 50; i++)
    {
        if(!i)continue;
        for(int j = -50; j <= 50; j++)
        {
            if(!j)continue;
            for(int k = -50; k <= 50; k++)
            {
                if(!k)continue;
                sum1[tot1++] = a * cnt[i + 50] + b * cnt[j + 50] + c * cnt[k + 50];
            }
        }
    }
    sort(sum1, sum1 + tot1);

    for(int i = -50; i <= 50; i++)
    {
        if(!i)continue;
        for(int j = -50; j <= 50; j++)
        {
            if(!j)continue;
            sum2[tot2++] = - d * cnt[i + 50] - e * cnt[j + 50];
        }
    }
    sort(sum2, sum2 + tot2);
    int ans = 0;
    for(int i = 0; i < tot2; i++)
    {
        ans += (upper_bound(sum1, sum1 + tot1, sum2[i]) - sum1) - (lower_bound(sum1, sum1 + tot1, sum2[i]) - sum1);
    }
    cout<<ans<<endl;
    return 0;
}