将链表后k个节点移到链表头
程序员文章站
2024-03-21 20:03:28
...
- 问题描述
给一个链表和整数k,将后k个节点移到链表头。
For example:
Given1->2->3->4->5->NULL , k =2,
return4->5->1->2->3->NULL. - 解题思路
1.将单链表变成循环链表
先遍历一遍,得出链表节点个数count,将尾结点next指针指向首节点,形成一个环
2.重新确定链表头和链表尾。
接着往后跑lcount-k%count(k可能大于链表节点数)步,断开,返回结果。 - 代码实现
class Solution {
public:
ListNode *rotateRight(ListNode *head, int k) {
if(NULL == head || k == 0)
return head;
//统计链表节点数,并将其变成循环链表
int count = 1;
ListNode *_head = head;
while(NULL != _head -> next)
{
count++;
_head = _head -> next;
}
_head -> next = head;//首尾相连
//确定新链表头,并断开循环链表
k = count - k % count;//k可能大于链表节点数,令k=k%count
while(k--)
{
_head = _head -> next;
}
head = _head -> next;//新链表头
_head -> next = NULL;//断开
return head;
}
};
上一篇: 使用一个二维数组打印杨辉三角
下一篇: 用Python写一个杨辉三角
推荐阅读