难度简单216
输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。
例如,一个链表有 6 个节点,从头节点开始,它们的值依次是 1、2、3、4、5、6 。这个链表的倒数第 3 个节点是值为 4 的节点。
示例:
给定一个链表: 1->2->3->4->5, 和 k = 2.
返回链表 4->5.
普通做法
class Solution {
public ListNode getKthFromEnd(ListNode head, int k) {
int len = 0;
ListNode a = head;
ListNode b = head;
if(head.next==null) return head;
while(a!=null){
len++;
a = a.next;
}
int t = len - k;
for(int i=0;i<len;i++){
if(i==t) return b;
b = b.next;
}
return null;
}
}
栈
class Solution {
public ListNode getKthFromEnd(ListNode head, int k) {
Stack<ListNode> stk = new Stack<ListNode>();
ListNode a = head;
while(a!=null){
stk.push(a);
a = a.next;
}
int n = stk.size();
for(int i=1;i<=n;i++){
if(i==k) return stk.peek();
stk.pop();
}
return null;
}
}
|