矩阵乘法
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。
很想问出题人题目的名字是什么情况……
完全和题目内容无关的名字啊……
其实咱是来学二维树状数组的……
思路:
第一眼CDQ……
然后发现好像不太会写……
那就整体二分吧。
那么考虑把输入的每个点按值排序,整体二分答案的值。
每次二分时,维护一个二维树状数组,将值小于当前二分值的点加入树状数组。
那么对于每个询问,以树状数组的查询结果是否大于k作为判断依据即可~
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N=509;
const int M=60009;
inline int read()
{
int x=0;char ch=getchar();
while(ch<'0' || '9'<ch)ch=getchar();
while('0'<=ch && ch<='9')x=x*10+(ch^48),ch=getchar();
return x;
}
struct node
{
int val,x,y;
bool operator < (node o)const{return val<o.val;}
}g[N*N];
struct qu
{
int id,x1,x2,y1,y2,k;
}q[M],tq[M];
int n,m,tot,cur;
int bit[N][N];
int ans[M];
inline void add(int x,int y,int val)
{
for(int i=x;i<=n;i+=i&-i)
for(int j=y;j<=n;j+=j&-j)
bit[i][j]+=val;
}
inline int query(int x,int y)
{
int ret=0;
for(int i=x;i;i-=i&-i)
for(int j=y;j;j-=j&-j)
ret+=bit[i][j];
return ret;
}
inline void solve(int st,int ed,int l,int r)
{
int mid=l+r>>1;
if(st>ed)return;
if(l==r)
{
for(int i=st;i<=ed;i++)
ans[q[i].id]=mid;
return;
}
while(cur<tot && g[cur+1].val<=mid)
cur++,add(g[cur].x,g[cur].y,1);
while(cur && g[cur].val>mid)
add(g[cur].x,g[cur].y,-1),cur--;
int ltp=st-1,rtp=ed+1;
for(int i=st;i<=ed;i++)
{
int val=query(q[i].x2,q[i].y2)-query(q[i].x2,q[i].y1-1)-query(q[i].x1-1,q[i].y2)+query(q[i].x1-1,q[i].y1-1);
if(q[i].k<=val)
tq[++ltp]=q[i];
else
tq[--rtp]=q[i];
}
for(int i=st;i<=ed;i++)
q[i]=tq[i];
solve(st,ltp,l,mid);
solve(rtp,ed,mid+1,r);
}
int main()
{
n=read();
m=read();
int mval=0;
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
g[++tot]=(node){read(),i,j};
mval=max(mval,g[tot].val);
}
sort(g+1,g+tot+1);
for(int i=1;i<=m;i++)
{
q[i].x1=read();
q[i].y1=read();
q[i].x2=read();
q[i].y2=read();
q[i].k=read();
q[i].id=i;
}
solve(1,m,1,mval);
for(int i=1;i<=m;i++)
printf("%d\n",ans[i]);
return 0;
}