1. 程式人生 > >[leetcode初級演算法]設計問題總結

[leetcode初級演算法]設計問題總結

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();

思路:

每一個位置都有可能跟剩下的n-1個位置進行交換

c++程式碼:

class Solution {
public:
    Solution(vector<int> nums) {
        vec=nums;
    }
    
    /** Resets the array to its original configuration and return it. */
    vector<int> reset() {
        return vec;
    }
    
    /** Returns a random shuffling of the array. */
    vector<int> shuffle
() { vector<int> tmp=vec; for(int i=0;i<vec.size();i++) { int j=rand()%vec.size(); //交換tmp[i],tmp[j] int t=tmp[i]; tmp[i]=tmp[j]; tmp[j]=t; } return tmp; } vector<int> vec; }; /** * Your Solution object will be instantiated and called as such: * Solution obj = new Solution(nums); * vector<int> param_1 = obj.reset(); * vector<int> param_2 = obj.shuffle(); */

最小棧

設計一個支援 push,pop,top 操作,並能在常數時間內檢索到最小元素的棧。 push(x) – 將元素 x 推入棧中。 pop() – 刪除棧頂的元素。 top() – 獲取棧頂元素。 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.

思路:

建立兩個stack,一個儲存所有值,一個儲存最小值

class MinStack {
public:
    /** initialize your data structure here. */
    MinStack() {
        
    }
    
    void push(int x) {
        s1.push(x);
        if(s2.empty()||s2.top()>=x)//注意這個等號,測試用例:入棧0 1 0,出棧0 1 
            s2.push(x);
    }
    
    void pop() {
        if(s1.top()==s2.top())
            s2.pop();
        s1.pop();
        
    }
    
    int top() {
        return s1.top();
    }
    
    int getMin() {
        return s2.top();
    }
    stack<int> s1,s2;
};

/**
 * Your MinStack object will be instantiated and called as such:
 * MinStack obj = new MinStack();
 * obj.push(x);
 * obj.pop();
 * int param_3 = obj.top();
 * int param_4 = obj.getMin();
 */