链表02--链表中倒数第K个节点-jz14

题目概述

  • 算法说明
    输入一个链表,输出该链表中倒数第k个结点。
  • 测试用例
    输入:
    {1,2,3,4,5},1
    输出:
    {5}

解析&参考答案

  • 解析
    方法1: 遍历一遍,获取链表长度,然后再遍历一遍获取下标为n-k+1的节点即可;
    方法2: 使用2个指针先后遍历,第一个遍历k-1个节点后开始第二个遍历,然后两个指针同时遍历,当第一个遍历完成后,第二个刚好是倒数第k个节点。
  • 参考答案
vim jz14.go
package main

import "fmt"

type ListNode struct {
Val int
Next *ListNode
}

func CreateList(array []int) *ListNode {
if len(array) == 0 {
return nil
}
head := &ListNode{array[0], nil}
p := head
for i, val := range array {
if i == 0 {
} else {
item := &ListNode{val, nil}
p.Next = item
p = p.Next
}
}
return head
}

func FindKthToTail(pHead *ListNode, k int) *ListNode {

if pHead == nil || k == 0 {
return nil
}
p1 := pHead
p2 := pHead
i := 0
for i = 0; i < k-1; i++ {
if p1 == nil {
break
}
p1 = p1.Next
}
if (i+2) < k || (p1 == nil) {
return nil
}
for p1.Next != nil {
p1 = p1.Next
p2 = p2.Next
}
return p2
}

func main() {
array := []int{1, 2, 3, 4, 5}
k := 1
head := CreateList(array)
p2 := FindKthToTail(head, k)
if p2 != nil {
fmt.Print(p2.Val)
} else {
fmt.Printf("K=%d is zero or too big", k)
}
}

注意事项

  1. 需要判断k为0 和 K 数组总长度的情况,这2种情况下直接设置为 nil

说明

  1. 当前使用 go1.15.8
  2. 参考​​牛客网--剑指offer​​ 标题中jzn(n为具体数字)代表牛客网剑指offer系列第n号题目,例如 jz01 代表牛客网剑指offer中01号题目。

注意!!!

  • 笔者最近在学习 golang,因此趁机通过数据结构和算法来进一步熟悉下go语言
  • 当前算法主要来源于剑指 offer,后续会进一步补充 LeetCode 上重要算法,以及一些经典算法
  • 此处答案仅为参考,不一定是最优解,欢迎感兴趣的读者在评论区提供更优解