leetcode#61 旋转链表

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

示例 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


/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {//题目不太清晰,【1,2】可以右旋转3?
public:
    ListNode* rotateRight(ListNode* head, int k) {
        //首先要找到倒数第k+1个节点,然后移到头部。如果k比链表长,那么求模
        if(!head) 
            return nullptr;
        int n = 0;
        ListNode *cur = head;
        while (cur) {
            ++n;
            cur = cur->next;
        }
        k %= n;//取模
        
        
        ListNode* current=head;
        while(k>0&&current)//让current成为快指针,比head先k
        {
            current=current->next;
            k--;
        }
        if(k!=0||current==nullptr) return head;
        ListNode* newHeader=head;
        while(current->next)
        {
            current=current->next;
            newHeader=newHeader->next;
        }
        current->next=head;
        ListNode* temp=newHeader->next;
        newHeader->next=nullptr;
        return temp;
    }
};

猜你喜欢

转载自www.cnblogs.com/lsaejn/p/9758223.html