剑指offer-打印链表倒数第k个结点

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

posted on 2015-10-19 10:53  qingfei  阅读(260)  评论(0编辑  收藏  举报