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

思想:截断+转移

代码:

struct ListNode* rotateRight(struct ListNode* head, int k){
    if (head == NULL) return head; 
    struct ListNode *newhead = head, *tail = head, *temp = head;
        
    int len = 0;
    while (temp != NULL) {
        if (temp->next == NULL)
            tail = temp;
        len++;
        temp = temp->next;
    }
        
    int move_len = len - k%len;
    if (move_len == len) return head;
    for (int i = 0; i < move_len-1; i++)
        newhead = newhead->next;
    temp = newhead;
    newhead = newhead->next;
    temp->next = NULL;
    tail->next = head;

    return newhead;
}
发布了42 篇原创文章 · 获赞 18 · 访问量 7536

猜你喜欢

转载自blog.csdn.net/weixin_44395686/article/details/105166723
今日推荐