Cracking the coding interview--Q2.2

原文：

Implement an algorithm to find the nth to last element of a singly linked list.

译文：

实现一个算法从一个单链表中返回倒数第n个元素。

package chapter_2_LinkedLists;

import java.util.Scanner;

/**
* 实现一个算法从一个单链表中返回倒数第n个元素。
*/
public class Question_2_2 {

/**
* 如果提前知道链表的长度,可以直接计算正向位置
*/
public static int findNthLast(LinkList linkList, int n) {
Node p;
p = linkList.head;
if(n > linkList.size) {
return -1;
}
int locat = linkList.size - n + 1;
int i = 1;
boolean flag = true;
while(flag) {
if(i != locat) {
i++;
p = p.next;
} else {
flag = false;
}
}
return p.data;
}

/**
* ********
* 如果提前不知道链表的长度,可以使用两个指针，间距保持n,只需要扫描一遍
*/
public static int findNthLast2(LinkList linkList, int n) {
Node p, q;
p = linkList.head;
int i = 1;
q = p;
while(i != n && q != null) {
q = q.next;
i ++;
}
if(q == null) {
return -1;
}
// 上面步骤保持n-1，开始往后移动
while(q.next != null) {
q = q.next;
p = p.next;
}
return p.data;
}

public static void main(String args[]) {

Scanner scanner = new Scanner(System.in);
LinkList linkList = new LinkList();
int temp;
int num = scanner.nextInt();
for(int i=0; i< num; i++) {
temp = scanner.nextInt();
linkList.add(temp);
}
scanner.nextLine();
int n = scanner.nextInt();

int result = findNthLast2(linkList, n);
if(result < 0) {
System.out.format("error : %3d\n", result);
} else {
System.out.format("%3d\n", result);
}
}
}