劍指offer-20200320

20200320

題目 ：隊列的最大值

請定義一個隊列並實現函數max_value得到隊列裏的最大值，要求函數max_value、push_back和pop_front的均攤時間複雜度都是$O(1)$。

若隊列爲空，pop_front 和 max_value 需要返回 -1

code

class MaxQueue {
	Queue<Integer> que;
    Deque<Integer> deq;
    
    public MaxQueue() {
		que = new LinkedList<>();
        deq = new LinkedList<>();
    }
    
    public int max_value() {
		return deq.size() > 0?deq.peek():-1;
    }
    
    public void push_back(int value) {
		que.offer(value);
        while(deq.size()>0 && deq.peekLast()<value){
            deq.pollLast();
        }
        deq.offerLast(value);//將value插入到deq的隊尾
    }
    
    public int pop_front() {
		int tmp = que.size() > 0 ? que.poll():-1;
        if(deq.size()>0 && tmp == deq.peek()){
            deq.poll();//如果出隊的元素是當前最大值，將deq的對手出隊
        }
        return tmp;
    }
}

題目 ：n個骰子的點數

把n個骰子扔在地上，所有骰子朝上一面的點數之和爲s。輸入n，打印出s的所有可能的值出現的概率。

你需要用一個浮點數數組返回答案，其中第 i 個元素代表這 n 個骰子所能擲出的點數集合中第 i 小的那個的概率。

例如：

輸入: 1
輸出: [0.16667,0.16667,0.16667,0.16667,0.16667,0.16667]

思路：

F(n,s)=F(n−1,s−1)+F(n−1,s−2)+F(n−1,s−3)+F(n−1,s−4)+F(n−1,s−5)+F(n−1,s−6)

class Solution{
    public double[] twoSum(int n){
        int[][] dp = new int[n+1][6*n+1];
        //邊界
        for(int s = 1;s <= 6;s++) dp[1][s] = 1;
        for(int i = 2;i <= n;i++){
            for(int s=i;s<=6*i;s++){
                for(int d=1;d<=6;d++){
                    if(s-d<i-1) break;//爲0了
                    dp[i][s] += dp[i-1][s-d];
                }
            }
        }
        double total = Math.pow((double)6,(double)n);
        double[] ans = new double[5*n+1];
        for(int i=n;i<=6*n;i++){
            ans[i-n]=(double)dp[n][i]/total;
        }
        return ans;
    }
}