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;
}