鲁棒性:
程序能够判断输入是否合乎规范要求,并对不合要求的输入予以合理的处理。通常会在函数的入口添加代码以验证用户输入是否符合要求,比如:如果输入是指针,需要考虑空指针的情况;如果输入是字符串,需要考虑空字符串的情况。
题目:
输入一个链表,输出该链表中倒数第k个结点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个结点。例如一个链表有6个结点,从头结点开始它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个结点是值为4的结点。
链表结点定义如下:
struct ListNode
{
int m_nValue;
ListNode* m_pNext;
};
需要遍历两次链表的方法:
为了得到倒数第k个结点,很自然会想先走到链表的尾部,再从尾部回溯k步,可是从链表结点的定义可以看出本题中链表是单向链表,只有从前往后的指针而没有从后往前的指针,因此该思路不可行。
既然不能从尾节点开始遍历,那就只能从头结点开始遍历。假设整个链表有n个结点,那么倒数第k个结点就是从头结点开始的第n-k+1个结点。
因此需要遍历链表两次:第一次统计出链表中结点的个数,第二次找到倒数第k个结点。
只需要遍历一次链表的方法:
定义两个指针:第一个指针从链表的头指针开始遍历向前走k-1,第二个指针保持不动;从第k步开始,第二个指针也开始从链表的头指针开始遍历,由于两个指针的距离保持在k-1,因此当第一个指针到达链表尾结点时,第二个指针正好是倒数第k个结点。
ListNode* FindKthToTail(ListNode* pListHead,unsigned int k)
{
ListNode *pAhead=pListHead;
ListNode *pBehind=NULL;
for(unsigned int i=0;i<k-1;i++)
pAhead=pAhead->m_pNext;
pBehind=pListHead;
while(pAhead->m_pNext!=NULL)
{
pAhead=pAhead->m_pNext;
pBehind=pBehind->m_pNext;
}
return pBehind;
}
上述代码存在不够鲁棒的问题:
1、输入的pListHead为空指针的时候
2、输入的链表总结点总数<k的时候
3、输入的参数k=0的时候
修改代码如下:
ListNode* FindKthToTail(ListNode* pListHead,unsigned int k)
{
if(pListHead==NULL || k==0)
return NULL;
ListNode *pAhead=pListHead;
ListNode *pBehind=NULL;
for(unsigned int i=0;i<k-1;i++)
{
if(pAhead->m_pNext!=NULL)
pAhead=pAhead->m_pNext;
else
{
return NULL;
}
}
pBehind=pListHead;
while(pAhead->m_pNext!=NULL)
{
pAhead=pAhead->m_pNext;
pBehind=pBehind->m_pNext;
}
retrun pBehind;
}
总结来说,当用一个指针遍历链表不能解决问题的时候,可以尝试用两个指针来遍历链表,可以让其中一个指针遍历速度快一些,或者让它先在链表上走若干步。