剑指OFFER之包含min函数的栈(九度OJ1522)

题目描述:

定义栈的数据结构,请在该类型中实现一个能够得到栈最小元素的min函数。

输入:

输入可能包含多个测试样例,输入以EOF结束。
对于每个测试案例,输入的第一行为一个整数n(1<=n<=1000000), n代表将要输入的操作的步骤数。
接下来有n行,每行开始有一个字母Ci。
Ci=’s’时,接下有一个数字k,代表将k压入栈。
Ci=’o’时,弹出栈顶元素。

输出:

对应每个测试案例中的每个操作,
若栈不为空,输出相应的栈中最小元素。否则,输出NULL。

样例输入:
7

s 3

s 4

s 2

s 1

o

o

s 0

 

样例输出:
3

3

2

1

2

3

0

解题思路:

  很容易想到,在栈中的元素做手脚就行啦。我们每次同时保存两个信息,一个是当前的元素的值,一个是当前的最小值

typedef struct stack{

    int num;

    int min;

}Stack;

typedef struct arr{

    Stack s[1000000];

}Arr;

 

  不使用栈,也可以考虑使用头插的链表,这样会节省内存。

代码:

#include <stdio.h>

#include <stdlib.h>

typedef struct stack{

    int num;

    int min;

}Stack;

typedef struct arr{

    Stack s[1000000];

}Arr;

int main(int argc, char const *argv[])

{

    int n,i,m;

    char c;

    while(scanf("%d",&n)!=EOF && n>=1 && n<=1000000){

        Arr *a = (Arr *)malloc(sizeof(Arr));

        int top = 0;

        for(i=0;i<n;i++){

            scanf("\n%c",&c);

            if(c == 's'){

                scanf("%d",&m);

                if(top == 0){

                    a->s[top].min = m;

                }else{

                    a->s[top].min = (m<a->s[top-1].min?m:a->s[top-1].min);

                }

                a->s[top].num = m;

                top++;

                printf("%d\n",a->s[top-1].min);

            }

            else{

                if(top > 1){

                    top--;

                    printf("%d\n",a->s[top-1].min);

                }else{

                    top = 0;

                    printf("NULL\n");

                }

            }

        }

    }

    return 0;

}

/**************************************************************

    Problem: 1522

    User: xhalo

    Language: C

    Result: Accepted

    Time:20 ms

    Memory:8728 kb

****************************************************************/

 

你可能感兴趣的:(函数)