bzoj 2738: 矩阵乘法 （整体二分+二维树状数组）

2738: 矩阵乘法

Time Limit: 20 Sec   Memory Limit: 256 MB
Submit: 1334   Solved: 579
[ Submit][ Status][ Discuss]

Description

　　给你一个N*N的矩阵，不用算矩阵乘法，但是每次询问一个子矩形的第K小数。

Input

 

　　第一行两个数N,Q，表示矩阵大小和询问组数；
　　接下来N行N列一共N*N个数，表示这个矩阵；
　　再接下来Q行每行5个数描述一个询问：x1,y1,x2,y2,k表示找到以(x1,y1)为左上角、以(x2,y2)为右下角的子矩形中的第K小数。

Output

　　对于每组询问输出第K小的数。

Sample Input

2 2
2 1
3 4
1 2 1 2 1
1 1 2 2 3

Sample Output

1
3

HINT

　　矩阵中数字是109以内的非负整数；

　　20%的数据：N<=100,Q<=1000；

　　40%的数据：N<=300,Q<=10000；

　　60%的数据：N<=400,Q<=30000；

　　100%的数据：N<=500,Q<=60000。

Source

[ Submit][ Status][ Discuss]

题解：整体二分+二维树状数组

将矩阵中的数按照权值排序，然后二分，将[l,mid]中的数加入到树状数组的对应位置，然后判断当前区间的询问，对于一个询问如果区域内的点的个数<=k，就把询问扔到[l,mid]，否则扔到[mid+1,r]，然后继续做。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define N 603
#define M 300000
using namespace std;
struct data
{
	int x,y,num;
}a[M];
struct node
{
	int x,y,x0,y0,k,num,opt;
}b[M];
int n,m,f[N][N],ans[M];
int cmp(data a,data b)
{
	return a.num<b.num;
}
int lowbit(int x)
{
	return x&(-x);
}
void change(int x,int y,int v)
{
	for (int i=x;i<=n;i+=lowbit(i))
	 for (int j=y;j<=n;j+=lowbit(j))
	  f[i][j]+=v;
}
int query(int x,int y)
{
	int ans=0;
	for (int i=x;i;i-=lowbit(i))
	 for (int j=y;j;j-=lowbit(j))
	  ans+=f[i][j];
	return ans;
}
int cmp1(node a,node b)
{
	return a.opt<b.opt;
}
void divide(int l,int r,int x,int y)
{
	if (l==r) {
		for (int i=x;i<=y;i++)
		 ans[b[i].num]=a[l].num;
		return;
	}
	int mid=(l+r)/2;
	for (int i=l;i<=mid;i++) change(a[i].x,a[i].y,1);
	int x1=0; int  y1=y-x+1;
	for (int i=x;i<=y;i++){
		int t=query(b[i].x,b[i].y);
		int t1=query(b[i].x0-1,b[i].y0-1);
		int t2=query(b[i].x0-1,b[i].y);
		int t3=query(b[i].x,b[i].y0-1);
		int sum=t-t2-t3+t1;
		if (sum>=b[i].k) b[i].opt=++x1;
		else b[i].opt=++y1,b[i].k-=sum;
	}
	for (int i=l;i<=mid;i++) change(a[i].x,a[i].y,-1);
	sort(b+x,b+y+1,cmp1);
	divide(l,mid,x,x+x1-1);
	divide(mid+1,r,x+x1,y);
}
int main()
{
	freopen("a.in","r",stdin);
	freopen("my.out","w",stdout);
	scanf("%d%d",&n,&m);
	int cnt=0;
	for (int i=1;i<=n;i++) 
	 for (int j=1;j<=n;j++){
		int x; scanf("%d",&x);
		a[++cnt].num=x; a[cnt].x=i; a[cnt].y=j;
	}
    sort(a+1,a+cnt+1,cmp);
   // for (int i=1;i<=cnt;i++)
     //cout<<a[i].x<<" "<<a[i].y<<" "<<a[i].num<<endl;
    for (int i=1;i<=m;i++) 
    	scanf("%d%d%d%d%d",&b[i].x0,&b[i].y0,&b[i].x,&b[i].y,&b[i].k),b[i].num=i;
    divide(1,cnt,1,m);
    for (int i=1;i<=m;i++) printf("%d\n",ans[i]);
}