题目传送门: http://acm.hdu.edu.cn/showproblem.php?pid=5818
题意:有两个栈,有3种操作。 第一种是往其中一个栈加入一个数; 第二种是取出其中一个栈的顶端数字; 第三种是将其中一个栈的所有元素放入另外一个栈,元素顺序依旧按照加入顺序来放。
然后官方题解是这样的: 比较简单巧妙的一个做法是引入一个新的栈C,每次合并的时候就把A和B合并到C上,然后把A和B都清空. push还是按正常做,pop注意当遇到要pop的栈为空时,因为题目保证不会对空栈进行pop操作,所以这时应直接改为对C栈进行pop操作. 这样做因为保证每个元素最多只在一次合并中被处理到,pop和push操作当然也是每个元素只做一次,所以总复杂度是O(N)的. 另一种做法是用链表来直接模拟,复杂度也是O(N),但代码量稍大一些.
其实这题用三个优先队列做就好了。。
七夕打比赛果然是有毒的QAQ。
详情见代码:
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <queue>
using namespace std;
typedef pair <int , int > pi;
priority_queue<pi>a;
priority_queue<pi>b;
priority_queue<pi>c;
int cnt;
void init()
{
while(!a.empty())
{
a.pop();
}
while(!b.empty())
{
b.pop();
}
while(!c.empty())
{
c.pop();
}
cnt=0;
}
void workpush(char aim)
{
if(aim=='A')
{
int n;
scanf("%d",&n);
a.push(make_pair(cnt++,n));
}
else
{
int n;
scanf("%d",&n);
b.push(make_pair(cnt++,n));
}
}
void workpop(char aim)//这里,如果没有merge过,就可以直接从原栈中取值。
//如果merge过了,就从混合后的队列中取值。
{
if(aim=='A')
{
if(!a.empty())
{
pi x=a.top();
a.pop();
printf("%d\n",x.second);
}
else
{
pi x=c.top();
c.pop();
printf("%d\n",x.second);
}
return ;
}
if(aim=='B')
{
if(!b.empty())
{
pi x=b.top();
b.pop();
printf("%d\n",x.second);
}
else
{
pi x=c.top();
c.pop();
printf("%d\n",x.second);
}
return ;
}
return ;
}
void workmer(char aim)//不管是谁往谁里面加,都全部放入一个队列中。
{
char ch;
scanf(" %c",&ch);
while(!a.empty())
{
pi x=a.top();
c.push(x);
a.pop();
}
while(!b.empty())
{
pi x=b.top();
c.push(x);
b.pop();
}
}
int main (void)
{
int num;
int cas=1;
while(~scanf("%d",&num))
{
if(num==0)
break;
init();
char ch[10];
char aim;
printf("Case #%d:\n",cas++);
while(num--)
{
scanf("%s %c",ch,&aim);
if(ch[1]=='u')
{
workpush(aim);
}
if(ch[1]=='o')
{
workpop(aim);
}
if(ch[0]=='m')
{
workmer(aim);
}
}
}
return 0;
}