程序员面试题精选100题(02)－设计包含min函数的栈[数据结构] - c实现

题目来源：http://zhedahht.blog.163.com/blog/static/25411174200712895228171/

题目：定义栈的数据结构，要求添加一个min函数，能够得到栈的最小元素。要求函数min、push以及pop的时间复杂度都是O(1)。

分析：这是去年google的一道面试题。

我看到这道题目时，第一反应就是每次push一个新元素时，将栈里所有逆序元素排序。这样栈顶元素将是最小元素。但由于不能保证最后push进栈的元素最先出栈，这种思路设计的数据结构已经不是一个栈了。

在栈里添加一个成员变量存放最小元素（或最小元素的位置）。每次push一个新元素进栈的时候，如果该元素比当前的最小元素还要小，则更新最小元素。

乍一看这样思路挺好的。但仔细一想，该思路存在一个重要的问题：如果当前最小元素被pop出去，如何才能得到下一个最小元素？

因此仅仅只添加一个成员变量存放最小元素（或最小元素的位置）是不够的。我们需要一个辅助栈。每次push一个新元素的时候，同时将最小元素（或最小元素的位置。考虑到栈元素的类型可能是复杂的数据结构，用最小元素的位置将能减少空间消耗）push到辅助栈中；每次pop一个元素出栈的时候，同时pop辅助栈。

补充：关键就是要制造一个辅助栈，栈顶元素始终为当前栈的最小值，当前栈弹出的元素为最小值时，则辅助栈弹出栈顶元素。

一下为我的c实现，栈功能是自己写的，没有用标准库函数

/**
2012-7-9 kimili BUPT CS
Email:lichunwen1987ATqq.com
**/

#include <stdio.h>
#include <stdlib.h>
#include <time.h> 
#define Elem int
#define StackSize 30         //栈初始大小
#define StackIncrease 10     //栈增加大小
#define Min 0x7fffffff
#define MAX 100 
 
typedef struct {
    Elem * top;     //栈顶位置
    Elem * base;    //栈底位置
    int size;     //栈的大小 
    int min;      //栈当前最小值
}SqStack;

//初始化栈
int InitStack(SqStack &s){
     s.base = (Elem *)malloc(StackSize * sizeof(Elem));
     if(! s.base) return -1;    //分配内存失败，初始化栈失败 
     s.top = s.base;
     s.size = StackSize;
     s.min = Min; 
     return 1;  //初始化栈成功 
} 

//是否为空，如果空，返回1，不空返回0 
int Empty(SqStack s){
    //如果为空，返回1 
    if (s.top == s.base) return 1;  
    else return 0;
}

//返回栈顶元素
Elem GetTop(SqStack s){
    Elem e; 
    if(!Empty(s)){ 
        e = *(s.top-1);
        return e;
    }else{
        return -1;
    } 
} 

//返回栈最小元素
Elem MinElem(SqStack s){
    return GetTop(s);
} 

//push元素，同时可能会更新栈sMin
int Push(SqStack &s, SqStack &sMin, Elem e){
    //栈满了，增加空间 
    if(s.top - s.base >= s.size -1 ){ 
        s.base = (Elem *)realloc(s.base, (s.size + StackIncrease) * sizeof(Elem));
        if(! s.base) return -1;
        //注意，此行代码很关键，realloc函数之后s.base的地址可能会变，这
        //时s.top的相对偏移就不准确了 ，所以要重新校准偏移 ，否则会出现错误数据
        s.top = s.base + s.size -1 ;         
        s.size += StackIncrease; 
    }
    * s.top ++ = e; //添加元素
    //如果新加的数更小，更新辅助栈 
    if (e < s.min) {
        s.min = e;
        //注意辅助栈，也要变化大小 
        if(sMin.top - sMin.base == sMin.size -1 ){ 
            sMin.base = (Elem *)realloc(sMin.base, (sMin.size + StackIncrease) * sizeof(Elem));
            if(! sMin.base) return -1;
            sMin.top = sMin.base + sMin.size -1; 
            sMin.size += StackIncrease; 
        }
        * sMin.top ++ = e; //
    } 
    return 1;
}

int Pop(SqStack &s, SqStack &sMin, Elem &e){
   // if(s.top != s.base){ 
    if(!Empty(s)){ 
        e = * --s.top ; //top指向的是栈顶值的上面一个
		//如果弹出的元素是栈最小（就是辅助栈的栈顶元素），则辅助栈弹出一个元素
        if(e == *(sMin.top-1)) 
			-- sMin.top;     
        return 1;
    } else{
        return -2;
    }   
} 
         
int main(){
    
    SqStack s,sMin;
    InitStack(s); 
    InitStack(sMin); //初始化最小值的栈
   
    int n,*a,i;
    Elem e; 
        
    printf("输入数组数量：");
    scanf("%d",&n);
    a=(int *)malloc((n+1)*sizeof(int));   
    srand((unsigned)time(NULL));  //随机生成数组，输入数组数量即可
    for(i=1;i<=n;i++){ 
        a[i] = rand()%MAX;
        printf("%d ",a[i]);
        Push(s,sMin,a[i]); 
    }
    
    printf("\n");
    for(i =1;i<=n;i++){
        Pop(s,sMin,e); 
        printf("pop= %d ,min = %d ", e,MinElem(sMin));
        printf("\n"); 
    } 
    printf("\n"); 
    
    //printf("%d\n", Min); 
    free(a); 
    free(s.base);
    free(sMin.base);
    system("pause"); 
    return 0;
}