题解 | 调整牛群顺序
主要思路:指针
使用指针指向需要放在末尾的节点和前一节点,以及链表的末端,方便操作,具体如下图所示:
当然这只是其中一种情况,对于需要移动的节点,根据所在位置or链表长度需要分类讨论:
- 如果链表为空,直接返回NULL;
- 如果链表长度只有1,即只有一个节点,则直接返回原链表头head;
- 如果需要移动的节点就是末节点,即n == 1,也只需要直接返回原链表头head;
- 需要移动的节点是中间节点,就按照上述操作对链表指向进行更改,具体做法是:
pre节点指向cur的后一节点;
end节点指向cur节点;
cur节点指向NULL;
5. 如果需要移动的节点是头节点,那么head指针后移一位,之后操作和普通情况相同
具体C++代码如下:
/**
* struct ListNode {
* int val;
* struct ListNode *next;
* ListNode(int x) : val(x), next(nullptr) {}
* };
*/
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param head ListNode类
* @param n int整型
* @return ListNode类
*/
ListNode* moveNthToEnd(ListNode* head, int n) {
// write code here
int len = 0;
ListNode* cur = head;
ListNode* end = NULL;
while (cur != NULL) {
len++;
end = cur;
cur = cur->next;
}
if (len == 0) return nullptr;
if (len == 1 || n == 1) return head;
cur = head;
ListNode* pre = NULL;
for (int i = 0; i < len - n; i++) {
pre = cur;
cur = cur->next;
}
if (cur == head) {
head = cur->next;
}else {
pre->next = cur->next;
}
end->next = cur;
cur->next = NULL;
return head;
}
};
