总结:设计这种要实现某些特殊操作的类实质上就是各种数据结构的综合应用,在原有数据结构内部添加一些额外的属性或者其他数据结构来辅助完成操作。所以要解决这样的问题要对常见的数据结构的构成和操作有清楚地认识,并能灵活应用这些数据结构来解决问题。
题目1:Shuffle an Array
打乱一个没有重复元素的数组。
示例:
// 以数字集合 1, 2 和 3 初始化数组。
int[] nums = {1,2,3};
Solution solution = new Solution(nums);
// 打乱数组 [1,2,3] 并返回结果。任何 [1,2,3]的排列返回的概率应该相同。
solution.shuffle();
// 重设数组到它的初始状态[1,2,3]。
solution.reset();
// 随机返回数组[1,2,3]打乱后的结果。
solution.shuffle();
思路:这个类的构造函数就是将传入的数组赋给类自身的数组,而reset()函数则是将该数组返回即可,而shuffle()则是要先将类自身的数组赋给一个临时数组,这样才能保证述数组不被真正打乱,在调用reset()时才能直接返回原有数组;在遍历数组的过程中每遍历到一个元素,就随机生成一个0到len-1的下标,生成下标对应元素和这个元素互换,这样就随机打乱了数组。
class Solution
{
private:
vector<int> arr;
public:
Solution(vector<int>& nums)
{
arr = nums;
}
/** Resets the array to its original configuration and return it. */
vector<int> reset()
{
return arr;
}
/** Returns a random shuffling of the array. */
vector<int> shuffle()
{
vector<int> temp = arr;
int len = arr.size();
for(int i=0;i<len;i++)
{
int random = rand()%len;
swap(temp[i],temp[random]);
}
return temp;
}
};
题目2:最小栈
设计一个支持 push,pop,top 操作,并能在常数时间内检索到最小元素的栈。
1.push(x) -- 将元素 x 推入栈中。
2.pop() -- 删除栈顶的元素。
3.top() -- 获取栈顶元素。
4.getMin() -- 检索栈中的最小元素。
示例:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.getMin(); --> 返回 -2.
思路:设计这个数据结构最关键在于能在常数时间内检索到栈内最小元素,所以在元素入栈的时候要记录好最小元素,调用getMin()的时候直接返回即可。使用两个栈,一个栈myStack用来完成入栈出栈和获取栈顶的操作,另一个栈Min用来记录栈的最小值。对于myStack来说,所有操作都和正常的栈一样;对于Min来说,元素入栈时,如果Min为空那么将该元素放入Min中,若不为空则将元素和栈顶元素比较,将小于等于栈顶元素的元素放入Min中。这样的话最小值就永远在Min的栈顶。出栈的时候,如果出栈的是最小元素那么要将最小元素在Min中删除掉。
class MinStack
{
private:
stack<int> myStack;
stack<int> Min;
public:
/**initialize your data structure here.*/
MinStack()
{
}
void push(int x)
{
myStack.push(x);
if(Min.empty()||x<=Min.top())
Min.push(x);
}
void pop()
{
if(myStack.top()==Min.top())
Min.pop();
myStack.pop();
}
int top()
{
return myStack.top();
}
int getMin()
{
return Min.top();
}
};