(POJ - 3685)Matrix(二分套二分)

原创于 2022-02-07 10:15:00 发布 · 360 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#二分

二分专栏收录该内容

19 篇文章

订阅专栏

该博客介绍了如何使用二分查找算法解决一个与矩阵相关的数值问题。具体而言，给定一个N阶方阵，其元素值由公式Aij=i^2+100000×i+j^2-100000×j+i×j计算得出，任务是找到第M小的值。文章通过分析公式确定了随着i的增大，Aij值递增，然后利用二分查找在每一列中寻找小于等于目标值的数的个数，以达到O(n*log(n)*log(q))的时间复杂度。文章提供了详细的C++代码实现，并演示了如何进行二分查找和暴力枚举列的过程。

中文题意：有一个N阶方阵第i行,j列的值Aij =i^2 + 100000 × i + j^2 - 100000 × j + i × j,需要找出这个方阵的第M小值.

先观察这个等式可以发现，Aij随i的增加而增加，所以我们就可以通过二分x来求取答案，每次二分的内容就是判断比x小的数的数目是否大于等于m，如果数目大于等于m，则当前二分值是大于等于答案的，反之当前二分值是小于答案的，具体二分过程如下：

我们暴力枚举列数，每一列的数是单调递增的，所以对于每一列我们可以二分枚举，这样总的复杂度就是n*log(n)*log(q)（q为答案可能值的范围大小）,细节比较多，下面附上代码

#include<cstdio>
#include<iostream>
#include<cstring>
#include<vector>
#include<algorithm>
#include<map>
#include<cmath>
#include<queue>
using namespace std;
typedef long long ll;
ll n,m;
ll cal(ll i,ll j)
{
	return i*i+100000*i+j*j-100000*j+i*j;
}
bool check(ll x)//若小于等于x的数大于等于m个则返回true，反之返回false 
{
	ll cnt=0;
	for(int i=1;i<=n;i++)//枚举列
	{
		ll l=0,r=n,mid;//二分枚举每一列中小于x的数的数目
		if(l==r) return cal(1,1)<=x;
		while(l<r)
		{
			mid=l+r+1>>1;
			if(cal(mid,i)<=x) l=mid;//找出大于等于x的最后一个数 
			else r=mid-1;
		}
		cnt+=l;
		if(cnt>=m) return true;
	}
	return false;
}
int main()
{
	int T;
	cin>>T;
	while(T--)
	{
		cin>>n>>m;
		ll l=-9999999999,r=9999999999,mid;//枚举数 
		while(l<r)
		{
			mid=l+r>>1;
			if(check(mid)) r=mid;//说明mid偏大
			else l=mid+1;
		}
		printf("%lld\n",l);
	}
	return 0;
}