Leetcode 61. 旋转链表 C++

Leetcode 61. 旋转链表

题目

给定一个链表，旋转链表，将链表每个节点向右移动 k 个位置，其中 k 是非负数。

测试样例

1.

输入: 1->2->3->4->5->NULL, k = 2
输出: 4->5->1->2->3->NULL
解释:
向右旋转 1 步: 5->1->2->3->4->NULL
向右旋转 2 步: 4->5->1->2->3->NULL

2.

输入: 0->1->2->NULL, k = 4
输出: 2->0->1->NULL
解释:
向右旋转 1 步: 2->0->1->NULL
向右旋转 2 步: 1->2->0->NULL
向右旋转 3 步: 0->1->2->NULL
向右旋转 4 步: 2->0->1->NULL						

题解

旋转是一个循环的过程，因此先求出链表的长度，根据长度判断我们要将后面几个节点挪到前面。然后再根据求出的位置，进行相关操作，具体过程见代码

代码

ListNode* rotateRight(ListNode* head, int k) {
        int length=0;
        ListNode*p = head;
        while(p){
            length++;
            p = p->next;
        }
        if(length == 0) return head;
        int place = length-k%length,i=0;		//根据链表的长度，旋转步数求出需要移动的节点的位置
        if(place == length) return head;
        p = head;
        ListNode *newhead;
        while(p){
            i++;
            if(i == place){
                newhead=p->next;			//新的头节点
                p->next=NULL;
                p=newhead;
                while(p){
                    if(p->next==NULL){		//找到最后一个节点，让其指向原头节点
                        p->next=head;
                        break;
                    }
                    p=p->next;
                }
                break;
            }
            p = p->next;
        }
        return newhead;
    }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/rotate-list
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。