258 字 1 分钟阅读

题目

给定一个奇数位升序,偶数位降序的单链表,编写一个函数,将其按照升序进行排序

注:假定链表第一个元素编号为1、第二个元素编号为2,依次类推;“奇数位”指的是编号为奇数的元素。

示例1:

给定链表 1->8->3->6->5->4->7->2->NULL, 
重新排列为 1->2->3->4->5->6->7->8->NULL

解法

先把链表按照奇数位和偶数位进行拆分,然后对偶数位的链表进行翻转得到两个升序链表。最后对两个升序链表进行归并排序即可。

public class Solution {

    public ListNode sortList(ListNode head) {
        if (head == null || head.next == null) {
            return head;
        }

        // 将链表按奇数位和偶数位进行拆分
        ListNode head2 = head.next;
        ListNode tmp1 = head;
        ListNode tmp2 = head2;
        while (tmp1 != null || tmp2 != null) {
            if (tmp1 != null) {
                tmp1.next = tmp1.next == null ? null : tmp1.next.next;
                tmp1 = tmp1.next;
            }
            if (tmp2 != null) {
                tmp2.next = tmp2.next == null ? null : tmp2.next.next;
                tmp2 = tmp2.next;
            }
        }
        // 翻转偶数位链表
        head2 = reverse(head2);
        // 合并两个链表
        return merge(head, head2);
    }

    /**
     * 翻转链表
     *
     * @param head
     * @return
     */
    public static ListNode reverse(ListNode head) {
        ListNode pre = head;
        ListNode cur = head.next;
        ListNode next;
        pre.next = null;
        while (cur != null) {
            next = cur.next;
            cur.next = pre;
            pre = cur;
            cur = next;
        }
        return pre;
    }

    /**
     * 升序合并链表
     *
     * @param head1
     * @param head2
     * @return
     */
    public static ListNode merge(ListNode head1, ListNode head2) {
        if (head1.val > head2.val) {
            ListNode tmp = head1;
            head1 = head2;
            head2 = tmp;
        }
        ListNode head = head1;
        ListNode cur = head1;
        head1 = head1.next;
        // 升序合并
        while (head1 != null && head2 != null) {
            if (head1.val < head2.val) {
                cur.next = head1;
                head1 = head1.next;
            } else {
                cur.next = head2;
                head2 = head2.next;
            }
            cur = cur.next;
        }
        if (head1 != null) {
            cur.next = head1;
        }
        if (head2 != null) {
            cur.next = head2;
        }
        return head;
    }
}