牛客网--链表分割

题目描述
编写代码,以给定值x为基准将链表分割成两部分,所有小于x的结点排在大于或等于x的结点之前
给定一个链表的头指针 ListNode* pHead,请返回重新排列后的链表的头指针。注意:分割以后保持原来的数据顺序不变。

思路:
1.我们可以定义两个链表(结构体类型变量)(比如small,large),分别放小于x的结点、大于或等于x的结点。
2.对原链表进行遍历,与x进行比较大小,对两个新链表进行尾插(所以还要分别定义两个指针哦)
3.遍历完后,结点位置都已经放好了,我们再把两个链表连起来就可以了

这里要用到 . 操作符,它是结构体类型变量访问内部成员的;而 -> 是通过指针访问结构体内部的变量

struct ListNode {
    int val;
    struct ListNode *next;
    ListNode(int x) : val(x), next(NULL) {}
};
class Partition {
public:
	ListNode* partition(ListNode* pHead, int x) {
		//先给两个链表,定义结构体类型的变量
		ListNode small(0);
		ListNode large(0);
		//给两个链表分别定义指针(进行尾插)
		ListNode* largeNode = &large;
		ListNode* smallNode = &small;
		while (pHead)
		{
			if (pHead->val < x)
			{
				smallNode->next = pHead;
				smallNode = pHead;
			}
			else
			{
				largeNode->next = pHead;
				largeNode = pHead;
			}
			pHead = pHead->next;
		}
		largeNode->next = NULL;//要让最后的结点指向空
		smallNode->next = large.next;//将两个链表连接
		return small.next;
	}
};
发布了78 篇原创文章 · 获赞 2 · 访问量 3321

猜你喜欢

转载自blog.csdn.net/weixin_43219708/article/details/104750151