傳送~: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;
}
else
l=mid+1;
}
printf("%d\n",ans);
}
}
int main()
{
cin>>n>>m>>q;
if(n==1) work1();
else work2();
return 0;
}