题目:输入一个链表,输出该链表的倒数第K个节点。为了符合大多数人的习惯,本题从1开始计数,即链表尾节点是倒数第一个节点。
解题思路:
解法一:一般情况下,单向链表无法从后一个节点获取到它前面的节点,可以通过两次遍历,第一次遍历获取链表中节点的个数,第二次遍历找到链表中第n-k+1个节点,就是链表的倒数第k个节点。
但是这种方法效率低,可以使用一次遍历得到倒数第K个节点
解法二:一次遍历得到倒数第K个节点。维护两个指针,第一个指针从链表头结点向前走k-1步,第二个节点指向头结点,从第K步开始,如果走在前面的节点有下一个节点,那两个节点一起向前走,
直到第一个节点走到尾节点,此时第二个节点指向倒数第K个节点
程序健壮性考虑:
1.输入链表为null 2.输入k为0或小于0 3.链表节点总数小于k
1 package Solution;
2
3 public class No15FindKthNodeFromEnd {
4
5 public static class ListNode {
6 int data;
7 ListNode next;
8
9 public ListNode(int value, ListNode next) {
10 this.data = value;
11 this.next = next;
12 }
13 }
14
15 public static ListNode findKthNodeFromEnd(ListNode head, int k) {
16 if (head == null)
17 throw new RuntimeException("待查找的链表不能为空");
18 if (k <= 0)
19 throw new RuntimeException("输入的位置数字不合法");
20 ListNode ahead = head;
21 ListNode behind = head;
22 // 第一个指针先指向K-1,并检验链表中节点个数是否大于k
23 int count = 1;
24 for (int i = 0; i < k - 1; i++) {
25 if (ahead.next != null) {
26 ahead = ahead.next;
27 count++;
28 } else
29 throw new RuntimeException("链表节点个数:" + count + " 小于输入K的个数:" + k);
30 }
31 while (ahead.next != null) {
32 ahead = ahead.next;
33 behind = behind.next;
34 }
35 return behind;
36 }
37
38 public static void main(String[] args) {
39 ListNode node1 = new ListNode(4, null);
40 ListNode node2 = new ListNode(3, node1);
41 ListNode node3 = new ListNode(2, node2);
42 ListNode head = new ListNode(1, node3);
43 ListNode find = findKthNodeFromEnd(head, 3);
44 System.out.println("找到的节点位" + find.data);
45 ListNode find2 = findKthNodeFromEnd(head, 5);
46 System.out.println("找到的节点位" + find2.data);
47 }
48 }