題目：https://www.luogu.org/problemnew/show/P1527

不難想到(?)可以用二維樹狀數組。但維護什么？怎么查詢是難點。

因為求第k小，可以考慮記權值樹狀數組，把比它小的和等于它的(因為可能有多個等于它的)記成1，其余記成0。

　　然后需要二分。

但一次詢問就這么麻煩，Q次詢問可能會超時吧。

　　所以竟然能想到整體二分！在一次二分的過程中把所有詢問都二分到。

代碼寫得很妙！那個 修改樹狀數組權值的地方 和 重新排序記id的地方 都寫得很好！

二分的范圍很嚴格！一開始 l 是0；如果寫成1的話全WA和RE！

　　而且從0到mx二分會T一個點，需要記一下mn才行！

#include<iostream> #include<cstdio> #include<cstring> #include<algorithm> 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<a.val;} }a[N*N]; struct Node{int x1,y1,x2,y2,k; }q[M]; void add(int x,int y,int k) {for(;x<=n;x+=(x&-x))for(int j=y;j<=n;j+=(j&-j))f[x][j]+=k; } int qry(int x,int y) {int ret=0;for(;x;x-=(x&-x))for(int j=y;j;j-=(j&-j))ret+=f[x][j];return ret; } int query(int x) {return qry(q[x].x2,q[x].y2)-qry(q[x].x1-1,q[x].y2)-qry(q[x].x2,q[x].y1-1)+qry(q[x].x1-1,q[x].y1-1); } void solve(int l,int r,int L,int R) {if(l>r||L>R)return;int mid=((L+R)>>1);while(a[T+1].val<=mid&&T<cnt)add(a[T+1].x,a[T+1].y,1),T++;//且是<= while(a[T].val>mid)add(a[T].x,a[T].y,-1),T--; // for(int i=1;i<=cnt;i++) // if(a[i].val<mid)add(a[i].x,a[i].y,1); // else add(a[i].x,a[i].y,-1);int ct=0;for(int i=l;i<=r;i++)if(query(id[i])>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; }

?

轉載于:https://www.cnblogs.com/Narh/p/9175945.html

總結

以上是生活随笔為你收集整理的洛谷1527（bzoj2738）矩阵乘法——二维树状数组+整体二分的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。