题目:
不难想到(?)可以用二维树状数组。但维护什么?怎么查询是难点。
因为求第k小,可以考虑记权值树状数组,把比它小的和等于它的(因为可能有多个等于它的)记成1,其余记成0。
然后需要二分。
但一次询问就这么麻烦,Q次询问可能会超时吧。
所以竟然能想到整体二分!在一次二分的过程中把所有询问都二分到。
代码写得很妙!那个 修改树状数组权值的地方 和 重新排序记id的地方 都写得很好!
二分的范围很严格!一开始 l 是0;如果写成1的话全WA和RE!
而且从0到mx二分会T一个点,需要记一下mn才行!
#include#include #include #include using namespace std;const int N=505,M=6e4+5;int n,m,f[N][N],id[M],tmp[M],ans[M],cnt,mx,mn=0x7fffffff,T;bool b[M];struct Dt{ int x,y,val; bool operator<(const Dt &a)const { return val r||L>R)return; int mid=((L+R)>>1); while(a[T+1].val<=mid&&T mid)add(a[T].x,a[T].y,-1),T--;// for(int i=1;i<=cnt;i++)// if(a[i].val q[id[i]].k-1) {b[i]=1;ct++;ans[id[i]]=mid;}//去左边 else b[i]=0; int l1=l,l2=l+ct; for(int i=l;i<=r;i++) if(b[i])tmp[l1++]=id[i];else tmp[l2++]=id[i]; for(int i=l;i<=r;i++)id[i]=tmp[i];// solve(l,l1-1,L,mid-1);solve(l1,l2-1,mid+1,R);}int main(){ scanf("%d%d",&n,&m); for(int i=1;i<=n;i++) for(int j=1;j<=n;j++) { a[++cnt].x=i;a[cnt].y=j;scanf("%d",&a[cnt].val); mx=max(mx,a[cnt].val);mn=min(mn,a[cnt].val); } sort(a+1,a+cnt+1); for(int i=1;i<=m;i++) scanf("%d%d%d%d%d",&q[i].x1,&q[i].y1,&q[i].x2,&q[i].y2,&q[i].k),id[i]=i; solve(1,m,mn,mx); for(int i=1;i<=m;i++)printf("%d\n",ans[i]); return 0;}