傳送~：https://www.luogu.org/problem/P2468

看了一下數據也發現是兩道題，后邊當他是一個序列（n==1）的時候直接主席樹二分區間前k大和就行了

但是有一個細節我覺得就是直接不去重的離散化會避免很多問題~所以直接按照最笨的方法排兩次序

然后按照順序添加主席樹查找就行。

但是那50%的數據n！=1有點不知道怎么做了，一開始想到主席樹肯定不行，然后二位前綴和也搞不出來，然后看了題解后才知道，只需要開一個三維數組來記錄val[i][j][k]表示從1,1~i,j的這個區間有多少大于k的數就行了，主要是k他不大，可以直接全部處理出來，然后再開一個三維num[i][j][k]來記錄個數，然后可以通過二維前綴和的方式相減得到某一個矩形情況，二分即可得到答案~

代碼如下：

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<set>
#include<stack>
#include<vector>
#include<map>
#include<queue>#define myself i,l,r
#define lson i<<1
#define rson i<<1|1
#define Lson i<<1,l,mid
#define Rson i<<1|1,mid+1,r
#define half (l+r)/2
#define inff 0x3f3f3f3f
#define lowbit(x) x&(-x)
#define PI 3.14159265358979323846
#define min4(a, b, c, d) min(min(a,b),min(c,d))
#define min3(x, y, z) min(min(x,y),min(y,z))
#define pii make_pair
#define pr pair<int,int>
const int dir[4][2] = {0, -1, -1, 0, 0, 1, 1, 0};
typedef long long ll;
const ll inFF = 9223372036854775807;
typedef unsigned long long ull;
using namespace std;
const int maxn=5e5+4;
int val[205][205][1005];
int num[205][205][1005];
int b[205][205];
struct nod
{ll x;int pos,id;
}a[maxn];
bool cmp(nod s,nod e) {return s.x<e.x;}
bool cmp1(nod s,nod e) {return s.id<e.id;}
struct node
{int l,r,cnt;ll sum;
}tr[maxn*100];
int root[maxn];
int t;
int n,m,q;
int get_sum(int x1,int y1,int x2,int y2,int x)
{return val[x2][y2][x]-val[x1-1][y2][x]-val[x2][y1-1][x]+val[x1-1][y1-1][x];
}
int get_num(int x1,int y1,int x2,int y2,int x)
{return num[x2][y2][x]-num[x1-1][y2][x]-num[x2][y1-1][x]+num[x1-1][y1-1][x];
}
void work2()
{int ma=0;for(int i=1;i<=n;i++)for(int j=1;j<=m;j++)scanf("%d",&b[i][j]),ma=max(ma,b[i][j]);for(int k=0;k<=ma;k++)for(int i=1;i<=n;i++)for(int j=1;j<=m;j++){val[i][j][k]=val[i-1][j][k]+val[i][j-1][k]-val[i-1][j-1][k]+(b[i][j]>=k?b[i][j]:0);num[i][j][k]=num[i-1][j][k]+num[i][j-1][k]-num[i-1][j-1][k]+(b[i][j]>=k?1:0);}int x1,y1,x2,y2,h;while(q--){scanf("%d %d %d %d %d",&x1,&y1,&x2,&y2,&h);if(get_sum(x1,y1,x2,y2,0)<h) {puts("Poor QLW");continue;}int l=0,r=ma,ans=-1;while(l<=r){int mid=half;if(get_sum(x1,y1,x2,y2,mid)>=h){ans=mid;l=mid+1;}else r=mid-1;}if(ans==-1) {puts("Poor QLW");continue;}else printf("%d\n",get_num(x1,y1,x2,y2,ans)-(get_sum(x1,y1,x2,y2,ans)-h)/ans);//這里(get_sum(x1,y1,x2,y2,ans)-h)/ans是可能有很多大小=ans的數，這個時候可能多算了哈//所以說算出來sum減掉h就是多出來的，然后看是不是多了幾個大小為ans的，把他們刪掉}
}
void insert(int &now,int pre,int l,int r,int k,ll x)
{now=++t;tr[now]=tr[pre];tr[now].cnt++;tr[now].sum+=x;if(l==r) return;int mid=half;if(k<=mid) insert(tr[now].l,tr[pre].l,l,mid,k,x);else insert(tr[now].r,tr[pre].r,mid+1,r,k,x);
}
ll query(int st,int ed,int l,int r,int k)//自己寫的主席樹好丑0A0
{if(k==0) return 0;if(l==r) return tr[ed].sum-tr[st].sum;int x=tr[tr[ed].r].cnt-tr[tr[st].r].cnt;ll ans=0;int mid=half;if(x<=k){ans+=tr[tr[ed].r].sum-tr[tr[st].r].sum;ans+=query(tr[st].l,tr[ed].l,l,mid,k-x);}else ans=query(tr[st].r,tr[ed].r,mid+1,r,k);return ans;
}
void work1()
{n=m;for(int i=1;i<=n;i++)scanf("%lld",&a[i].x),a[i].id=i;sort(a+1,a+1+n,cmp);for(int i=1;i<=n;i++) a[i].pos=i;sort(a+1,a+1+n,cmp1);for(int i=1;i<=n;i++)insert(root[i],root[i-1],1,n,a[i].pos,a[i].x);int x1,y1,x2,y2;ll h;while(q--){scanf("%d %d %d %d %lld",&x1,&y1,&x2,&y2,&h);if(query(root[y1-1],root[y2],1,n,y2-y1+1)<h) {puts("Poor QLW");continue;}int l=0,r=y2-y1+1,ans=-1;while(l<=r){int mid=half;if(query(root[y1-1],root[y2],1,n,mid)>=h){ans=mid;r=mid-1;}elsel=mid+1;}printf("%d\n",ans);}
}
int main()
{cin>>n>>m>>q;if(n==1) work1();else work2();return 0;
}

?

總結

以上是生活随笔為你收集整理的洛谷 P2468 粟粟的书架 二分（主席树+前缀和）的全部內容，希望文章能夠幫你解決所遇到的問題。

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