LeetCode 61. 旋转链表

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

先通过循环遍历,计算出链表的长度num,然后用k%num,因为如果k%num=0,我们就不需要去旋转链表,如果k>num我们只需要去旋转取余次,定义一个指针变量temp,使tem->next=头节点,然后开始循环找到倒数第二个节点也就是p1和最后一个节点p2,把倒数第二个节点p1->next赋值为null,把最后一个节点p2->next赋值给第一个节点temp->next,最后把temp指向p2,这样就OK了。

示例 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 {
     int val;
     struct ListNode *next;
};

struct ListNode* rotateRight(struct ListNode* head, int k){
    if((k == 0) || (head == NULL)) {
        return head;
    }
    struct ListNode temp, *p1, *p2, *p3;
    int num = 0;
    p3 = head;
    while(p3 != NULL) {
        p3 = p3->next;
        num++;
    }
    k %= num;
    temp.next = head;
    while(k--) {
        p1 = &temp;
        p2 = p1->next;
        while(p2->next != NULL) {
            p1 = p1->next;
            p2 = p1->next;
        }
        p1->next = NULL;
        p2->next = temp.next;
        temp.next = p2;
    }
    return temp.next;
}

你可能感兴趣的:(LeetCode 61. 旋转链表)