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[];
 const int maxn = 1e6 + ;
 ll sum1[maxn];
 ll sum2[maxn];
 int main()
 {
     ll a, b, c, d, e;
     cin >> a >> b >> c >> d >> e;
     for(int i = -; i <= ; i++)cnt[i + ] = i * i * i;
     int tot1 = , tot2 = ;
     for(int i = -; i <= ; i++)
     {
         if(!i)continue;
         for(int j = -; j <= ; j++)
         {
             if(!j)continue;
             for(int k = -; k <= ; k++)
             {
                 if(!k)continue;
                 sum1[tot1++] = a * cnt[i + ] + b * cnt[j + ] + c * cnt[k + ];
             }
         }
     }
     sort(sum1, sum1 + tot1);

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