链表--2-链表中倒数第k个元素

题目

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

思路:

定义两个指针。第一个指针从链表的头指针开始遍历向前走k-1,第二个指针保持不动;从第k步开始,第二个指针也开始从链表的头指针开始遍历。由于两个指针的距离保持在k-1,当第一个(走在前面的)指针到达链表的尾结点时,第二个指针(走在后面的)指针正好是倒数第k个结点。

注意:

        1、头指针为空

        2、k = 0

        3、k的大小大于总结点个数

注意上述三种条件都是不合法的输入,要注意判断

/*
struct ListNode {
	int val;
	struct ListNode *next;
	ListNode(int x) :
			val(x), next(NULL) {
	}
};*/
class Solution {
public:
    ListNode* FindKthToTail(ListNode* pListHead, unsigned int k) {
    if (pListHead == nullptr || k == 0)    //注意条件判断
			return nullptr;
		ListNode *p1 = pListHead;
		ListNode *p2 = pListHead;
		for (int i = 1; i < k; i++)
		{
            if (p2->next != nullptr)
			    p2 = p2->next;
            else                        //此时是k大于节点总数的情况
                return nullptr;
		}
		while (p2->next != nullptr)
		{
			p2 = p2->next;
			p1 = p1->next;
		}
		return p1;
    }
};

猜你喜欢

转载自blog.csdn.net/ax_hacker/article/details/80960259