1. 程式人生 > >leetcode (Maximum Average Subarray I)

leetcode (Maximum Average Subarray I)

Title:Maximum Average Subarray I    643

Difficulty:Easy

原題leetcode地址:https://leetcode.com/problems/maximum-average-subarray-i/

 

1.申請了一個數組,將每連續的k個數想加後存放,然後找到這個陣列中最大的數

時間複雜度:O(n^2),巢狀for迴圈。

空間複雜度:O(n),最長需要申請的空間長度為n。

    /**
     * 申請了一個數組,將每連續的k個數想加存放,然後找到這個陣列中最大的數,再求平均數
     * @param nums
     * @param k
     * @return
     */
    public static double findMaxAverage(int[] nums, int k) {

        if (nums == null || nums.length <= 0 || k <= 0 || k > nums.length) {
            return 0;
        }

        int max = Integer.MIN_VALUE;
        int tmp[] = new int[nums.length - k + 1];

        for (int i = 0; i < nums.length - k + 1; i++) {
            for (int j = i; j < k + i; j++) {
                tmp[i] += nums[j];
            }
        }

        for (int i = 0; i < tmp.length; i++) {
            if (tmp[i] >= max) {
                max = tmp[i];
            }
        }
        System.out.println(max);

        return max / (k * 1.0);
    }

2. 先計算出第一個K個數的乘積,然後加上後面一個數減去前面一個數,再進行判斷大小值

時間複雜度:O(n),兩次一層for迴圈。

空間複雜度:O(1),沒有申請額外空間。

    /**
     * 先計算出第一個K個數的乘積,然後加上後面一個數減去前面一個數,再進行判斷大小值,最後計算平均值
     * @param nums
     * @param k
     * @return
     */
    public static double findMaxAverage1(int[] nums, int k) {

        if (nums == null || nums.length <= 0 || k <= 0 || k > nums.length) {
            return 0;
        }

        int sum = 0;
        for (int i = 0; i < k; i++) {
            sum += nums[i];
        }

        int max = sum;
        for (int i = k; i < nums.length; i++) {
            sum += (nums[i] - nums[i - k]);
            max = max > sum ? max : sum;
        }
        System.out.println(max);

        return max / (k * 1.0);
        
    }