链表分割

最新推荐文章于 2024-07-25 11:23:22 发布

原创最新推荐文章于 2024-07-25 11:23:22 发布 · 310 阅读

0 ·

CC 4.0 BY-SA版权

数据结构专栏收录该内容

55 篇文章

订阅专栏

本文介绍了一种链表分割算法，该算法将链表中小于给定值x的所有节点排在大于或等于x的节点之前，同时保持原有节点顺序不变。通过实现Partition类中的partition方法，实现了这一功能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

编写代码，以给定值x为基准将链表分割成两部分，所有小于x的结点排在大于或等于x的结点之前

给定一个链表的头指针 ListNode* pHead，请返回重新排列后的链表的头指针。注意：分割以后保持原来的数据顺序不变。

/*struct ListNode 
{
	int val;    
	struct ListNode *next;    
	ListNode(int x):val(x), next(NULL) {    }
};*/
class Partition{
public:    
	ListNode* partition(ListNode* pHead, int x) 
	{        
		// write code here        
		ListNode* min = NULL;        
		ListNode* max = NULL;        
		ListNode* lastmin = NULL;        
		ListNode* lastmax = NULL ;        
		ListNode* p = pHead ;        
		while(p)            
		{                       
			if(p->val<x)                
			{                 
				if(min == NULL)                     
				{                     
					min = p;                     
					lastmin = min;                     
					p = p->next;                     
					lastmin->next = NULL;                 
				}                
				else                    
				{                    
					lastmin->next = p;                    
					p = p->next;                    
					lastmin = lastmin->next;                    
					lastmin->next = NULL;                                   
				}                                       
			}            
			else                
			{                
				if(max == NULL)                    
				{                    
					max = p;                    
					lastmax = max;                    
					p = p->next;                    
					lastmax->next = NULL;                
				}                
				else                    
				{                    
					lastmax->next = p;                    
					p = p->next;                    
					lastmax = lastmax->next;                    
					lastmax->next = NULL;                
				}            
			}                  
		}        
		if(lastmin)            
		{            
			lastmin->next = max;            
			return min;        
		}        
		return pHead;    
	}
};