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

「这是我参与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

解题思路如下:

  1. 定义 next 指针初始化指向 head
  2. next 指针向后走 k-1 步,找到正数第 k 个节点
  3. next 指针继续向后走一步,一共从 head 节点位置走了 k
  4. head 指针和 next 指针一起向后走,直到 next 指针指向 null,此时 head 指针指向倒数第 k 个节点
  5. 交换正数第 k 个节点和倒数第 k 个节点的值
  6. 返回头节点

代码如下:

var swapNodes = function(head, k) {
  let pre = null,
  cur = head;
  const vhead = new ListNode(0);
  vhead.next = head;
  while(k){
    // 正数第k个节点
    if(k===1) pre = cur;
    cur = cur.next,k--;
  }
  while(cur){
    cur = cur.next;
    head = head.next;
  }
  // 倒数第k个节点
  const headVal = head.val;
  head.val = pre.val;
  pre.val = headVal;
  
  return vhead.next;
};
复制代码

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

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

猜你喜欢

转载自juejin.im/post/7031516322924068894