1057. Stack (30) - 树状数组
来源:互联网 发布:手机淘宝4.0 编辑:程序博客网 时间:2024/06/02 19:50
题目如下:
Stack is one of the most fundamental data structures, which is based on the principle of Last In First Out (LIFO). The basic operations include Push (inserting an element onto the top position) and Pop (deleting the top element). Now you are supposed to implement a stack with an extra operation: PeekMedian -- return the median value of all the elements in the stack. With N elements, the median value is defined to be the (N/2)-th smallest element if N is even, or ((N+1)/2)-th if N is odd.
Input Specification:
Each input file contains one test case. For each case, the first line contains a positive integer N (<= 105). Then N lines follow, each contains a command in one of the following 3 formats:
Push keyPop
PeekMedian
where key is a positive integer no more than 105.
Output Specification:
For each Push command, insert key into the stack and output nothing. For each Pop or PeekMedian command, print in a line the corresponding returned value. If the command is invalid, print "Invalid" instead.
Sample Input:17PopPeekMedianPush 3PeekMedianPush 2PeekMedianPush 1PeekMedianPopPopPush 5Push 4PeekMedianPopPopPopPopSample Output:
InvalidInvalid322124453Invalid
这个题目我最初用的是string、stringstream和vector来做,发现会严重超时,后来在网上参考了和山米兰的解法,发现他的方法很有技巧,分析如下:
①对命令的解析,只看第二位,如果是o,说明是Pop,如果是e,说明是PeekMedian,否则是push,是push则应当再读入一次数字,这比用getline要好的多,因为getline还需要排除第一个输入的N。
②求中位数的思想,不是排序找中间的值,而是通过统计从1开始的每个元素的个数放到数组C中,这样从前到后,数组C的子列和为题目要求的位置时,拿到的就是中位数。
③求子列和的思想,因为是从前到后的前缀和,可以利用树状数组,下面的代码利用add实现了添加和删除两种操作,利用value的不同,1表示添加,2表示删除。树状数组的基本思想就是数组C中不同元素管辖不同的区域,如果要添加一个元素,则所有满足区域条件的位置都要+value,反之如果删除,所有满足条件的区域都要-value。本题要求的是统计1~100000的元素个数,因此value=+1或者-1。
④求子列和为题目要求的值,利用二分查找。
#include<stdio.h>#include<cstring>#include<iostream>#include<string>using namespace std;const int N=100001;int c[N];int lowbit(int i){return i&(-i);}void add(int pos,int value){while(pos<N){c[pos]+=value;pos+=lowbit(pos);}}int sum(int pos){int res=0;while(pos>0){res+=c[pos];pos-=lowbit(pos);}return res;}int find(int value){int l=0,r=N-1,median,res;while(l<r-1){if((l+r)%2==0)median=(l+r)/2;elsemedian=(l+r-1)/2;res=sum(median);if(res<value)l = median;elser = median;}return l+1;}int main(){char ss[20];int stack[N],top=0,n,pos;memset(c,0,sizeof(c));scanf("%d",&n);while(n--){scanf("%s",ss);if(ss[1]=='u'){scanf("%d",&pos);stack[++top]=pos;add(pos,1);}else if(ss[1]=='o'){if(top==0){printf("Invalid\n");continue;}int out=stack[top];add(out,-1); // 删除元素outprintf("%d\n",stack[top--]);}else if(ss[1]=='e'){if(top==0){printf("Invalid\n");continue;}int res;if(top%2==0)res=find(top/2);elseres=find((top+1)/2);printf("%d\n",res);}else{printf("Invalid\n");}}return 0;}
- 1057. Stack (30) - 树状数组
- PAT 1057. Stack (30) 中位数;树状数组
- 1057. Stack (30)(树状数组)
- 1057. Stack (30)-PAT甲级真题(树状数组)
- PAT 1057 Stack 树状数组
- 1057. Stack (30)【栈+树状数组】——PAT (Advanced Level) Practise
- PAT 1057 Stack (30) (树状数组+二分 典型题)
- pat-1057 Stack 树状数组+二分查找
- PAT-1057 Stack (树状数组 + 二分查找)
- 树状数组
- 树状数组
- 树状数组
- 树状数组
- 树状数组
- 树状数组
- 树状数组
- 树状数组
- 树状数组
- UVa 11078 - Open Credit System(维护最大值)
- 有序数组和二分法
- OC_交叉引入头文件问题
- hadoop Hdfs文件上传下载
- unbutu下安装与配置phpmyadmin
- 1057. Stack (30) - 树状数组
- HDU 1016 Prime Ring Problem
- 旋转数组
- 面试题10_二进制中1的个数
- RobotFramework+Selenium2Library环境安装
- 存储对象的数组
- Hadoop hdfs增删该查 简单写法
- C语言(3)--Fibonacci数列、候选人得票统计、字符串复制函数
- BZOJ 1068 [SCOI2007]压缩 区间DP