码迷,mamicode.com
首页 > 编程语言 > 详细

算法题:双栈排序

时间:2015-08-21 00:12:38      阅读:184      评论:0      收藏:0      [点我收藏+]

标签:

题目描述

请编写一个程序,按升序对栈进行排序(即最大元素位于栈顶),要求最多只能使用一个额外的栈存放临时数据,但不得将元素复制到别的数据结构中。

给定一个int[] numbers(C++中为vector<int>),其中第一个元素为栈顶,请返回排序后的栈。请注意这是一个栈,意味着排序过程中你只能访问到第一个元素。

测试样例:
[1,2,3,4,5]
返回:[5,4,3,2,1]


class TwoStacks {
public:
    vector<int> twoStacksSort(vector<int> numbers) {
        // write code here
        stack<int> st1;
        stack<int> st2;
        int n = numbers.size();
        if(n<=1)return numbers;
        int temp;
        for(int i=0;i<n;i++)
        {
              temp = numbers[i];
			  for(;st1.empty()==false;)
              {
              		if(temp<st1.top())
                     {
                        st2.push(st1.top());
                     	st1.pop();   
                     }
                  	 else
                     {
                         break;
                     }
              }
            st1.push(temp);
            while(st2.empty()==false)
            {
                st1.push(st2.top());
                st2.pop();
            }
        }
        vector<int> v;
        while(st1.empty()==false)
        {
            v.push_back(st1.top());
            st1.pop();
        }
        return v;
    }
};


版权声明:本文为博主原创文章,未经博主允许不得转载。

算法题:双栈排序

标签:

原文地址:http://blog.csdn.net/liuhuiyan_2014/article/details/47818901

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!