快慢指针解决链表定位问题(go语言代码)

快慢指针解决链表定位

在做到leetcode中链表的问题时,发现一类需要定位链表某个位置,然后做操作的问题

19. 删除链表的倒数第 N 个结点

题目:给你一个链表,删除链表的倒数第 n 个结点,并且返回链表的头结点。

做到这个题目时 ,我采用了快慢指针,接下来我就详细说一下快慢指针
顾名思义,快慢指针就是一个指针先走,一个指针后走,一个循环,复杂度是O(n)

 //定义了两个链表指针 l r ,分别是 快,慢 指针;n是快慢指针差的步数
 for {
      if l.Next != nil {
          temp = l.Next
          l = temp
          len++ //计算长度,实际上也是快指针走的次数
      }
      if len > n {
          // 当快指针的步数大于慢指针,慢指针开始动
          temp = l2.Next
          l2 = temp
      }
      if l.Next == nil {
          break
          //快指针结束 就可以停止循环,此时慢指针到最后一个节点的距离就是n了
      }
  }

于是回到上面的题,其实删除第n个节点,就是通过快慢指针,把距离最后一个结点为n的节点删掉(按照我的写法,慢指针的下一个就是要删除的)
因此我的完整做法做法贴在下面

/**
 * Definition for singly-linked list.
 * type ListNode struct {
 *     Val int
 *     Next *ListNode
 * }
 */
func removeNthFromEnd(head *ListNode, n int) *ListNode {
    l := new(ListNode)
    l2 := new(ListNode)
    temp := new(ListNode)
    l = head
    l2 = head
    var len int = 0
    for {
        if l.Next != nil {
            temp = l.Next
            l = temp
            len++
        }
        if len > n {
            temp = l2.Next
            l2 = temp
        }
        if l.Next == nil {
            break
        }
    }
    if(len==0){
        return nil
    }
    
    if(n == len + 1){
        head = l2.Next
        l2.Next = nil
        return head
    }
    temp = l2.Next
    l2.Next = temp.Next
    return head
}
posted @ 2022-04-12 20:54  JiaoMaster  阅读(91)  评论(1编辑  收藏  举报