链表中倒数第k个数
/*
public class ListNode {
int val;
ListNode next = null;
ListNode(int val) {
this.val = val;
}
}*/
public class Solution {
public ListNode FindKthToTail(ListNode head,int k) {
if(head == null || k<1){
return null;
}
ListNode tempA = head;
for(int i=0;i<k-1;i++){
if(tempA.next != null){
tempA = tempA.next;
}else{
return null;
}
}
ListNode tempB = head;
while(tempA.next != null){
tempB = tempB.next;
tempA = tempA.next;
}
return tempB;
}
}
概述
鲁棒是英文Robust的音译,有时也翻译成健壮性。所谓的鲁棒性是指程序能够判断输入是否合乎规范要求,并对不符合要求的输入予以合理的处理。
当我们用一个指针 遍 历 链 表 不 能 解 决 问 题 的 时 候 ,可以尝试用两个指针 来 遍 历 链 表 可 以 让 其 中 一 个 指 针 遍 历 的 速 度 快 一 些 (比如一次在链表上 走 两 步 ),或者让它先在链表上走若千步
问题
输入一个链表,输出该链表中倒数第k个结点。
思路
为了得到倒数第k 个节点,很自然的想法是先走到链表的尾端,再从尾端回溯k步。可是我们从链表节点的定义可以看出,本题中的链表是单向链表,单向链表的节点只有从前往后的指针而没有从后往前的指针,因此这种思路行不通。
遍历两次
既然不能从尾节点开始遍历这个链表,我们还是把思路回到头节点上来。假设整个链表有n个节点,那么倒数第k • 个节点就是从头节点开始的第n-k+1个节点。如果我们能够得到链表中节点的个数n 那么只要从头节点 开 始 往 后 走 n-k+1步 就 可 以 了 。如何得到节点数n,这个不难,只需要从头开始遍历链表,每经过一个节点,计数器加1就行了。
也就是说我们需要遍历链表两次,第一次统计出链表中节点的个数,第二次就能找到倒数第A 个节点。但是当我们把这种思路解释给面试官之后,他会告诉我们他期待的解法只需要遍历链表一次
public static void main(String[] args) {
ListNode head = new ListNode(1);
head.next = new ListNode(2);
head.next.next = new ListNode(3);
head.next.next.next = new ListNode(4);
head.next.next.next.next = new ListNode(5);
FindKthToTail(head,2);
}
public static ListNode FindKthToTail(ListNode head, int k) {
if(head==null){
return null;
}
int count=1;
ListNode old=head;
while(head.next!=null){
head=head.next;
count++;
}
if(k>count){
return null;
}
for(int i=0;i<count-k;i++){
old=old.next;
}
return old;
}
}
public class ListNode {
int val;
ListNode next;
ListNode(int x) { val = x; }
}
遍历一次
我们可以定义两个指针。第一个指针从链表的头指针开始遍历向前走k-1,第二个指针保持不动;从第k步开始,第二个指针也开始从链表的头指针开始遍历。由于两个指针的距离保持在k-1,当第一个(走在前面的)指针到达链表的尾结点时,第二个指针(走在后面的)指针正好是倒数第k个结点。
除此之外,要注意代码的鲁棒性。需要判断传入参数合法性问题。
public ListNode removeNthFromEnd(ListNode head, int n) {
// 输入的链表不能为空,并且k大于0
if (n < 1 || head == null) {
return null;
}
// 指向头结点
ListNode pointer = head;
// 倒数第k个结点与倒数第一个结点相隔k-1个位置
// pointer先走k-1个位置
for (int i = 1; i < n; i++) {
// 说明还有结点
if (pointer.next != null) {
pointer = pointer.next;
}
// 已经没有节点了,但是i还没有到达k-1说明k太大,链表中没有那么多的元素
else {
// 返回结果
return null;
}
}
// pointer还没有走到链表的末尾,那么pointer和head一起走,
// 当pointer走到最后一个结点即,pointer.next=null时,head就是倒数第k个结点
while (pointer.next != null) {
head = head.next;
pointer = pointer.next;
}
// 返回结果
return head;
}
/*
class ListNode {
int val;
ListNode next = null;
ListNode(int val) {
this.val = val;
}
}*/
方法2
public static ListNode FindKthToTail(ListNode head,int k) {
if(head == null) return null;
ListNode p1 = head,p2 = head;
while (k>0 && p2 != null){
p2 = p2.next;
k--;
}
//p2 指向head说明k<=0,p2==null && k>0说明 k超过了链表的长度
if(p2 == head || (p2 == null && k>0)) return null;
while (p2!=null){
p1 = p1.next;
p2 = p2.next;
}
return p1;
}
相关题目
求链表的中间节点。如果链表中的节点总数为奇数,则返回中间节点;如果节点总数是偶数,则返回中间两个节点的任意一个。为了解决这个问题,我们也可以定义两个指针,同吋从链表的头节点出发,一个指针一次走一步,另一个指针一次走两步。当走得快的指针走到链表的末尾时,走得慢的指针正好在链表的中间。
当我们用一个指针 遍 历 链 表 不 能 解 决 问 题 的 时 候 ,可以尝试用两个指针 来 遍 历 链 表 可 以 让 其 中 一 个 指 针 遍 历 的 速 度 快 一 些 (比如一次在链表上 走 两 步 ),或者让它先在链表上走若千步。
LeetCode
给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
说明:
给定的 n 保证是有效的。
public ListNode removeNthFromEnd(ListNode head, int n) {
// 输入的链表不能为空,并且k大于0
if (n < 1 || head == null) {
return null;
}
// 指向头结点
ListNode pointer = head;
// 倒数第k个结点与倒数第一个结点相隔k-1个位置
// pointer先走k-1个位置
for (int i = 1; i < n; i++) {
// 说明还有结点
if (pointer.next != null) {
pointer = pointer.next;
}
// 已经没有节点了,但是i还没有到达k-1说明k太大,链表中没有那么多的元素
else {
// 返回结果
return null;
}
}
// pointer还没有走到链表的末尾,那么pointer和head一起走,
// 当pointer走到最后一个结点即,pointer.next=null时,head就是倒数第k个结点
while (pointer.next != null) {
head = head.next;
pointer = pointer.next;
}
// 返回结果
return head;
}