剑指Offer——题22（链表中的倒数第 K 个节点）

【题目】：

输入一个链表，输出该链表中倒数第k个结点。为了符合大多数人的习惯，本题从1开始计数，即链表的尾结点是倒数第1个结点。例如一个链表有6个结点，从头结点开始它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个结点是值为4的结点。

【思路】：

方法一：设置两个指针，第一个指针先遍历 k-1 步；从第 k 步开始，第二个指针指向头结点，两个结点同时往后遍历，当第一个指针到达最后一个结点时，第二个指针指向的正好是倒数第k个结点。

方法二： 使用栈，先奖链表的节点全部入栈，然后出栈，第 K 次出栈的即为倒数第 K 个节点。

package com.offer.offer22;


import java.util.Stack;

/**
 * @author zth
 * @Date 2019-08-11 16:12
 */

class ListNode {
    int val;
    ListNode next = null;

    ListNode(int val) {
        this.val = val;
    }
}
public class FindKthToTail {
    /**
     * 方法一：通过双指针完成
     * @return 参数不合法返回 null ,否则返回对应的 ListNode
     */
    public ListNode FindKthToTail0(ListNode head,int k) {
        if (null==head || k<=0){
            return null;
        }
        ListNode PAhead = head;
        for (int i = 1; i < k; i++) {
            PAhead = PAhead.next;
            if (PAhead == null){
                return null;
            }
        }
        ListNode PBehind = head;
        while (null != PAhead.next){
            PAhead = PAhead.next;
            PBehind = PBehind.next;
        }
        return PBehind;
    }

    /**
     * 用栈实现
     */
    public ListNode FindKthToTail1(ListNode head,int k) {
        if (null == head || k<=0){
            return null;
        }
        int len = 0;
        Stack<ListNode> nodeStack = new Stack<>();
        ListNode node = head;
        while (node != null){
            nodeStack.push(node);
            len++;
            node = node.next;
        }

        if (len<k){
            return null;
        }else {
            for (int i = 0; i < k; i++) {
                node = nodeStack.pop();
            }
        }
        return node;
    }

    //=====================测试代码=======================

    /*
     * null
     */
    void test1() {
        ListNode head=null;
        ListNode result=FindKthToTail1(head, 1);
        if(result==null)
            System.out.println("test1 passed!");
        else
            System.out.println("test1 failed!");
    }

    /*
     * k超出范围
     */
    void test2() {
        ListNode head=new ListNode(2);
        ListNode result=FindKthToTail1(head, 2);
        if(result==null)
            System.out.println("test2 passed!");
        else
            System.out.println("test2 failed!");
    }

    /*
     * 单个结点
     */
    void test3() {
        ListNode head=new ListNode(1);
        ListNode result=FindKthToTail1(head, 1);
        if(result.val==1)
            System.out.println("test3 passed!");
        else
            System.out.println("test3 failed!");
    }

    /*
     * 尾结点
     */
    void test4() {
        ListNode node1=new ListNode(1);
        ListNode node2=new ListNode(2);
        ListNode node3=new ListNode(3);
        ListNode node4=new ListNode(4);
        node1.next=node2;
        node2.next=node3;
        node3.next=node4;
        ListNode result=FindKthToTail1(node1, 1);
        if(result.val==4)
            System.out.println("test4 passed!");
        else
            System.out.println("test4 failed!");
    }

    /*
     * 中间结点
     */
    void test5() {
        ListNode node1=new ListNode(1);
        ListNode node2=new ListNode(2);
        ListNode node3=new ListNode(3);
        ListNode node4=new ListNode(4);
        node1.next=node2;
        node2.next=node3;
        node3.next=node4;
        ListNode result=FindKthToTail1(node1, 2);
        if(result.val==3)
            System.out.println("test5 passed!");
        else
            System.out.println("test5 failed!");
    }

    /*
     * 头结点
     */
    void test6() {
        ListNode node1=new ListNode(1);
        ListNode node2=new ListNode(2);
        ListNode node3=new ListNode(3);
        ListNode node4=new ListNode(4);
        node1.next=node2;
        node2.next=node3;
        node3.next=node4;
        ListNode result=FindKthToTail1(node1, 4);
        if(result.val==1)
            System.out.println("test6 passed!");
        else
            System.out.println("test6 failed!");
    }

    public static void main(String[] args) {
        FindKthToTail demo = new FindKthToTail();
        demo.test1();
        demo.test2();
        demo.test3();
        demo.test4();
        demo.test5();
        demo.test6();
    }
}