链表02--链表中倒数第K个节点
原创
©著作权归作者所有:来自51CTO博客作者昕光xg的原创作品,请联系作者获取转载授权,否则将追究法律责任
链表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)
}
}
注意事项
- 需要判断k为0 和 K 数组总长度的情况,这2种情况下直接设置为 nil
说明
- 当前使用 go1.15.8
- 参考牛客网--剑指offer 标题中jzn(n为具体数字)代表牛客网剑指offer系列第n号题目,例如 jz01 代表牛客网剑指offer中01号题目。
注意!!!
- 笔者最近在学习 golang,因此趁机通过数据结构和算法来进一步熟悉下go语言
- 当前算法主要来源于剑指 offer,后续会进一步补充 LeetCode 上重要算法,以及一些经典算法
- 此处答案仅为参考,不一定是最优解,欢迎感兴趣的读者在评论区提供更优解