[华为机试练习题]26.铁路栈问题

题目

描述: 题目标题：铁路栈问题

铁路的调度站如下：

火车编号为：1~9，且不重复。

如：编号分别为“1”、“2”、“3”、“4”、“5”的5个火车顺序进站，那么进站序列为“12345”，全部进站后再顺序出站，则出站序列为“54321”，如果先进1,2，然后2出站，然后1出站，然后再3进站、出站，4进站、出站，5进站、出站，那么出站序列就为21345.

详细描述：

int JudgeTrainSequence (int maxNum, char *pOutSeq);

输入参数：

int maxNum：进站的火车最大编号

char* pOutSeq：使用字符串表示火车出站序列

输出参数（指针指向的内存区域保证有效）：

无。

返回值：

Int: 根据输入的进站序列判断，如果输入的出站序列是可能的，返回1，否则返回0；

练习阶段: 高级

代码

/*--------------------------------------- * 日期：2015-07-01 * 作者：SJF0115 * 题目：铁路栈问题 * 来源：华为机试练习题 -----------------------------------------*/
#include <iostream>
#include <string.h>
#include <stack>
using namespace std;

/* 详细描述： int JudgeTrainSequence (int maxNum, char *pOutSeq); 输入参数： int maxNum：进站的火车最大编号 char* pOutSeq：使用字符串表示火车出站序列 输出参数（指针指向的内存区域保证有效）： 无。 返回值： Int: 根据输入的进站序列判断，如果输入的出战序列是可能的，返回1，否则返回0； */

int JudgeTrainSequence (int maxNum, char *pOutSeq){
    if(pOutSeq == NULL || maxNum <= 0){
        return 0;
    }//if
    int size = strlen(pOutSeq);
    stack<int> trainSeq;
    // 初始
    int index = 1;
    for(int i = 0;i < size;){
        if(trainSeq.empty() || trainSeq.top() < pOutSeq[i] - '0'){
            trainSeq.push(index);
            ++index;
        }//if
        else{
            trainSeq.pop();
            ++i;
        }//else
    }//for
    if(!trainSeq.empty()){
        return 0;
    }//if
    return 1;
}