LeetCode 23:Merge k Sorted Lists 解题与思考

LeetCode 23:Merge k Sorted Lists 解题与思考

[原题链接]

题目描述

将若干个有序数列合并成一个，简单点说就是归并排序的第二步

思路

很显然，只需要对所有数列的头部进行遍历，每遍历一次取出最值就好。

当然，作为通过率较低的hard题目，其需要考虑的情况也不少。

我最先想到的是需要判断数列顺序的升降序，结果没能ac，因为leetcode还特别喜欢测试空数组的情况。不过只要好好考虑了这些情况，剩下的就很好办了。

算法

1、先判断数列空的情况
2、倘若数列不为空，则寻找至少两个或以上数字的数列，判断升降序
3、依次遍历每个数组的头部，找到最大（小）值输出。

代码

#include <iostream>
#include <vector>

using namespace std;

class Solution {
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        ListNode* result = NULL, *current;
        if ( lists.size() == 0 )return result;
        int nowVal, nowPos;
        bool decending;
        for ( auto i : lists ) {
            if ( i != NULL && i->next != NULL ) {
                if ( i->val > i->next->val ) {
                    decending = true;
                    break;
                }
                if ( i->val < i->next->val ) {
                    decending = false;
                    break;
                }
            }
        }
        if ( decending ) {
            nowVal = -100000;
            nowPos = -1;
            int pos = 0;
            for ( auto i : lists ) {
                if ( (i != NULL) && (i->val > nowVal) ) {
                    nowVal = i->val;
                    nowPos = pos;
                }
                pos++;
            }
            if ( nowPos == -1 )return result;
            result = lists.at(nowPos);
            lists.at(nowPos) = lists.at(nowPos)->next;
            current = result;
            while ( true ) {
                nowVal = -100000;
                nowPos = -1;
                int pos = 0;
                for ( auto i : lists ) {
                    if ( (i != NULL) && (i->val > nowVal) ) {
                        nowVal = i->val;
                        nowPos = pos;
                    }
                    pos++;
                }
                if ( nowPos == -1 )return result;
                current->next = lists.at(nowPos);
                lists.at(nowPos) = lists.at(nowPos)->next;
                current = current->next;
            }
        }
        else {
            nowVal = 100000;
            nowPos = -1;
            int pos = 0;
            for ( auto i : lists ) {
                if ( (i != NULL) && (i->val < nowVal) ) {
                    nowVal = i->val;
                    nowPos = pos;
                }
                pos++;
            }
            if ( nowPos == -1 )return result;
            result = lists.at(nowPos);
            lists.at(nowPos) = lists.at(nowPos)->next;
            current = result;
            while ( true ) {
                nowVal = 100000;
                nowPos = -1;
                int pos = 0;
                for ( auto i : lists ) {
                    if ( (i != NULL) && (i->val < nowVal) ) {
                        nowVal = i->val;
                        nowPos = pos;
                    }
                    pos++;
                }
                if ( nowPos == -1 )return result;
                current->next = lists.at(nowPos);
                lists.at(nowPos) = lists.at(nowPos)->next;
                current = current->next;
            }
        }
        return result;
    }
};

思考

还是稍微有点懒了，很多代码部分没有做优化，部分地方复制粘贴，其实代码可以更省些的。不过反正题目也不难，所以也不打算花太多时间在上面了。

leetcode真的是喜欢空输入这个极端状况啊，只要有机会就会弄个这个玩意儿卡一下你，有点伤脑筋。