[路飞]_leetcode-1721-交换链表中的节点

384 阅读2分钟

「这是我参与11月更文挑战的第13天,活动详情查看:2021最后一次更文挑战

[题目地址] [B站地址]

给你链表的头节点 head 和一个整数 k 。

交换 链表正数第 k 个节点和倒数第 k 个节点的值后,返回链表的头节点(链表 从 1 开始索引)。

示例 1:

输入: head = [1,2,3,4,5], k = 2
输出: [1,4,3,2,5]

示例 2:

输入: head = [7,9,6,6,7,8,3,0,9,5], k = 5
输出: [7,9,6,6,8,7,3,0,9,5]

示例 3:

输入: head = [1], k = 1
输出: [1]

示例 4:

输入: head = [1,2], k = 1
输出: [2,1]

示例 5:

输入: head = [1,2,3], k = 2
输出: [1,2,3]

提示:

  • 链表中节点的数目是 n
  • 1 <= k <= n <= 105
  • 0 <= Node.val <= 100

本题其实很简单,只需要两步操作

首先找到正数第 k 个节点以及倒数 k 个节点

然后交换两个节点的值即可

解题思路如下:

  1. 定义 prenext 、 end 指针初始化指向 head
  2. end 指针向后走 k-1 步,找到正数第 k 个节点,此时让 pre 指针记录该位置
  3. end 指针继续向后走一步,一共从 head 节点位置走了 k 步,此时 end 指针和 next 指针距离为 k
  4. next 指针和 end 指针一起向后走,直到 end 指针指向 null,此时 next 指针指向倒数第 k 个节点
  5. 交换正数第 k 个节点(pre 指针所在位置)和倒数第 k 个节点(next 指针所在位置)的值
  6. 最后返回头节点即可返回头节点

代码如下:

var swapNodes = function(head, k) {
  // 定义三个指针初始化指向 head
  let pre = next = end = head;
  while(k){
    // 此时 pre 指向正数第 `k`个节点
    if(k===1) pre = end;
    end = end.next;
    k--;
  }
  // 此时next 和 end 的距离为 k
  // 当 end 不为空时,两个指针一起向后走
  // end指向空的时候,next指向倒数第 k 个节点
  while(end){
    next = next.next;
    end = end.next;
  }
  // 交换pre 和 next 指针的值
  const nextVal = next.val;
  next.val = pre.val;
  pre.val = nextVal;
  // 返回头节点
  return head;
};

至此我们就完成了 leetcode-1721-交换链表中的节点

如有任何问题或建议,欢迎留言讨论!