PAT.A1057 Stack

返回目录

题意

给出一个栈的入栈(Push)、出栈(Pop)过程，并随时通过PeekMedian命令要求输出栈中中位数(Pop命令输出出栈的数)。当栈中没有元素时，Pop命令和PeekMedian命令都应该输出“Invalid"。

样例（可复制）

17
Pop
PeekMedian
Push 3
PeekMedian
Push 2
PeekMedian
Push 1
PeekMedian
Pop
Pop
Push 5
Push 4
PeekMedian
Pop
Pop
Pop
Pop

//output
Invalid
Invalid
3
2
2
1
2
4
4
5
3
Invalid

注意点

1. 本题如果对每次查询都遍历一次table则会超时，必须使用分块或其他方法，这里提供分块思想和树状数组的方法。
2. 分块的大小可以根据数据自行修改，在不知道测试数据的情况下，可以考虑分成sqrt（maxn)块。

分块思想

#include <bits/stdc++.h>
using namespace std;

int w[10010],dp[110]={0};
bool cho[10011][110],flag[10011];//cho[i][v]用来记录dp[i][v]时选择的策略，flag[i]用来存放第i号物品是否选择
bool cmp(int a,int b){
	return a>b;
}
int main(){
    int n,m;
    cin>>n>>m;
    for(int i=1;i<=n;i++)scanf("%d",&w[i]);
    sort(w+1,w+n+1,cmp);
    for(int i=1;i<=n;i++){
    	for(int v=m;v>=w[i];v--){
    		if(dp[v]<=dp[v-w[i]]+w[i]){//放入
    			dp[v]=dp[v-w[i]]+w[i];
    			cho[i][v]=1;
			}else{//不放入
				cho[i][v]=0;
			}
		}
	}
	if(dp[m]!=m)cout<<"No Solution";
	else{
		int k=n,num=0;//num记录选择的总数
		while(k>=0){
			if(cho[k][v]==1){
				flag[k]=true;
				m-=w[k];
				num++;
			}else{
				flag[k]=false;
			}
			k--;
		}
		for(int i=n;i>=1;i--){
			if(flag[i]){
				printf("%d",w[i]);
				num--;
				if(num>0)printf(" ");
			}
		}
	}
	return 0;
}

3. 这里说明下#define lowbit(i) ((i)&-(i))语句，该语句定义了一个函数，函数的作用是返回一个不大于i的2的最大次方因子。实现的方式为将该数i与其-i的补码进行与操作，例如lowbit(1)，则会用0000 0001和1111 1111进行与操作，返回1，再例如lowbit(6),则会用0000 0110和1111 1010相与，返回2，可以注意到2^2<6, 2^3>6。该语句还可以使用参考代码中注释的部分代替。

树状数组

#include<bits/stdc++.h>
using namespace std;

#define lowbit(i) ((i)&-(i))
const int N=100010;
stack<int> s;
int c[N];//树状数组 
// int lowbit(int x){
//     return x&(-x);
// }
void update(int x,int v){
	for(int i=x;i<N;i+=lowbit(i))c[i]+=v;
}
int getsum(int x){
	int sum=0;
	for(int i=x;i>0;i-=lowbit(i))sum+=c[i];
	return sum;
}
void peek(){//二分法求第k大
	int l=1,r=N,mid,k=(s.size()+1)/2;
	while(l<r){
		mid=(l+r)/2;
		if(getsum(mid)>=k)r=mid;
		else l=mid+1;
	}
	printf("%d\n",l);
}
int main(){
	int n,x;
	char cmd[20];
	cin>>n;
	for(int i=0;i<n;i++){
		scanf("%s",cmd);
		if(strcmp(cmd,"Push")==0){
			scanf("%d",&x);
			s.push(x);
			update(x,1);
		}else if(strcmp(cmd,"Pop")==0){
			if(s.empty())printf("Invalid\n");
			else{
				printf("%d\n",s.top());
				update(s.top(),-1);
				s.pop();
			}
		}else{
			if(s.empty())printf("Invalid\n");
			else peek();
		}
	}
    return 0;
}