码迷,mamicode.com
首页 > 其他好文 > 详细

POJ - 3685 Matrix 二分

时间:2015-05-14 01:03:26      阅读:192      评论:0      收藏:0      [点我收藏+]

标签:

题目大意:有一个N * N的矩阵,其中Aij = i * i + i * 100000 - 100000 * j + j * j + i * j,问这个矩阵中,第M小的数是多少

解题思路:观察这个式子,可以发现j不变的情况下,随着i的增大,Aij也相应增大,由这个受到启发
二分枚举第M小的数,然后按列寻找,找到第一个大于这个数的位置,就可以知道该列中有多少个数是大于这个数的了

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define INF 0x3f3f3f3f
typedef long long ll;
ll N, M;

ll Count(int i, int j) {
    return ll(i + 100000 + j) * i + ll(j - 100000) * j;
}

bool judge(ll mid) {
    ll cnt = 0;
    for(int j = 1; j <= N; j++) {
        int l = 1, r = N;
        while(l <= r) {
            int Mid = (l + r) / 2;
            if( Count(Mid,j) <= mid)
                l = Mid + 1;
            else
                r = Mid - 1;
        }
        cnt += l - 1;
    }
    return cnt < M;
}

ll solve() {
    ll l = -((ll)N * N * 3 + 100000 * N) , r = (ll)N * N * 3 + 100000 * N;

    while(l <= r) {
        ll mid = (l + r) / 2; 
        if(judge(mid))
            l = mid + 1;
        else
            r = mid - 1;
    }
    return l;
}

int main(){
    int test;
    scanf("%d", &test);
    while(test--) {
        scanf("%lld%lld", &N, &M);
        printf("%lld\n",solve());
    }
    return 0;
}

POJ - 3685 Matrix 二分

标签:

原文地址:http://blog.csdn.net/l123012013048/article/details/45704235

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!