iOS 接雨水算法Swift

題目描述:

給定 n 個(gè)非負(fù)整數(shù)表示每個(gè)寬度為 1 的柱子的高度圖臊岸,計(jì)算按此排列的柱子址否,下雨之后能接多少雨水占婉。


接雨水題目描述.png

上面是由數(shù)組 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度圖酒请,在這種情況下量愧,可以接 6 個(gè)單位的雨水(藍(lán)色部分表示雨水)。 感謝 Marcos 貢獻(xiàn)此圖溃槐。
示例:
輸入: [0,1,0,2,1,0,1,3,2,1,2,1]
輸出: 6

解題思路:

思路一匣砖,時(shí)間復(fù)雜度由希爾排序決定,最大N*logN昏滴,以[90,100,56,10,0,80,30,60]為例:

  1. 降序數(shù)組猴鲫,數(shù)組內(nèi)容為值及對應(yīng)的原始位置(location,value)[(1,100),(0,90),(5,80),(7,60),(2,56),(6,30),(3,10),(4,0)]
  2. 遍歷排序后的數(shù)組谣殊,根據(jù)能夠接水的必要條件是兩個(gè)柱子之間必須有間隙拂共,即不能相鄰,拿到可以接水的區(qū)間:
  • 100跟90之間沒有間隙姻几,找下一個(gè)宜狐;
  • 100跟80之間有間隙势告,符合可以接水的必要條件,存儲(chǔ)該區(qū)間gap =(1抚恒,5)咱台,并計(jì)算該區(qū)間可以接水的量:遍歷區(qū)間(2~4)內(nèi)的數(shù)據(jù)container += min(100,80) - arr[i];
  • 繼續(xù)循環(huán),看是否在gap區(qū)間對應(yīng)(0俭驮,6)范圍內(nèi)(其中0和6回溺,如果在,由于已經(jīng)計(jì)算表鳍,continue),80不再區(qū)間祥诽,根據(jù)位置關(guān)系更新gap為(1譬圣,7),計(jì)算新增區(qū)間(5雄坪,7)可以接水的量container += min(100,80) - arr[i]厘熟;
  • 比較gap區(qū)間跟整個(gè)數(shù)組范圍(1,count -2)關(guān)系维哈,如果小于等于gap绳姨,計(jì)算完畢,沒有繼續(xù)c步驟阔挠。
    思路一.png

    代碼如下:

static func trap(_ height: [Int]) -> Int {
        if height.count <= 2 {
            return 0
        }
        let sortHeight = sortArr(arr: height)///降序飘庄,拿到值及對應(yīng)的原始位置 [(原始位置,值)]
        
        var container = 0
        var gap :(Int,Int)?
        
        for k in 0..<sortHeight.count-1 {
            let firstL = sortHeight[k]
            guard let gugap = gap else {
                var i = k+1
                
                while (gap ?? nil) == nil && i < sortHeight.count{///從最大值(如果不是最大值開始购撼,可能漏算)開始跪削,拿到第一段可以接水的區(qū)間。
                    let secondL = sortHeight[I]
                    if abs(firstL.0 - secondL.0) > 1 { ///可以接水的必要條件是迂求,兩個(gè)柱子之間有間隙
                        let start = min(secondL.0, firstL.0)
                        let end = max(secondL.0, firstL.0)
                        gap = (start,end)///保存區(qū)間
                        for g in start + 1...end-1 {///計(jì)算此區(qū)間可以接水的量
                            container += secondL.1 - height[g] < 0 ? 0:secondL.1 - height[g]///雖然中間有間隔碾盐,但是不能保證能接水,去掉不能接水揩局,即中間的柱子不是最小
                            
                        }
                        if start <= 1 && end >= sortHeight.count - 2{
                            return container
                        }
                    }
                    i += 1
                }
                continue
            }
            if firstL.0 <= gugap.1 + 1 && firstL.0 >= gugap.0 - 1  {///在已經(jīng)結(jié)算的區(qū)間內(nèi)毫玖,continue
                continue
            }else{///不再已經(jīng)結(jié)算的區(qū)間,根據(jù)位置關(guān)系凌盯,更新結(jié)算區(qū)間
                var start = 0
                var end  = 0
                if firstL.0 > gugap.1 {///在右邊
                    start = gugap.1
                    end = firstL.0
                    gap = (gugap.0,end)
                }else{///在左邊
                    start = firstL.0
                    end = gugap.0
                    gap = (start,gugap.1)
                }
                for g in start + 1...end-1 {///計(jì)算新增區(qū)間可以接水的量付枫,同樣可能會(huì)有不能接水的可能,剔除
                    container += firstL.1 - height[g] < 0 ? 0:firstL.1 - height[g]
                    
                }
                if start <= 1 && end >= sortHeight.count - 2{
                    return container
                }
            }
        }
        return container
    }
    static func sortArr(arr:[Int]) -> [(Int,Int)] {///希爾排序
        let count = arr.count
        var locValueTump :[(Int,Int)] = []
        for i in 0..<count {
            locValueTump.append((i,arr[i]))///記錄位置及對應(yīng)的值
        }
        
        var gap = count/2
        while gap > 0 {
            for i in gap..<count{
                var j = I
                let temp = locValueTump[j]
                if locValueTump[j].1 > locValueTump[j-gap].1 {
                    while j-gap>=0 && temp.1 > locValueTump[j-gap].1 {
                        locValueTump[j] = locValueTump[j-gap]
                        j -= gap
                    }
                    locValueTump[j] = temp
                }
            }
            gap /= 2
        }
        
        return locValueTump
    }

