61. 旋转链表

题目描述

点这里

思路分析

链表模拟题。
思路在注释中有体现。
大概就是找到新的头节点。
链表模拟题都是这样。考察指针的改变，遍历等等。

代码实现

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* rotateRight(ListNode* head, int k) {
        if(!head)  return head; //判空
        auto dummy=new ListNode(-1); //虚拟头节点
        dummy->next=head;
        int cnt=0; //总节点数量
        auto p=head;
        while(p) cnt++,p=p->next;
        k%=cnt; //k对节点数量取模，因为多旋转cnt次=不旋转
        if(!k) return head; //对K判空
        p=head;  //求新的头节点
        for(int i=1;i<=cnt-k;i++) p=p->next;
        auto newhead=p;
        while(p->next) p=p->next; //把最后的尾节点指向原头节点
        p->next=dummy->next;
        dummy->next=newhead; //虚拟头节点指向新的头节点
        p=head;
        while(p->next!=dummy->next) p=p->next; //找到新的尾节点，指向NULL
        p->next=NULL;
        return dummy->next; //返回虚拟头节点的指针
    }
};