牛客-C语言解法-合并k个已排序的链表

 题目链接：

合并k个已排序的链表_牛客题霸_牛客网 (nowcoder.com)

题目简介：

描述

合并 k 个升序的链表并将结果作为一个升序的链表返回其头节点。

数据范围：节点总数 0≤n≤5000，每个节点的val满足 ∣val∣<=1000

要求：时间复杂度 O(nlogn)

题目解法：

#include "stdio.h"
#include "stdlib.h"


/*
 *BM5 合并k个已排序的链表
描述
    合并 k 个升序的链表并将结果作为一个升序的链表返回其头节点。

数据范围：节点总数  0≤n≤5000，每个节点的val满足 |val| <= 1000
要求：时间复杂度 O(nlogn)O(nlogn)

如输入[{1,2,3},{4,5,6,7}]时，返回{1,2,3,4,5,6,7}，
*/

/**************************code*****************************************/
 struct ListNode {
	int val;
	struct ListNode *next;
 };

/**
 * 
 * @param lists ListNode类一维数组 
 * @param listsLen int lists数组长度
 * @return ListNode类
 */
//根据索引获得链表的元素
int getListElemByIndex(struct ListNode* list, int index){
    for(int i=0; i<index; i++)
        list = list->next;
    return list->val;
}

//计算数组指定长度的前N项和
int getSumLen(int *eachLen, int listsLen){
    int sumLen = 0;
    for(int i=0; i<listsLen; i++){
        sumLen += eachLen[i];
    }
    return sumLen;
}
//找到链表首部最小的值，并返回链表索引
int findListHeadMin(struct ListNode** lists, int listsLen, int* eachIndex, int* eachLen){
    int listIndex = 0;
    int listNodeMin = 1001;
    int tmpVal;
    struct ListNode* tmpList;
    for(int i=0; i<listsLen; i++){
        if(eachIndex[i] >= eachLen[i])continue; //如果该链表已经用完，跳过
        
        tmpVal = getListElemByIndex(lists[i], eachIndex[i]);
        if(tmpVal < listNodeMin){
            listNodeMin = tmpVal;
            listIndex = i;
        }
    }
    return listIndex;
}

struct ListNode* mergeKLists(struct ListNode** lists, int listsLen ) {
    // write code here
    struct ListNode* dummyHead = (struct ListNode*)malloc(sizeof(struct ListNode));
    struct ListNode* tmpNode = dummyHead;
    int *eachLen = (int*)malloc(sizeof(int)*listsLen);  //保存各个链表的长度
    int *eachIndex = (int*)malloc(sizeof(int)*listsLen);  //保存各个链表的索引
    int sumLen = 0;//各个链表相加的总长度
    int listMinIndex; //各个链表中当前节点值最小的链表索引

    for(int i=0; i<listsLen; i++){      //获得各个链表的长度
        int len = 0;
        eachLen[i] = 0;
        eachIndex[i] = 0;           //清空索引
        struct ListNode* tmp = *(lists + i);
        while(tmp){
            eachLen[i]++;
            tmp = tmp->next;
        }
    }
    sumLen = getSumLen(eachLen, listsLen);  //计算各个链表的元素总和
    if(sumLen == 0) return NULL;            //如果 0个链表 或者 有多个链表但每个链表都为空，则返回NULL
    for(int curLen=0; curLen<sumLen; curLen++){ //根据元素总和进行循环
        tmpNode->val = 0;               //先初始化当前链表值

        listMinIndex = findListHeadMin(lists, listsLen, eachIndex, eachLen);    //找到各个链表中未使用的首元素最小的链表索引
        tmpNode->val = getListElemByIndex(lists[listMinIndex], eachIndex[listMinIndex]);    //根据链表以及链表索引找到未使用的元素
        eachIndex[listMinIndex]++;                      //记录当前链表已使用的元素数量，方便findListHeadMin函数判断未使用的元素以及是否到末尾了

        if(curLen==sumLen-1)continue;           //如果已经到最后一个元素，那么不再申请节点
        tmpNode->next = (struct ListNode*)malloc(sizeof(struct ListNode));    //初始化新节点
        tmpNode = tmpNode->next;    //指向下一个节点
    }

    return dummyHead;
}

/**************************end******************************************/

int main ()
{
    struct ListNode LN13={3,NULL};
    struct ListNode LN12={2,&LN13};
    struct ListNode LN1 ={1,&LN12};

    struct ListNode LN24={7,NULL};
    struct ListNode LN23={6,&LN24};
    struct ListNode LN22={5,&LN23};
    struct ListNode LN2 ={4,&LN22};
    struct ListNode *input1[2] = {&LN1, &LN2};
    
    struct ListNode LN33={3,NULL};
    struct ListNode LN32={2,&LN33};
    struct ListNode LN3 ={1,&LN32};

    struct ListNode LN43={3,NULL};
    struct ListNode LN42={3,&LN43};
    struct ListNode LN4 ={3,&LN42};

    struct ListNode *pLN = mergeKLists(input1, 2);

    return 0;
}