Advertisement
sweet1cris

Untitled

Jan 9th, 2018
83
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Java 1.26 KB | None | 0 0
  1. public class Solution {
  2.     /**
  3.      * @param nums an array with positive and negative numbers
  4.      * @param k an integer
  5.      * @return the maximum average
  6.      */
  7.     public double maxAverage(int[] nums, int k) {
  8.         // Write your code here
  9.         double l = Integer.MAX_VALUE, r = Integer.MIN_VALUE;
  10.         for (int i = 0; i < nums.length; ++i) {
  11.             if (nums[i] < l)
  12.                 l = nums[i];
  13.             if (nums[i] > r)
  14.                 r = nums[i];
  15.         }
  16.        
  17.        
  18.         while (r - l >= 1e-6) {
  19.             double mid = (l + r) / 2.0;
  20.  
  21.             if (check_valid(nums, mid, k)) {
  22.                 l = mid;
  23.             }
  24.             else {
  25.                 r = mid;
  26.             }
  27.         }
  28.  
  29.         return l;
  30.     }
  31.    
  32.     private boolean check_valid(int nums[], double mid, int k) {
  33.         int n = nums.length;
  34.         double min_pre = 0;
  35.         double[] sum = new double[n + 1];
  36.         sum[0] = 0;
  37.         for (int i = 1; i <= n; ++i) {
  38.             sum[i] = sum[i - 1] + nums[i - 1] - mid;
  39.             if (i >= k && sum[i] - min_pre >= 0) {
  40.                 return true;
  41.             }
  42.             if (i >= k)
  43.                 min_pre = Math.min(min_pre, sum[i - k + 1]);
  44.         }
  45.         return false;
  46.     }
  47. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement