此题本意是对一个栈中的数据进行排序,最多用另一个栈来缓存。但在牛客网上利用vector来模拟栈,还规定只能访问头元素。。。
题目原意:编写程序,按升序对栈进行排序(即最大元素位于栈顶)。最多只能用一个额外的栈存放临时数据,但不得将元素复制到别的数据结构中(如数组)。
class TwoStacks {
public:
vector<int> twoStacksSort(stack<int> nums) { //利用两个栈排序
stack<int> nums_copy;
while (nums.empty()) {
int tmp = nums.top();
nums.pop();
while (!nums_copy.empty() && nums_copy.top() > tmp) {
nums.push(nums_copy.top());
nums_copy.pop();
}
nums_copy.push(tmp);
}
return nums_copy;
}
}
【无甚意义】牛客网山的OJ代码
class TwoStacks {
public:
vector<int> twoStacksSort(vector<int> numbers) { //这题目出的太烂!
sort(numbers.begin(), numbers.end(), my_compare);
return numbers;
}
static int my_compare(int a, int b) { //写在类内的比较函数,必须声明为static的!!!
if(a >= b)
return 1;
return 0;
}
};