题解 | 合并k个已排序的链表

import java.util.*;

/*
 * public class ListNode {
 *   int val;
 *   ListNode next = null;
 *   public ListNode(int val) {
 *     this.val = val;
 *   }
 * }
 */

public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param lists ListNode类ArrayList 
     * @return ListNode类
     */
    public ListNode mergeKLists (ArrayList<ListNode> lists) {
        
        ListNode nNode = minNode(lists);
        ListNode tmp = nNode;
        while(lists.size()>0){
            tmp.next = minNode(lists);
            tmp = tmp.next;
        }
        return nNode;
        // write code here
    }
// 获取最小的那一个node，并重设lists中存储的header
    public static ListNode minNode(ArrayList<ListNode> lists){
        lists.removeIf(itm->itm==null);

        if(lists.size()<1){
            return null;
        }
        ListNode min = lists.get(0);
        int idx = 0 ;
        for(int i=1;i<lists.size();i++){
            if(min.val>lists.get(i).val){
                min = lists.get(i);
                idx = i;
            }
        }
        if(min.next!=null){
            lists.set(idx,min.next);
        }else{
            lists.remove(idx);
        }
        return min;
    }
    
}