Java 算法-最大平均值子數(shù)組 (二分法)

??今天遇到了一道題,本來(lái)題的不難的,但是就是因?yàn)榧尤肓撕芏嗟南拗疲瑢?dǎo)致了這個(gè)題比較棘手。

題意

給出一個(gè)整數(shù)數(shù)組,有正有負(fù)。找到這樣一個(gè)子數(shù)組,他的長(zhǎng)度大于等于 k,且
平均值最大。

樣例

給出 nums = [1, 12, -5, -6, 50, 3], k = 3

返回 15.667 // (-6 + 50 + 3) / 3 = 15.667

注意事項(xiàng)

保證數(shù)組的大小 >= k

1.傳統(tǒng)的方法(超時(shí))

??這個(gè)題的傳統(tǒng)非常的容易,這里不便再贅述

public static double maxAverage(int[] nums, int k) {
        //記錄當(dāng)前k(或者大于k)位數(shù)字和
        double sum = 0;
        double max = Integer.MIN_VALUE;
        if(nums == null || nums.length == 0 || k == 0){
            return 0;
        }
        for(int j = k; j <= nums.length; j++){
            //初始化為0
            sum = 0;
            for(int i = 0; i < nums.length; i++){
                if(i >= j){//當(dāng)i > j時(shí),計(jì)算最大值
                    max = ((sum * 1.0 / j) > max) ? sum * 1.0 / j : max;
                    sum -= nums[i - j]; //需要減去當(dāng)前選擇的數(shù)字的第一個(gè),因?yàn)榧磳⒃诩尤胍粋€(gè)
                }
                //加入一個(gè)數(shù)字
                sum += nums[i];
            }
            max = ((sum * 1.0 / j) > max) ? sum * 1.0 / j : max;
        }
        return max;
    }

2.二分法

??對(duì)的,又是二分法。跟前面的快慢指針一樣,這里也是非常規(guī)的二分法使用。

(1).題意介紹

??這個(gè)題讓我們求的是一個(gè)數(shù)組的子數(shù)組(長(zhǎng)度大于等于k)的最大平均值,記住不是最大值,是最大平均值。因?yàn)榈贸鲎畲笾挡灰欢軌蛲瞥鲎畲蟮钠骄担驗(yàn)槠骄挡粌H受和的影響,還要受個(gè)數(shù)影響。

(2).算法概述

??首先一個(gè)數(shù)組的和的平均值一定在這個(gè)數(shù)組中的最小值和最大值之間。這一點(diǎn)肯定沒(méi)有爭(zhēng)議。
??其次,我們現(xiàn)在要做的就是--二分,在這個(gè)范圍里面,不斷的二分,直到找到我們想要的值,但是又應(yīng)該怎么二分呢?
??我們可以這樣來(lái)假設(shè),假設(shè) 最大值為max,最小值為min,中間值為 mid = (min + max ) /2.0(這里為什么要除以2.0,而不是2呢?那是我們?cè)赱min, max]區(qū)間求得我們想要的那個(gè)平均值,而這個(gè)平均值不可能總是整數(shù),有可能是小數(shù)(double),所以,我們除以2.0而不是2)。假設(shè)nums數(shù)組(nums[0], nums[1], nums[2], nums[3].......),如果存在兩個(gè)下標(biāo)i,j(i > j),長(zhǎng)度i - j >= k,使其(相當(dāng)于是nums數(shù)組的子數(shù)組)平均值大于等于mid的值,那么:

(nums[i] + nums[i + 1] + ...... + nums[j]) / (i - j) >= mid.

??所以,得出我們想要的答案必定在[mid, max]區(qū)間里面;反之,如果不存在這兩個(gè)下標(biāo),也就是所有的子數(shù)組的平均值都小于mid,那么我們求得最終的答案肯定小于mid,所以最終的答案肯定在[min, mid]。
??那么,怎么判斷一個(gè)子數(shù)組(長(zhǎng)度L >= k)和的平均值大于mid呢?
??這個(gè)是我們這個(gè)問(wèn)題的一個(gè)難點(diǎn),我們用數(shù)學(xué)公式來(lái)表達(dá),如圖所示:



??如圖上的結(jié)論,我們還可以假設(shè)s數(shù)組,其中

                      s[0] = 0;

                      s[1] = b[0];

                      s[2] = b[0] + b[1] = s[1] + b[1];

                      s[3] = b[0] + b[1] + b[2] = s[2] + b[2];

