LeetCode-面试题 02.02. 返回倒数第 k 个节点(暴力解法、双指针解法)

Problem: 面试题 02.02. 返回倒数第 k 个节点

文章目录

  • 题目——难度:简单
  • 一、暴力解法
  • 思路
  • 解题方法
  • 复杂度
  • Code
  • 二、双指针解法
  • 思路
  • 解题方法
  • 复杂度
  • Code
  • 总结

题目——难度:简单

链接: 面试题 02.02. 返回倒数第 k 个节点

(与2009年408数据结构的算法题类似)

实现一种算法,找出单向链表中倒数第 k 个节点。返回该节点的值。

示例:
输入: 1->2->3->4->5 和 k = 2
输出: 4

说明:给定的 k 保证是有效的。

一、暴力解法

思路

倒数第k个数是正序的第n-k+1个数,构造一个函数计算链表的长度n,然后遍历链表找到第m个数

解题方法

暴力解法

复杂度

  • 时间复杂度:

l e n ( ) 为 O ( n ) , k t h T o L a s t ( ) 为 O ( n ) ,所以总的时间复杂度为 O ( n ) len()为O(n),kthToLast()为O(n),所以总的时间复杂度为O(n) len()O(n)kthToLast()O(n),所以总的时间复杂度为O(n)

  • 空间复杂度:

O ( 1 ) O(1) O(1)

Code

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    int len(ListNode* head) { //统计链表长度并返回长度n
        ListNode* p = head;
        int n = 0;
        while(p) {
            p = p->next;
            n++;
        }
        return n;
    }

    int kthToLast(ListNode* head, int k) {
        int m = len(head) - k + 1;
        if(m <= 0) return 0;
        ListNode* q = head;
        int j = 1;
        while(j < m) {
            q = q->next;
            j++;
        }
        return q->val;
    }
};

二、双指针解法

思路

从头到尾遍历单链表,并用指针p指向当前结点的前k个结点。当遍历到链表的最后一个结点时,指针p所指向的结点即为所查找的结点。

解题方法

增加两个指针变量和一个整型变量,从链表头向后遍历,其中指针first指向当前遍历的结点,指针p指向first所指向结点的第前k个结点,如果first之前没有k个结点,那么p指向表头结点。用整型变量i表示当前遍历了多少个结点,当i>k时,指针p随着每次遍历,也向前移动一个结点。当遍历完成时,p或者指向头结点,或者指向链表中倒数第k个位置上的结点。

复杂度

  • 时间复杂度:

O ( n ) O(n) O(n)

  • 空间复杂度:

O ( 1 ) O(1) O(1)

Code

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    int kthToLast(ListNode* head, int k) {
        ListNode* p = head;
        ListNode* first = p->next;
        int i = 1;
        while(first) {
            if(i > k - 1) {
                p = p->next;
            }
            first = first->next;
            i++;
        }
        if(p == head) return head->val;
        return p->val;
    }
};

总结

虽然两种方法的时间复杂度和空间复杂度都是一样的,但是由于暴力解法需要遍历链表2次,而双指针只需要1次,所以双指针解法是更好一点的。

你可能感兴趣的:(刷题,leetcode,算法,链表,c++,数据结构)