输入一个链表,输出该链表中倒数第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),也就是健壮性,要注意判断各种可能出现的不合法的输入。