题意
栈的简单应用,给出元素个数,入栈顺序O1和出栈顺序O2,每个序列里有n个元素,问能不能利用栈把O1变成O2。
版本1:
依此考虑出栈的元素(记为temp),首先判断为了temp出栈,是否需要有元素入栈,一种是需要,则入栈,入栈之后的栈顶元素则为temp;若不需要入栈,那么因为栈的特点,在正常情况下,栈顶元素为temp,若不是,则O1不能转换成O2,此时则退出循环。而对于栈顶元素为temp的情况,则出栈,继续考虑下一个出栈元素。
#include<iostream>
#include<stack>
using namespace std;
#define N 10
// 入栈和出栈序列
char in[N];
char out[N];
int main(){
int x;
while(~scanf("%d%s%s", &x, in, out)){
int res[N*2]; // 用于记录输入、输出的过程,入栈为1,出栈为0
int point = 0; // 指向res
int flag = 1; // 能否转换的标记
stack <char> stk;
// 上次入栈的元素的后面元素的下标
int start = 0;
// 判断每个输出的元素能否正确输出
for (int i = 0; i < x; ++i) {
char temp;
temp = out[i];
// 确定当前输出元素,在输入时的下标
int j = 0;
for (; j < x; ++j) {
// 因为元素是唯一的,所以可以通过判断元素相等来确定位置
if(in[j] == temp)
break;
}
// 确定本轮从哪里开始入栈
int k = 0;
if(j<start)//不需要入栈
{
// 是栈顶,出栈
if(stk.top() == temp){
stk.pop();
res[point ++] = 1;
}
// 出错
else{
flag = 0;
break;
}
}
// 需要入栈
else{
// 从上次入栈的位置之后入栈
k = start;
for (; k <= j; ++k) {
stk.push(in[k]);
res[point ++] = 0;
}
// 栈顶元素直接出栈
stk.pop();
res[point ++] = 1;
start = ++j;
}
}
// 输出结果
if(flag != 0){
printf("Yes.\n");
for(int i = 0; i < point; i ++){
if(res[i] == 1){
printf("out\n");
}
else
printf("in\n");
}
}
else
printf("No.\n");
printf("FINISH\n");
}
return 0;
}
出错点:
- 字符串的输入。若非一次接收out,而是单独获取字符元素,若上次执行失败(还有元素未接收),会影响下次的输入。造成Time Limit Exceeded
- 内存空间的分配。记录入栈和出栈顺序的数组的大小应该是入栈或出栈序列长度的二倍。造成|Memory Limit Exceeded
- 输出格式,是‘Yes.’和‘No.’而不是YES.和NO. 造成**Wrong Answer
版本2
参考: 杭电OJ–1021 Train Problem I
熟练使用入栈出栈的模板,先入栈,且在入栈的同时判断是否要出栈。
#include<iostream>
#include<stack>
#include<vector>
using namespace std;
#define N 10
// 入栈和出栈序列
char in[N];
char out[N];
int main(){
int x;
while(~scanf("%d%s%s", &x, in, out)){
stack <char> stk;
vector<string> pr;//建立string类型的数组
// 指向入栈和出栈元素
int point_in = 0;
int point_out = 0;
while(point_in < x)
{
// 入栈
stk.push(in[point_in]);
point_in ++;
pr.push_back("in");
// 出栈
while(!stk.empty() && out[point_out] == stk.top()){
stk.pop();
pr.push_back("out");
point_out ++;
}
}
// 输出结果
if(stk.empty()){
printf("Yes.\n");
int len = pr.size();
for(int i = 0; i < len; i ++){
// 将pr[i]以C语言形式输出,或者cout<<pr[i]<<endl;
printf("%s\n", pr[i].c_str());
}
}
else
printf("No.\n");
printf("FINISH\n");
}
return 0;
}