1. 程式人生 > 其它 >【LeetCode】1721 交換連結串列中的節點,快慢指標實現

【LeetCode】1721 交換連結串列中的節點,快慢指標實現

技術標籤:Leetcode刷題總結leetcode連結串列java快慢指標

題目描述

給你連結串列的頭節點 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

來源:力扣(LeetCode)
連結:https://leetcode-cn.com/problems/swapping-nodes-in-a-linked-list

解題思路:

定義一個node結點和快慢指標都指向頭結點,fast快指標一直往前走直到遇見null,當node結點走到正數第k個結點時停止,此時slow慢指標開始出發,最後fast快指標走完連結串列時,slow慢指標所在位置剛好就是倒數第k個結點。

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode() {}
 *     ListNode(int val) { this.val = val; }
 *     ListNode(int val, ListNode next) { this.val = val; this.next = next; }
 * }
 */
class Solution {
    public ListNode swapNodes
(ListNode head, int k) { ListNode node = head; ListNode slow = head; ListNode fast = head; int count = 1; while(fast.next != null) { if(count < k){ node = node.next; }else{ slow = slow.next; } fast = fast.next; count++; } int temp = node.val; node.val = slow.val; slow.val = temp; return head; } }