?? 那么b[j]到b[i](i > j)區(qū)間和為s[i + 1] - s[j]。要想找出區(qū)間和(長(zhǎng)度大于等于k)的最大值,等價(jià)于找i,j,滿(mǎn)足i - j >= k并且使得s[i] - s[j]最大。要想求出s[i + 1] - s[j]的最大值,我們固定i,只要s[j]最小的話(huà),那么s[i + 1] - s[j]就最大了。

(3).代碼

public static double maxAverage(int[] nums, int k) {
        if (nums == null || nums.length == 0 || k == 0) {
            return 0;
        }
        double max = Integer.MIN_VALUE;
        double min = Integer.MAX_VALUE;
        for(int i = 0; i < nums.length; i++){
            if(nums[i] > max){
                max = nums[i];
            }
            if(nums[i] < min){
                min = nums[i];
            }
        }
        while (max - min >= 1e-6) {
            double mid = (max + min) / 2.0;
            if(binarySearch(nums, mid, k)){//表示最大的平均值在[min, mid]里面
                min = mid;
            }else{//表示最大的平均值在[mid, max]里面
                max = mid;
            }
        }
        return min;
    }

    private static boolean binarySearch(int[] nums, double mid, int k) {
        //用來(lái)記錄區(qū)間和的
        double sum[] = new  double[nums.length + 1];
        double min = 0;
        sum[0] = 0;
        for (int i = 1; i <= nums.length; i++) {
            //計(jì)算每個(gè)點(diǎn)的區(qū)間和
            sum[i] = sum[i - 1] + (nums[i - 1] - mid);
            //min相當(dāng)于是s[j],不過(guò)這里是一步一步的取小值得出來(lái)的
            if (i >= k && sum[i] >= min) {//表示有b數(shù)組的和>=0,則取值范圍變?yōu)閇mid, max]
                return true;
            }  if (i >= k) {
                min = Math.min(min, sum[i - k + 1]); //一步一步的取小值
            }
        }
        return false;
    }
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書(shū)系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。
  • 序言:七十年代末,一起剝皮案震驚了整個(gè)濱河市,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 230,501評(píng)論 6 544
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場(chǎng)離奇詭異,居然都是意外死亡,警方通過(guò)查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 99,673評(píng)論 3 429
  • 文/潘曉璐 我一進(jìn)店門(mén),熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái),“玉大人,你說(shuō)我怎么就攤上這事。” “怎么了?”我有些...
    開(kāi)封第一講書(shū)人閱讀 178,610評(píng)論 0 383
  • 文/不壞的土叔 我叫張陵,是天一觀的道長(zhǎng)。 經(jīng)常有香客問(wèn)我,道長(zhǎng),這世上最難降的妖魔是什么? 我笑而不...
    開(kāi)封第一講書(shū)人閱讀 63,939評(píng)論 1 318
  • 正文 為了忘掉前任,我火速辦了婚禮,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 72,668評(píng)論 6 412
  • 文/花漫 我一把揭開(kāi)白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上,一...
    開(kāi)封第一講書(shū)人閱讀 56,004評(píng)論 1 329
  • 那天,我揣著相機(jī)與錄音,去河邊找鬼。 笑死,一個(gè)胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 44,001評(píng)論 3 449
  • 文/蒼蘭香墨 我猛地睜開(kāi)眼,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼!你這毒婦竟也來(lái)了?” 一聲冷哼從身側(cè)響起,我...
    開(kāi)封第一講書(shū)人閱讀 43,173評(píng)論 0 290
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒(méi)想到半個(gè)月后,有當(dāng)?shù)厝嗽跇?shù)林里發(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 49,705評(píng)論 1 336
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 41,426評(píng)論 3 359
  • 正文 我和宋清朗相戀三年,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 43,656評(píng)論 1 374
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 39,139評(píng)論 5 364
  • 正文 年R本政府宣布,位于F島的核電站,受9級(jí)特大地震影響,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 44,833評(píng)論 3 350
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開(kāi)封第一講書(shū)人閱讀 35,247評(píng)論 0 28
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)。三九已至,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開(kāi)封第一講書(shū)人閱讀 36,580評(píng)論 1 295
  • 我被黑心中介騙來(lái)泰國(guó)打工, 沒(méi)想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個(gè)月前我還...
    沈念sama閱讀 52,371評(píng)論 3 400
  • 正文 我出身青樓,卻偏偏與公主長(zhǎng)得像,于是被迫代替她去往敵國(guó)和親。 傳聞我的和親對(duì)象是個(gè)殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 48,621評(píng)論 2 380

推薦閱讀更多精彩內(nèi)容