思路二驰怎,比較暴力時(shí)間復(fù)雜度O(n^2):

  1. 從數(shù)組的第二個(gè)元素開始到倒數(shù)第二個(gè)元素遍歷励背,,向左向右分別取到最大值max_left,max_right砸西;
  2. 計(jì)算接水量 container += min(max_left,max_right) - height[I]叶眉。
    思路二.png
static func trap(_ height: [Int]) -> Int {
        if height.count <= 2 {
            return 0
        }
        var ans = 0;
        let size = height.count;
        for i in 0..<size - 1 {//向左尋找最大
            var max_left = 0, max_right = 0;
            for j in 0...i { //Search the left part for max bar size
                max_left = max(max_left, height[j]);
            }
            for j in i..<size { //向右邊查找最大值
                max_right = max(max_right, height[j]);
            }
            ans += min(max_left, max_right) - height[i];
        }
        return ans;
}

接雨水

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末址儒,一起剝皮案震驚了整個(gè)濱河市,隨后出現(xiàn)的幾起案子衅疙,更是在濱河造成了極大的恐慌莲趣,老刑警劉巖,帶你破解...
    沈念sama閱讀 216,496評論 6 501
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件饱溢,死亡現(xiàn)場離奇詭異喧伞,居然都是意外死亡,警方通過查閱死者的電腦和手機(jī)绩郎,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 92,407評論 3 392
  • 文/潘曉璐 我一進(jìn)店門潘鲫,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人肋杖,你說我怎么就攤上這事溉仑。” “怎么了状植?”我有些...
    開封第一講書人閱讀 162,632評論 0 353
  • 文/不壞的土叔 我叫張陵浊竟,是天一觀的道長。 經(jīng)常有香客問我津畸,道長振定,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 58,180評論 1 292
  • 正文 為了忘掉前任肉拓,我火速辦了婚禮后频,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘暖途。我一直安慰自己徘郭,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,198評論 6 388
  • 文/花漫 我一把揭開白布丧肴。 她就那樣靜靜地躺著残揉,像睡著了一般。 火紅的嫁衣襯著肌膚如雪芋浮。 梳的紋絲不亂的頭發(fā)上抱环,一...
    開封第一講書人閱讀 51,165評論 1 299
  • 那天,我揣著相機(jī)與錄音纸巷,去河邊找鬼镇草。 笑死,一個(gè)胖子當(dāng)著我的面吹牛瘤旨,可吹牛的內(nèi)容都是我干的梯啤。 我是一名探鬼主播,決...
    沈念sama閱讀 40,052評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼存哲,長吁一口氣:“原來是場噩夢啊……” “哼因宇!你這毒婦竟也來了七婴?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 38,910評論 0 274
  • 序言:老撾萬榮一對情侶失蹤察滑,失蹤者是張志新(化名)和其女友劉穎打厘,沒想到半個(gè)月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體贺辰,經(jīng)...
    沈念sama閱讀 45,324評論 1 310
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡户盯,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,542評論 2 332
  • 正文 我和宋清朗相戀三年,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了饲化。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片莽鸭。...
    茶點(diǎn)故事閱讀 39,711評論 1 348
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡,死狀恐怖吃靠,靈堂內(nèi)的尸體忽然破棺而出硫眨,到底是詐尸還是另有隱情,我是刑警寧澤撩笆,帶...
    沈念sama閱讀 35,424評論 5 343
  • 正文 年R本政府宣布捺球,位于F島的核電站缸浦,受9級特大地震影響夕冲,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜裂逐,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,017評論 3 326
  • 文/蒙蒙 一歹鱼、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧卜高,春花似錦弥姻、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,668評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至薪缆,卻和暖如春秧廉,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背拣帽。 一陣腳步聲響...
    開封第一講書人閱讀 32,823評論 1 269
  • 我被黑心中介騙來泰國打工疼电, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人减拭。 一個(gè)月前我還...
    沈念sama閱讀 47,722評論 2 368
  • 正文 我出身青樓蔽豺,卻偏偏與公主長得像,于是被迫代替她去往敵國和親拧粪。 傳聞我的和親對象是個(gè)殘疾皇子修陡,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 44,611評論 2 353