day5--Os nós na lista encadeada são invertidos a cada k grupos

Primeiro, crie um dummynó de cabeçalho virtual chamado , e aponte seu nextponteiro para o nó de cabeçalho da lista encadeada original.

Em seguida, usamos três ponteiros pre, cur, nexpara realizar a operação de flip:

Primeiro, calcule o comprimento da lista encadeada e, em seguida, inverta k nós a cada vez até que restem menos de k nós na lista encadeada.

Durante o processo de inversão, um loop é usado para trocar dois nós adjacentes até que a inversão seja concluída.

Por fim, retornamos o ponteiro do nó de cabeça virtual next, que é o ponteiro de cabeça da lista encadeada após a inversão.

#include <iostream>
using namespace std;

struct ListNode {
    int val;
    ListNode *next;
    ListNode(int x) : val(x), next(NULL) {}
};
class Solution {
public:
    ListNode* reverseKGroup(ListNode* head, int k) {
        if (head == NULL || k == 1) {
            return head;
        }
        ListNode *dummy = new ListNode(0);
        dummy->next = head;
        ListNode *pre = dummy, *cur = head, *nex = NULL;
        int len = 0;
        while (head != NULL) {
            len++;
            head = head->next;
        }
        while (len >= k) {
            cur = pre->next;
            nex = cur->next;
            for (int i = 1; i < k; i++) {
                cur->next = nex->next;
                nex->next = pre->next;
                pre->next = nex;
                nex = cur->next;
            }
            pre = cur;
            len -= k;
        }
        return dummy->next;
    }
};
int main() {
    ListNode *head = new ListNode(1);
    head->next = new ListNode(2);
    head->next->next = new ListNode(3);
    head->next->next->next = new ListNode(4);
    head->next->next->next->next = new ListNode(5);
    Solution solution;
    ListNode *result = solution.reverseKGroup(head, 2);
    while (result != NULL) {
        cout << result->val << " ";
        result = result->next;
    }
    return 0;
}

Acho que você gosta

Origin blog.csdn.net/qq_54809548/article/details/130850388
Recomendado
Clasificación