Problem: 面试题 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次,所以双指针解法是更好一点的。