Given a list, rotate the list to the right by k places, where k is non-negative.
For example:
Given 1->2->3->4->5->NULL
and k = 2
,
return 4->5->1->2->3->NULL
.
题意:将链表右移k个位置。
这题不难,但是有几个细节要注意:
1)k=0,这个时候直接返回head.
2)计算链表的长度len,如果k>len,令k=k%len,这个要留意,我做这道题的时候就是忽略了这点,认为k肯定小于len,从而wrong answer了多次。
3)之后定义两个指针cur和post,先让post走k步,然后cur,post一起向前走,当post为NULL的时候,那么这个时候的cur就是头节点head。在此过程中,分别定义temp1,temp2作为cur和post的前节点,用于将链表重新组合,这个在代码的最后面有体现,这里就不阐述了。
class Solution {
public:
ListNode *rotateRight(ListNode *head, int k) {
if(head==NULL)
return NULL;
ListNode *p=head;
int len=1;
while(p->next)
{
len++;
p=p->next;
}
k=k%len;
if(k==0)
return head;
ListNode *cur=head,*post=head;
while(k--&&post!=NULL)
{
post=post->next;
}
if(post==NULL)
return NULL;
ListNode *temp1=NULL,*temp2=NULL;
while(post!=NULL)
{
temp1=cur;
cur=cur->next;
temp2=post;
post=post->next;
}
temp2->next=head;
temp1->next=NULL;
head=cur;
return head;
}
};