————— 第二天 —————
什么意思呢?我们以下面这个链表为例:
给定链表的头结点,但并不知道链表的实际长度,要求我们找到链表的倒数第n个结点。
假设n=3,那么要寻找的结点就是元素1:
如何利用队列呢?小灰的思路如下:
1.创建一个长度为n的队列,遍历原始链表,让结点逐一进入队列:
2.当队列已满时,让队尾元素出队,新结点入队:
3.当链表全部结点遍历完毕时,队尾的元素就是倒数第n个结点(因为队列长度是n):
————————————
首先,我们创建两个指针P1和P2,P1指向链表的头结点,P2指向链表的正数第n个结点(也就是例子中的第3个结点):
接下来,我们让指针P1和P2同时循环右移,每次右移一步,直到指针P2移动到链表的末尾:
此时,由于P2指向链表的尾结点,且P1和P2的距离是n-1,因此P1所指的结点就是我们要寻找的链表倒数第n个结点:
显然,这个方法从头到尾只需要对链表做一次遍历,而且仅仅使用了两个指针,算法的空间复杂度是O(1)。
- public class NthFromEnd {
- public static Node findNthFromEnd(Node head, int n){
- Node p1 = head;
- Node p2 = head;
- //把p2指针移动到正数第n个结点
- for(int i=1; i<n; i++){
- p2 = p2.next;
- if(p2 == null){
- throw new IllegalArgumentException("参数n超出链表长度!");
- }
- }
- //p1和p2一起右移,直到p2指向链表尾结点
- while (p2.next != null){
- p1 = p1.next;
- p2 = p2.next;
- }
- return p1;
- }
- //快速创建链表
- private static Node buildLinkList(int[] array){
- Node head = new Node(array[0]);
- Node p = head;
- for(int i=1; i<array.length; i++){
- p.next = new Node(array[i]);
- p = p.next;
- }
- return head;
- }
- //链表节点
- private static class Node {
- int data;
- Node next;
- Node(int data) {
- this.data = data;
- }
- }
- public static void main(String[] args) {
- int[] inputs = {5,3,7,2,4,1,9,8};
- Node head = buildLinkList(inputs);
- Node node = findNthFromEnd(head,3);
- System.out.println("链表倒数第3个元素是:" + node.data);
- }
- }
本文转载自微信公众号「程序员小灰」,可以通过以下二维码关注。转载本文请联系程序员小灰公众号。