【题目】
给定一个单向链表的头节点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
【要求】
- 如果链表长度为N,时间复杂度要求为O(N),额外空间复杂度为O(1)。
- 如果不满足1 <= from <= to <= N,则不用调整。
【解答】
- 先判断是否满足1 <= from <= to <= N,如果不满足,则直接返回原来的头节点。
- 找到第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。
- 如果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名企算法与数据结构题目最优解)》左程云