题目描述
给定一个链表,旋转链表,将链表每个节点向右移动 k 个位置,其中 k 是非负数。
解题思路
- leetcode 链表题目中,链表结构体都是没有头结点的。每次我都会自动构造一个头结点。
- 链表题目其实都不难,画画图基本都能够解出来。
本题目使用双指针算法即可。
定义两个指针 first 和 second,首先令 first 指向第 n 个结点。然后初始化 second 指向头结点,而后 first 、second 指针依次向后移动,当 first 指向最后一个结点时,first 就指向倒数第 n + 1 个结点了。此时调整指针指向,便实现了旋转链表。
本题目有个坑就是 k 可能大于元素个数。
实现代码
ListNode* rotateRight(ListNode* head, int k)
{
ListNode* phead = new ListNode(-1);
phead->next = head;
ListNode* first = phead,*second = phead;
int count = 0; // 统计链表元素个数
while(first->next)
{
first = first->next;
count++;
}
if(count == 0)
{
return phead->next;
}
k = k % count; // 如果 k 等于元素个数,旋转后链表等于原链表
first = phead;
while(k--)
{
first = first->next;
}
while(first->next)
{
first = first->next;
second = second->next;
}
// 调整指针位置,有点绕
first->next = phead->next;
phead->next = second->next;
second->next = NULL;
return phead->next;
}