P2468 [SDOI2010]粟粟的书架----主席树部分

P2468 [SDOI2010]粟粟的书架

题目描述

幸福幼儿园B29班的粟粟是一个聪明机灵、乖巧可爱的小朋友，她的爱好是画画和读书，尤其喜欢Thomas H. Cormen的文章。粟粟家中有一个R行C列的巨型书架，书架的每一个位置都摆有一本书，上数第i行、左数第j列摆放的书有Pi,j页厚。

粟粟每天除了读书之外，还有一件必不可少的工作就是摘苹果，她每天必须摘取一个指定的苹果。粟粟家果树上的苹果有的高、有的低，但无论如何凭粟粟自己的个头都难以摘到。不过她发现，如果在脚下放上几本书，就可以够着苹果；她同时注意到，对于第i天指定的那个苹果，只要她脚下放置书的总页数之和不低于Hi，就一定能够摘到。

由于书架内的书过多，父母担心粟粟一天内就把所有书看完而耽误了上幼儿园，于是每天只允许粟粟在一个特定区域内拿书。这个区域是一个矩形，第i天给定区域的左上角是上数第x1i行的左数第y1i本书，右下角是上数第x2i行的左数第y2i本书。换句话说，粟粟在这一天，只能在这﹙x2i－x1i＋1﹚×﹙y2i－y1i＋1﹚本书中挑选若干本垫在脚下，摘取苹果。

粟粟每次取书时都能及时放回原位，并且她的书架不会再撤下书目或换上新书，摘苹果的任务会一直持续M天。给出每本书籍的页数和每天的区域限制及采摘要求，请你告诉粟粟，她每天至少拿取多少本书，就可以摘到当天指定的苹果。

输入格式

输入文件susu.in第一行是三个正整数R, C, M。

接下来是一个R行C列的矩阵，从上到下、从左向右依次给出了每本书的页数Pi,j。

接下来M行，第i行给出正整数x1i, y1i, x2i, y2i, Hi，表示第i天的指定区域是﹙x1i, y1i﹚与﹙x2i, y2i﹚间的矩形，总页数之和要求不低于Hi。

保证1≤x1i≤x2i≤R，1≤y1i≤y2i≤C。

输出格式

输出文件susu.out有M行，第i行回答粟粟在第i天时为摘到苹果至少需要拿取多少本书。如果即使取走所有书都无法摘到苹果，则在该行输出“Poor QLW”（不含引号）。

输入输出样例

输入 #1复制

5 5 7
14 15 9 26 53
58 9 7 9 32
38 46 26 43 38
32 7 9 50 28
8 41 9 7 17
1 2 5 3 139
3 1 5 5 399
3 3 4 5 91
4 1 4 1 33
1 3 5 4 185
3 3 4 3 23
3 1 3 3 108

输出 #1复制

6
15
2
Poor QLW
9
1
3

输入 #2

1 10 7
14 15 9 26 53 58 9 7 9 32
1 2 1 9 170
1 2 1 9 171
1 5 1 7 115
1 1 1 10 228
1 4 1 4 45704571
1 1 1 1 1
1 7 1 8 16

输出 #2

6
7
3
10
Poor QLW
1
2

说明/提示

【数据规模和约定】

对于10%的数据，满足R, C≤10；

对于20%的数据，满足R, C≤40；

对于50%的数据，满足R, C≤200，M≤200,000；

另有50%的数据，满足R＝1，C≤500,000，M≤20,000；

对于100%的数据，满足1≤Pi,j≤1,000，1≤Hi≤2,000,000,000。

 

一道传说中的组合题，但我只是想复习一下主席树的板子，所以只写了主席树部分。

题意：

当R=1时，给出一个区间L-R，给出条件K，求出在L-R区间内，选取最少的x个数使得它们加起来大于等于K

输出最少的数目x，如果L-R里所有的数字加起来小于K输出“Poor QLW”。

分析：

构建一颗主席树，存放节点个数和累加起来的值a[i]（见代码），然后先求出右边的数字总和w，如果w>=k说明右边的数字就可以满足条件，所以直接走右边（因为右边的数字一定大！！依照题目意思肯定选大的一些数），如果w<k说明就算把右边全加了也不能满足条件，所以我们直接把右边有多少个数字全部加了，然后找左边，这时只需要把右边的总数减掉就可以了。

见代码：（50分代码，只有主席树部分）

#include<bits/stdc++.h>
using namespace std;
const int maxn=5e5+6;
int a[maxn];
int root[maxn]; 
int cnt;
struct node{
	int l,r,sum,v;
}sz[maxn*40];
void updata(int l,int r,int &x,int y,int p)
{
	sz[++cnt]=sz[y];
	sz[cnt].v++;
	sz[cnt].sum+=p;
	x=cnt;
	if(l==r) return ;
	int mid=l+r>>1;
	if(mid>=p)
		updata(l,mid,sz[x].l,sz[y].l,p);
	else 
		updata(mid+1,r,sz[x].r,sz[y].r,p);
}
int getSum(int l,int r,int x,int y,int k)
{
	if(l==r) return (l+k-1)/l;
	int w=sz[sz[y].r].sum-sz[sz[x].r].sum;//求出右边总和 
	int mid=l+r>>1;
	int ans=0;
	if(w>=k)//如果右边总和大于k,说明右边就可以满足条件，直接找右边就可以了 
		ans+= getSum(mid+1,r,sz[x].r,sz[y].r,k);
	else//如果右边还少了，就把右边加起来，然后找左边 
		ans+= (sz[sz[y].r].v-sz[sz[x].r].v)+getSum(l,mid,sz[x].l,sz[y].l,k-w);
	return ans;
}
int main()
{
	int m,n,q;
	scanf("%d %d %d",&m,&n,&q);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
	}
	for(int i=1;i<=n;i++)
	{
		updata(1,1e9,root[i],root[i-1],a[i]);//更新 
	}
	for(int i=1;i<=q;i++)
	{
		int aaa,x,bbb,y,k;
		scanf("%d %d %d %d %d",&aaa,&x,&bbb,&y,&k);
		if(sz[root[y]].sum-sz[root[x-1]].sum<k)//如果所有的都不满足，直接输出 Poor QLW
			printf("Poor QLW\n");
		else
			printf("%d\n",getSum(1,1e9,root[x-1],root[y],k));
	}
}