思路
1、计算链表总长度,并把链表变为循环链表
2、计算旋转后链表头节点的前一个节点,将p移至此处
3、将p->next保存ret=p->next,再将p->next=null,断开循环链表,返回ret
举例
输入:head = [1,2,3,4,5], k = 2 输出:[4,5,1,2,3] 1、链表长度为num_node=5,链表变为循环链表,p指向5(原链表的尾部)
2、计算旋转后链表头节点的前一个节点的位置 num_node - k%num_node=5-2=3,旋转后链表头节点的前一个节点的位置是在第3个节点,将p移至此处
3、使得ret指向p->next,ret就指向val=4的节点,断开循环链表p->next=null,返回ret
关键: 只要理解了第2步计算的意义,此题也就迎刃而解了
代码
方法1
struct ListNode* rotateRight(struct ListNode* head, int k){
if(head==NULL) return NULL;
struct ListNode *p=head,*ret;
int num_node=1;
while(p->next!=NULL){
p=p->next;
num_node++;
}
p->next=head;
num_node=num_node-k%num_node;
while(num_node>0){
p=p->next;
num_node--;
}
ret=p->next;
p->next=NULL;
return ret;
}
方法2(省去方法1中的ret,直接返回head)
struct ListNode* rotateRight(struct ListNode* head, int k){
if(head==NULL) return NULL;
struct ListNode *p=head;
int num_node=1;
while(p->next!=NULL){
p=p->next;
num_node++;
}
p->next=head;
num_node=num_node-k%num_node;
while(num_node>0){
p=p->next;
num_node--;
}
head=p->next;
p->next=NULL;
return head;
}
|