题目描述:2041. 干草堆 - AcWing题库
贝茜对她最近在农场周围造成的一切恶作剧感到抱歉,她同意帮助农夫约翰把一批新到的干草捆堆起来。
开始时,共有 N 个空干草堆,编号 1∼N。
约翰给贝茜下达了 K 个指令,每条指令的格式为 A B
,这意味着贝茜要在 A..B 范围内的每个干草堆的顶部添加一个新的干草捆。
例如,如果贝茜收到指令 10 13
,则她应在干草堆 10,11,12,13 中各添加一个干草捆。
在贝茜完成了所有指令后,约翰想知道 N 个干草堆的中值高度——也就是说,如果干草堆按照高度从小到大排列,位于中间的干草堆的高度。
方便起见,N 一定是奇数,所以中间堆是唯一的。
请帮助贝茜确定约翰问题的答案。
输入格式
第一行包含 N 和 K。
接下来 K 行,每行包含两个整数 A,B,用来描述一个指令。
输出格式
输出完成所有指令后,NN 个干草堆的中值高度。
数据范围
1≤N≤106,
1≤K≤25000,
1≤A≤B≤N
输入样例:
7 4
5 5
2 4
4 6
3 5
输出样例:
1
样例解释
贝茜完成所有指令后,各堆高度为 0,1,2,3,3,1,0。
将各高度从小到大排序后,得到 0,0,1,1,2,3,3,位于中间的是 1。
解题思路:前缀和and差分
看题目会觉得很简单,因为只需让在这个区间的每一个草堆高度元素加一即可,可是,如果直接用for循环,外面还好有一个while来控制指令个数,时间复杂度O(NK)还是会超时,这时候,就要考虑时间复杂度为O(N)的解法了,那怎么让每一个草堆高度加一的复杂度变成O(1)呢?前缀和and差分闪亮登场!!
对于不了解的小伙伴,先将一下前缀和以及差分,了解就可以直接A题了
前缀和:
预处理求出s[1~n],然后就能快速求出任意数组任意区间里一段数的和,如求l~r区间的和,就是s[r]-s[l-1],数组要从1开始从,s[0]=0,这样比如算s[1]-s[9],就可以用s[9]-s[0],而s[0]其实就是0,没有元素,避免越界
S[i] = a[1] + a[2] + ... a[i]
a[l] + ... + a[r] = S[r] - S[l - 1]
差分:
差分,就是前缀和的逆运算,对一个数组A[a1,a2,...an],构造一个数组B[b1,b2,...bn]使得ai=b1+b2+...+bi;那么A数组就是B数组的前缀和,B数组就是A数组的差分。构造差分的时候,其实就是b1=a2-a1,b3=a3-a2...bn=an-an-1
假如给b[l]+c那么由B数组来求A数组的时候,从a[l]~a[n]都要加上b[l],则每一个元素都加上了c,但如果只让a[l]~a[r]加上c,则再让从r+1之后的都减去c,同理,a[l+1]~a[n]每一个元素都要加上b[l+1],那么让b[l+1]-c即可
给区间[l, r]中的每个数加上c:B[l] += c, B[r + 1] -= c
最后怎么求数组A在经过几次区间加c之后的最终序列呢?当然是用b来求,我们知道A是B的前缀和,可以这样实现
for(int i=1;i<=n;i++) b[i]+=b[i-1];//每一项都变成这一项和前i-1项的和
for(int i=1;i<=n;i++) printf("%d ",b[i]);
//最终得到的B数组就是前缀和A
最终代码实现:
#include<iostream>
#include<algorithm>
using namespace std;
const int N=1e6+10;
int heap[N];
int main()
{
int n,t,a,b;
cin>>n>>t;
while(t--){
cin>>a>>b;
heap[a]++;
heap[b+1]--;
}
for(int i=1;i<n;i++) heap[i]+=heap[i-1];
sort(heap,heap+n);
cout<<heap[n/2]<<endl;
return 0;
}