美文网首页
剑指offer 链表中的倒数第K个节点 Python and C

剑指offer 链表中的倒数第K个节点 Python and C

作者: GhostintheCode | 来源:发表于2020-01-21 18:52 被阅读0次

    题目描述

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

    思路

    假设链表中的节点数大于等于k个,那么一定会存在倒数第k个节点,首先使用一个快指针先往前走k步,然后两个指针每次走一步,两个指针之间始终有k的距离,当快指针走到末尾时,慢指针所在的位置就是倒数第k个节点。

    做题可能出现的问题

    考虑k小于0的情况,其次就是链表的长度是否够k个长度

    python

    # -*- coding:utf-8 -*-
    # class ListNode:
    #     def __init__(self, x):
    #         self.val = x
    #         self.next = None
    
    class Solution:
        def FindKthToTail(self, head, k):
            # write code here
            if not head  or k<=0:
                return None
            p = q = head
            count = 0
            while q is not None and count<k:
                q = q.next
                count += 1
            if count<k:
                return None
            while q is not None:
                q = q.next
                p = p.next
            return p
    

    C++

    /*
    struct ListNode {
        int val;
        struct ListNode *next;
        ListNode(int x) :
                val(x), next(NULL) {
        }
    };*/
    class Solution {
    public:
        ListNode* FindKthToTail(ListNode* pListHead, unsigned int k) {
            ListNode *ptr1 = pListHead; // 指针1
            ListNode *ptr2 = pListHead; //指针2
            if (pListHead == NULL || k == 0)//考虑全面
                return NULL;
            int i;
            for(i=0;i<k-1;i++) // p1 走k-1步
            {
                if(ptr1->next) // 这个条件等价于方法一种的,if(n<k)return NULL; 因为当i<k-1时,如果发生ptr1->next==null,说明k太大了,超出了链表的总结点个数
                    ptr1 = ptr1->next;
                else 
                    return NULL;
            }
            while(ptr1->next) // p1走到end, p2开始走
            {
                ptr1=ptr1->next;            
                ptr2=ptr2->next;
            }
            return ptr2;
        }
    };
    

    相关文章

      网友评论

          本文标题:剑指offer 链表中的倒数第K个节点 Python and C

          本文链接:https://www.haomeiwen.com/subject/grdtzctx.html