Matrix （POJ - 3685，二分答案 + 二分）

一.题目链接：

POJ-3685

二.题目大意：

给 n 和 m.

表示有一个 n × n 的矩阵，求第 m 小的元素的值.

矩阵元素的计算公式：

三.分析：

易得矩阵的元素在同一列上单增.

第一层二分答案.

因为是求第 m 小的数，所以必有 m 个数 ≤ 答案（数有重复）.

check(mid) 函数里先枚举列，对于每一列再二分有多少个数 ≤ mid即可.

ps: 控制边界好累额

四.代码实现：

#include <set>
#include <map>
#include <ctime>
#include <queue>
#include <cmath>
#include <stack>
#include <vector>
#include <cstdio>
#include <sstream>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#define eps 1e-6
#define pi acos(-1.0)
#define ll long long int
using namespace std;
const int M = (int)1e5;
const ll mod = (ll)1e9 + 7;
const ll inf = 0x3f3f3f3f3f;

ll n, m;

ll f(ll i, ll j)
{
    return i * i + j * j + M * (i - j) + i * j;
}

bool check(ll x)
{
    ll cnt = 0;
    for(int i = 1; i <= n; ++i)
    {
        int l = 0;
        int r = n;
        while(l < r)
        {
            int mid = (l + r + 1) >> 1;
            if(f(mid, i) > x)
                r = mid - 1;
            else
                l = mid;
        }
        cnt += l;
    }
    return cnt >= m;
}

int main()
{
    int T;
    scanf("%d", &T);
    while(T--)
    {
        scanf("%lld %lld", &n, &m);
        ll l = -inf;
        ll r = inf;
        while(l < r)
        {
            ll mid = (l + r) >> 1;
            if(check(mid))
                r = mid;
            else
                l = mid + 1;
        }
        printf("%lld\n", r);
    }
    return 0;
}