如何轻松用JavaScript在LeetCode上解决K个一组翻转链表？

2023-12-19 01:51:40

前言

欢迎来到LeetCode之旅！今天，我们将共同征服一道精彩的算法题：“K个一组翻转链表”。这道题不仅考验你的编程能力，更需要你对数据结构和算法的深刻理解。

题目解析

给你一个链表，每k个节点一组进行翻转，请你返回翻转后的链表。k是一个正整数，它的值小于或等于链表的长度。如果节点总数不是k的倍数，那么最后剩下的节点保持原样。

JavaScript实现

/**
 * Definition for singly-linked list.
 * function ListNode(val) {
 *     this.val = val;
 *     this.next = null;
 * }
 */

/**
 * @param {ListNode} head
 * @param {number} k
 * @return {ListNode}
 */
const reverseKGroup = (head, k) => {
  if (head === null || k === 1) {
    return head;
  }

  let dummy = new ListNode(0);
  dummy.next = head;
  let prev = dummy;
  let curr = head;
  let next;

  while (curr !== null) {
    let count = 0;
    while (curr !== null && count < k) {
      count++;
      next = curr.next;
      curr.next = prev;
      prev = curr;
      curr = next;
    }

    if (count === k) {
      head = prev;
    }

    prev = curr;
    curr = next;
  }

  return head;
};