剑指:链表中倒数第k个节点

时间:2019-06-28 21:01:17   收藏:0   阅读:85

题目描述

输入一个链表,输出该链表中倒数第k个结点。

 

解法

pre 指针走 k-1 步。之后 cur 指针指向 phead,然后两个指针同时走,直至 pre 指针到达尾结点。

即cur与pre始终相距k-1

当用一个指针遍历链表不能解决问题的时候,可以尝试用两个指针来遍历链表。可以让其中一个指针遍历的速度快一些。

此题需要考虑一些特殊情况。比如 k 的值小于 0 或者大于链表长度。

 

package demo;

public class Solution {
    
    public static class ListNode{
        int val;
        ListNode next = null;
        
        ListNode(int val){
            this.val = val;
        }
    }
    
    public static ListNode findKthtoTail(ListNode head, int k){
        if(head == null || k<1){
            return null;
        }
        ListNode pre = head;
        for(int i=0; i<k-1;i++){//先将pre指针移到第k-1位
            if(pre.next != null)
                pre = pre.next;
            else
                return null;
        }
        
        ListNode cur = head;
        while(pre.next != null){//遍历
            pre = pre.next;
            cur = cur.next;
        }
        return cur;
    }

    public static void main(String[] args) {
        ListNode head = new ListNode(0);
        
        ListNode node1 = new ListNode(1); head.next = node1;
        ListNode node2 = new ListNode(2); node1.next = node2;
        ListNode node3 = new ListNode(3); node2.next = node3;
        ListNode node4 = new ListNode(4); node3.next = node4;
        ListNode node5 = new ListNode(5); node4.next = node5;
        ListNode node6 = new ListNode(6); node5.next = node6;
        ListNode node7 = new ListNode(7); node6.next = node7;
        ListNode node8 = new ListNode(8); node7.next = node8;
        ListNode node9 = new ListNode(9); node8.next = node9;
        ListNode node10 = new ListNode(10); node9.next = node10;
        
        ListNode cur = head;
        while(cur.next != null){
            cur = cur.next;
            System.out.print(cur.val + " ");
        }
        
        int k = 4;
        ListNode resNode = findKthtoTail(head, k);
        System.out.println("\n链表倒数第"+k+"个元素为:" + resNode.val);

    }
}

 

只要从头走n-k+1步即可。

 

评论(0
© 2014 mamicode.com 版权所有 京ICP备13008772号-2  联系我们:gaon5@hotmail.com
迷上了代码!