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;
}