[剑指offer] 链表03:链表中倒数第k个结点

题目描述

输入一个链表,输出该链表中倒数第k个结点。

解题思路

经典的双指针法。定义两个指针,第一个指针从链表的头指针开始遍历向前走k-1步,第二个指针保持不动,从第k步开始,第二个指针也开始从链表的头指针开始遍历,由于两个指针的距离保持在k-1,当第一个指针到达链表的尾节点时,第二个指针刚好指向倒数第k个节点。

关注要点
1)链表头指针是否为空,若为空则直接返回回null
2)k是否为0,k为0也就是要查找倒数第0个节点,由于计数一般是从1开始的,所有输入0没有实际意义,返回null
3)k是否超出链表的长度,如果链表的节点个数少于k,则在指针后移的过程中会出现next指向空指针的错误,所以程序中要加一个判断

参考代码(Java)

  • 结构定义
public class ListNode {
  int val;
  ListNode next = null;

  ListNode(int val) {
    this.val = val;
 }
}
  • 解法一代码
public class Demo03 {
  public ListNode FindKthToTail(ListNode head,int k) {
    if(head==null || k==0) {
        return null;
    }
    
    ListNode p1 = head;
    //判断k是否超过链表节点的个数,注意是 i 从1~k-1
    for (int i = 1; i < k; i++) {
        if(p1.next!=null) {
            p1=p1.next;
        }else {
            return null;
        }
    }
    
    ListNode p2 = head;
    
    while(p1.next!=null) {
        p1=p1.next;
        p2=p2.next;
    }

    return p2;
    
  }
}
  • 测试主方法
public static void main(String[] args) {    
    ListNode listNode = new ListNode(1);
    ListNode listNode2 = new ListNode(2);
    ListNode listNode3 = new ListNode(3);
    ListNode listNode4 = new ListNode(4);
    listNode3.next = listNode4;
    listNode2.next = listNode3;
    listNode.next = listNode2;
    ListNode tail = FindKthToTail(listNode,2);
    System.out.println(tail.val);
}
  • 输出结果
 3 
  • 解法二代码(用栈)
public class Demo03 {
  public ListNode FindKthToTail(ListNode head,int k) {
    if (head == null || k == 0) {
        return null;
    }

    // 可以先把链表反转,然后找出第k个
    Stack<ListNode> stack = new Stack<ListNode>();
    int count = 0;
    while (head != null) {
        stack.push(head);
        head = head.next;
        count++;
    }

    if (count < k) {
        return null;
    }

    ListNode knode = null;
    for (int i = 0; i < k; i++) {
        knode = stack.pop();
    }

    return knode;
    
  }
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容