力扣 61. 旋转链表 链表+模拟

https://leetcode-cn.com/problems/rotate-list/
力扣 61. 旋转链表 链表+模拟_第1张图片

思路:其实就是把链表循环右移 k k k步,首先计算出链表长度 n n n,令 k % = n k\%=n k%=n,然后找到倒数第 k k k个链表作为头节点即可,那么需要把链表从 k − 1 k-1 k1处断开,并把末尾连接到头部。注意特判 k k k 0 0 0的情况。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* rotateRight(ListNode* head, int k) {
        if(!head)
            return head;
        int n=0;
        ListNode *cur=head;
        while(cur&&++n)
            cur=cur->next;
        //循环向右位移k步
        k%=n;
        if(!k)
            return head;
        ListNode *pre=head,*nxt=head;
        while(k--)
            nxt=nxt->next;
        while(nxt->next)
            pre=pre->next,nxt=nxt->next;
        //pre->next 为头节点即可
        ListNode *tmp=pre->next;
        pre->next=nullptr;
        nxt->next=head;
        return tmp;
    }
};

你可能感兴趣的:(面试题,链表,模拟)