洛谷 P2468 粟粟的书架 二分（主席树+前缀和）

传送~：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;
}