链表问题05：反转部分单向链表

【题目】

给定一个单向链表的头节点head，以及两个整数from和to，在单向链表上把第from个节点到第to个节点这一部分进行反转。

例如：

1 -> 2 -> 3 -> 4 -> 5 -> null，from = 2，to = 4

调整结果为：1 -> 4 - > 3 -> 2 -> 5 -> null

再如：

1 -> 2 -> 3 -> null，from = 1，to = 3

调整结果为：3 -> 2 -> 1 -> null

 

【要求】

1. 如果链表长度为N，时间复杂度要求为O(N)，额外空间复杂度为O(1)。
2. 如果不满足1 <= from <= to <= N，则不用调整。

 

【解答】

1. 先判断是否满足1 <= from <= to <= N，如果不满足，则直接返回原来的头节点。
2. 找到第from-1个节点fPre和第to+1个节点tPos。fPre就是要反转部分的前一个节点，tPos是反转部分的后一个节点。把反转的部分先反转，然后正确地连接fPre和tPos。例如：1 -> 2 -> 3 -> 4 -> null，假设fPre为节点1，tPos为节点4，要反转部分为2 -> 3。先反转成3 -> 2，然后fPre连向节点3，节点2连向tPos，就变成1 -> 3 -> 2 -> 4 -> null。
3. 如果fPre为null，说明反转部分是包含头节点的，则返回新的头节点，也就是没反转之前反转部分的最后一个节点，也是反转之后反转部分的第一个节点；如果fPre不为null，则返回旧的头节点。

 

【代码实现】

struct  Node
{
	int value;
	Node *next;
	Node(int data){
		value = data;
	}
};


static Node *reversePartList(Node *head, int from, int to){
	int len = 0;
	Node *node1 = head;
	Node *fPre = nullptr;
	Node *tPos = nullptr;
	while (node1 != nullptr)
	{
		len++;
		fPre = len == from - 1 ? node1 : fPre;
		tPos = len == to + 1 ? node1 : tPos;
		node1 = node1->next;
	}
	if (from > to || from < 1 || to > len){
		return head;
	}
	node1 = fPre == nullptr ? head : fPre->next;
	Node *node2 = node1->next;
	node1->next = tPos;
	Node *next = nullptr;
	while (node2 != tPos)
	{
		next = node2->next;
		node2->next = node1;
		node1 = node2;
		node2 = next;
	}
	if (fPre != nullptr){
		fPre->next = node1;
		return head;
	}
	return node1;
}

 

【来源】

《程序员代码面试指南(IT名企算法与数据结构题目最优解)》左程云