1517:チェーンテーブルの最後からk番目のノード@jobdu

1913 ワード

タイトル1517:チェーンテーブルの最後からk番目のノード
時間制限:1秒
メモリ制限:128メガ
特殊問題:いいえ
コミット:460
解決:233
タイトルの説明:
チェーンテーブルを入力し、チェーンテーブルの最後からk番目のノードを出力します.(hint:チェーンテーブルは必ず使用してください.)
入力:
入力には複数のテストサンプルが含まれ、入力はEOFで終了します.各テストケースについて、入力される第1の動作の2つの整数nおよびk(0<=n<=1000、0<=k<=1000):nは、入力するチェーンテーブル要素の個数を表し、kは、クエリの最後から数番目の要素を表す.入力された2行目は、チェーンテーブルの要素を表すn個の数t(1<=t<=100000,000)を含む.
出力:
各テストケースに対応し、結果があれば、対応する検索結果を出力します.そうでなければNULLを出力します.
サンプル入力:
5 2
1 2 3 4 5
1 0
5

サンプル出力:
4
NULL

クラシックなテーマ、肝心なのはcorner caseです
import java.io.BufferedInputStream;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.util.Scanner;


public class S15 {

	public static void main(String[] args) throws FileNotFoundException {
		BufferedInputStream in = new BufferedInputStream(new FileInputStream("in.in"));
		System.setIn(in);
		Scanner cin = new Scanner(System.in);
		
		while (cin.hasNextInt()) {
			int n = cin.nextInt();
			int k = cin.nextInt();
			ListNode dummy = new ListNode(-1);
			ListNode cur = dummy;
			for(int i=0; i<n; i++){
				cur.next = new ListNode(cin.nextInt());
				cur = cur.next;
			}
			getLastKth(dummy, k);
		}
	}
	
	public static void getLastKth(ListNode dummy, int k){
		ListNode head = dummy.next;
		ListNode probe = head;
		int i;
		for(i=0; i<k; i++){
			if(probe != null){
				probe = probe.next;
			}else{
				break;
			}
		}
		if(i != k){
			System.out.println("NULL");
			return;
		}
		ListNode cur = head;
		while(probe != null){
			probe = probe.next;
			cur = cur.next;
		}
		if(cur != null){
			System.out.println(cur.val);			
		}else{
			System.out.println("NULL");
		}
		
	}
	
	static class ListNode{
		int val;
		ListNode next;
		public ListNode(int val){
			this.val = val;
			this.next = null;
		}
	}
	
	
}