一刷剑指offer(15)——输出链表倒数第k个结点

鲁棒性:

程序能够判断输入是否合乎规范要求,并对不合要求的输入予以合理的处理。通常会在函数的入口添加代码以验证用户输入是否符合要求,比如:如果输入是指针,需要考虑空指针的情况;如果输入是字符串,需要考虑空字符串的情况。

题目:
输入一个链表,输出该链表中倒数第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;
}

总结来说,当用一个指针遍历链表不能解决问题的时候,可以尝试用两个指针来遍历链表,可以让其中一个指针遍历速度快一些,或者让它先在链表上走若干步。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值