leetcode 86 分隔链表 / partition list

最新推荐文章于 2024-05-13 06:12:34 发布

原创最新推荐文章于 2024-05-13 06:12:34 发布 · 447 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#leetcode #onlineJudge

刷题专栏收录该内容

26 篇文章

订阅专栏

本文探讨了链表操作中的分隔问题，通过两种不同的实现方式对比，展示了一种更简洁明了的方法。作者分享了自己的代码，并反思了其在代码效率和可读性上的不足。

题目描述：

又是关于链表的操作题，感觉自己做这类题比较缺乏经验，想到哪写到哪，代码原没有别人简洁明了，还需要多多学习。

参考代码：

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* partition(ListNode* head, int x) {
        ListNode dummy1(-1),dummy2(-1);
        ListNode *root1=&dummy1,*root2=&dummy2;
        while(head){
            if(head->val<x){
                root1->next=head;
                head=head->next;
                root1=root1->next;
                root1->next=NULL;
            }
            else{
                root2->next=head;
                head=head->next;
                root2=root2->next;
                root2->next=NULL;
            }
        }
        root1->next=dummy2.next;
        return dummy1.next;
    }
};

我的：

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
	ListNode* partition(ListNode* head, int x) {
		if(head == NULL) return head;
		ListNode* q = head;
		ListNode* temp;
		int total = 1;
		while(q -> next != NULL){
			total += 1;
			q = q -> next;
		} // q记录最后一个节点
		ListNode* p = head;
		int cnt = 0;
		while(p -> val >= x && cnt < total){ // 找到第一个小于x的节点
			temp = p;
			q -> next = temp;
			p = p -> next;
			temp -> next = NULL;
			q = temp;
			cnt += 1;
		}
		if(cnt > total) return p;
		cnt += 1;
		if(cnt == total) return p;
		ListNode* begin = p;
		while(cnt < total){
			if(p -> next != NULL && p -> next -> val < x){
				p = p -> next;
				cnt += 1;
			}
			else if(p -> next != NULL && p -> next -> val >= x){
				temp = p -> next;
				q -> next = temp;
				p -> next = p -> next -> next;
				temp -> next = NULL;
				q = temp;
				cnt += 1;
			}
		}
		return begin;
    }
};