美文网首页
剑指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