POJ3685 Matrix (二分-查找第K大的值)

题目链接：https://vjudge.net/problem/POJ-3685

题目大意：求解第m小的数字

题目思路：二分之查找第K大的值，用多个二分，二分的判断条件中加一个二分。前提是保证所给的序列有序。

总结：有毒。思路一样，却总得不出正确的结果。

代码：

#include <iostream>    
#include <algorithm>    
#include <cmath>    
#include <vector>    
#include <string>   
#include <cstdio> 
#include <cstring>    
#pragma warning(disable:4996)    
using namespace std;  
  
#define maxn 1e12  
typedef long long ll;  
  
ll m, n;  
ll le, ri, mid;  
  
ll cal(ll i, ll j)  
{  
    return i*i + 100000 * i + j*j - 100000 * j + i*j;  
}  
  
ll check(ll x)  
{  
    ll i, le, ri, mid, cnt;  
      
    ll temp;  
    cnt = 0;  
    for (i = 1; i <= n; i++)  
    {  
        le = 1;//mid不能取到0，所以这里的le取1  
        ri = n + 1;//因为mid可能要取到n，所以这里的ri要取到比n大的数  
          
        mid = le + (ri - le) / 2;  
        while (le < ri)  
        {  
            temp = cal(mid, i);  
            if (cal(mid, i) < x)  
            {  
                le = mid + 1;  
            }  
            else  
            {  
                ri = mid ;  
            }  
            mid = le + (ri - le) / 2;  
        }  
        cnt += (mid - 1);//经过计算，这里始终是多计算了一个1，所以要在这里把它扣掉  
    }  
    return cnt;  
}  
  
int main()  
{  
    //freopen("i.txt", "r", stdin);  
    //freopen("o.txt", "w", stdout);  
  
    int test;  
    scanf("%d", &test);  
  
    while (test--)  
    {  
        scanf("%lld%lld", &n, &m);  
        le = -maxn;  
        ri = maxn;  
  
        mid = le + (ri - le) / 2;  
        while (le < ri)  
        {  
            if (check(mid) < m)//检查小于 mid 的个数  
            {  
                le = mid + 1;  
            }  
            else  
            {  
                ri = mid;  
            }  
            mid = le + (ri - le) / 2;  
        }  
        printf("%lld\n", mid-1);//有m个小于mid的数，所以第m大的数就是mid-1  
    }  
  
    //system("pause");  
    return 0;  
}