链表之分隔链表

题目:

给你一个链表的头节点 head 和一个特定值 x ,请你对链表进行分隔,使得所有 小于 x 的节点都出现在 大于或等于 x 的节点之前。
你应当 保留 两个分区中每个节点的初始相对位置。
链表之分隔链表

解析:
直观来说我们只需维护两个链表small 和 large 即可,small 链表按顺序存储所有小于 x 的节点,large 链表按顺序存储所有大于等于 x 的节点。遍历完原链表后,我们只要将small 链表尾节点指向 large 链表的头节点即能完成对链表的分隔。

为了实现上述思路,我们设 smallHead 和 largeHead 分别为两个链表的哑节点,即它们的 next 指针指向链表的头节点,这样做的目的是为了更方便地处理头节点为空的边界条件。同时设 small 和 large 节点指向当前链表的末尾节点。开始时smallHead=small,largeHead=large。随后,从前往后遍历链表,判断当前链表的节点值是否小于 x,如果小于就将small 的next 指针指向该节点,否则将 large 的 next 指针指向该节点。

遍历结束后,我们将large 的next 指针置空,这是因为当前节点复用的是原链表的节点,而其 next 指针可能指向一个小于 x 的节点,我们需要切断这个引用。同时将small 的next 指针指向 largeHead 的 next 指针指向的节点,即真正意义上的large 链表的头节点。最后返回smallHead 的next 指针即为我们要求的答案。

链表之分隔链表

链表之分隔链表

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode() {}
 *     ListNode(int val) { this.val = val; }
 *     ListNode(int val, ListNode next) { this.val = val; this.next = next; }
 * }
 */
class Solution {
    public ListNode partition(ListNode head, int x) {
         ListNode small = new ListNode(0);
         ListNode smallhead = small;
         ListNode large = new ListNode(0);
         ListNode largehead = large;

         while(head!=null){
             if(head.val<x){
                 small.next=head;
                 small=small.next;
             }else{
                 large.next=head;
                 large=large.next;
             }
             head=head.next;
         }
         large.next=null;
         small.next=largehead.next;
         return   smallhead.next;
    }
}

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/partition-list
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

上一篇:21.03.11 LeetCode295. 数据流的中位数


下一篇:推荐:腾讯开源的词向量精简版本下载