小bug in23. Merge k Sorted Lists

這道題注意 priorityqueue里面不能放null, 所以每次入堆的時候要判斷是不是null. 也就是下面這幾個if語句:

for (ListNode list : lists){
    if (list != null){
      pq.offer(list);
    }
}

以及

 while (!pq.isEmpty()){
            ListNode top = pq.poll();
            curt.next = top;
            if (top.next != null){
                pq.offer(top.next);
            }
            curt = curt.next;
            curt.next = null;
        }        

其實就是一開始集體入最小堆的時候要檢查是不是有空鏈表糟趾,然后就是把鏈表的當前節(jié)點加入答案里后归榕,還要把它的next放入最小堆,這時候也要檢查curt.next是否為空。

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode mergeKLists(ListNode[] lists) {
        if (lists == null || lists.length == 0){
            return null;
        }
        int k = lists.length;
        ListNode dummy = new ListNode(0);
        ListNode curt = dummy;
        PriorityQueue<ListNode> pq = new PriorityQueue<>(k, cmp);
        for (ListNode list : lists){
            if (list != null){
                pq.offer(list);
            }
        }
        while (!pq.isEmpty()){
            ListNode top = pq.poll();
            curt.next = top;
            if (top.next != null){
                pq.offer(top.next);
            }
            curt = curt.next;
            curt.next = null;
        }        
        return dummy.next;
    }
    
    private Comparator<ListNode> cmp = new Comparator<ListNode>(){
        public int compare(ListNode l1, ListNode l2){
            return l1.val - l2.val;
        }
    };
}
?著作權歸作者所有,轉載或內容合作請聯系作者
  • 序言:七十年代末贵白,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子趴捅,更是在濱河造成了極大的恐慌羔味,老刑警劉巖,帶你破解...
    沈念sama閱讀 218,284評論 6 506
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件蛆橡,死亡現場離奇詭異舌界,居然都是意外死亡,警方通過查閱死者的電腦和手機泰演,發(fā)現死者居然都...
    沈念sama閱讀 93,115評論 3 395
  • 文/潘曉璐 我一進店門呻拌,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人睦焕,你說我怎么就攤上這事藐握。” “怎么了垃喊?”我有些...
    開封第一講書人閱讀 164,614評論 0 354
  • 文/不壞的土叔 我叫張陵猾普,是天一觀的道長。 經常有香客問我本谜,道長抬闷,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 58,671評論 1 293
  • 正文 為了忘掉前任耕突,我火速辦了婚禮笤成,結果婚禮上,老公的妹妹穿的比我還像新娘眷茁。我一直安慰自己炕泳,他們只是感情好,可當我...
    茶點故事閱讀 67,699評論 6 392
  • 文/花漫 我一把揭開白布上祈。 她就那樣靜靜地躺著培遵,像睡著了一般。 火紅的嫁衣襯著肌膚如雪登刺。 梳的紋絲不亂的頭發(fā)上籽腕,一...
    開封第一講書人閱讀 51,562評論 1 305
  • 那天,我揣著相機與錄音纸俭,去河邊找鬼皇耗。 笑死,一個胖子當著我的面吹牛揍很,可吹牛的內容都是我干的郎楼。 我是一名探鬼主播万伤,決...
    沈念sama閱讀 40,309評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼呜袁!你這毒婦竟也來了敌买?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 39,223評論 0 276
  • 序言:老撾萬榮一對情侶失蹤阶界,失蹤者是張志新(化名)和其女友劉穎虹钮,沒想到半個月后,有當地人在樹林里發(fā)現了一具尸體膘融,經...
    沈念sama閱讀 45,668評論 1 314
  • 正文 獨居荒郊野嶺守林人離奇死亡芙粱,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 37,859評論 3 336
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現自己被綠了托启。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片宅倒。...
    茶點故事閱讀 39,981評論 1 348
  • 序言:一個原本活蹦亂跳的男人離奇死亡攘宙,死狀恐怖屯耸,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情蹭劈,我是刑警寧澤疗绣,帶...
    沈念sama閱讀 35,705評論 5 347
  • 正文 年R本政府宣布,位于F島的核電站铺韧,受9級特大地震影響多矮,放射性物質發(fā)生泄漏。R本人自食惡果不足惜哈打,卻給世界環(huán)境...
    茶點故事閱讀 41,310評論 3 330
  • 文/蒙蒙 一塔逃、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧料仗,春花似錦湾盗、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,904評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至氛改,卻和暖如春帐萎,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背胜卤。 一陣腳步聲響...
    開封第一講書人閱讀 33,023評論 1 270
  • 我被黑心中介騙來泰國打工疆导, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人葛躏。 一個月前我還...
    沈念sama閱讀 48,146評論 3 370
  • 正文 我出身青樓是鬼,卻偏偏與公主長得像肤舞,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子均蜜,可洞房花燭夜當晚...
    茶點故事閱讀 44,933評論 2 355

推薦閱讀更多精彩內容

  • 背景 一年多以前我在知乎上答了有關LeetCode的問題, 分享了一些自己做題目的經驗李剖。 張土汪:刷leetcod...
    土汪閱讀 12,745評論 0 33
  • //leetcode中還有花樣鏈表題,這里幾個例子囤耳,冰山一角 求單鏈表中結點的個數----時間復雜度O(n)這是最...
    暗黑破壞球嘿哈閱讀 1,518評論 0 6
  • 動態(tài)規(guī)劃 111. 爬樓梯思路類似斐波那契數列注意考慮第 0 階的特殊情況 272. 爬樓梯 II思路類似上題篙顺,只...
    6默默Welsh閱讀 2,431評論 0 1
  • 【原文】(11.4) 子曰:“回也非助我者也,于吾言無所不說充择〉旅担” 【通譯】 孔子說:“顏回也不是...
    錢江潮369閱讀 439評論 0 1
  • 盛開的太陽花 和 天空移動的云彩 還有你微笑的臉 那是一幢老舊的樓房 樓下的公交站 是你回家的最后一點 ...
    姚智公閱讀 207評論 0 1