输入一个链表,输出该链表中倒数第k个结点。
看到这个题目,我们很容易想到一种做法,遍历两次链表,一次得到链表长度,另外一次得到倒数第k个结点。
不过要注意的是边界值的测试
-
- 链表为空
- k大于链表长度
于是我的做法如下:
public static ListNode FindKthToTail(ListNode head,int k) { if(head==null) { return null; } ListNode prehead=new ListNode(0); prehead.next=head; int length=0; ListNode p=prehead; while(p!=null) { p=p.next; length++; } if(k>=length) { return null; } p=prehead; int i=0; while(p!=null) { if(length-i==k) { return p; } p=p.next; i++; } return null; }
后来看了剑指offer以后,发现其实还有一种做法,不需要遍历链表两次,用的是两个指针。
思路如下:
-
- 使用两个指针p1,p2,p1先移动k-1步,到达第k个结点
- p2指向头结点,此时p2和p1之间相差k-1
- 此时p1,p2同时向前移动一个结点,直到p1到达最后一个结点
- 此时,p2所在结点就是倒数第k个结点
public static ListNode FindKthToTail(ListNode head,int k) { if(head==null||k<=0) { return null; } //先让p1指向头结点 ListNode p1=head;//指向头指针算是移动了1次,所以i从1开始 //p1移动k-1次,则可以移动到第k个节点 for(int i=1;i<k;i++){ p1=p1.next; //因为有可能k大于整个链表的长度,所以需要判断p1是否等于空 if(p1==null) { break; } } //如果p1等于空,说明k大于链表长度 if(p1==null) { return null; } //让p2指向head,现在p1和p2相差k-1 ListNode p2=head; //在指向最后一个结点之前,p1,p2同时移动一步。 while(p1.next!=null) { p1=p1.next; p2=p2.next; } //当p1到达最后一个结点时,p2所在的位置就是倒数第k个节点。 return p2; }
而且还有一个边界值需要测试的,就是当k小于等于0的时候,应该返回空,若不加以判断,在上面这个程序中,返回值是不对的,因为k=0,所以p1没有移动过。这时p1,p2同时移动到最后一个值,最后返回的是最后一个值。
这道题提醒我们要注意代码的鲁棒性(rebust),也就是健壮性,要注意判断各种可能出现的不合法的输入。