LookupCSV.java
4.11 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
/******************************************************************************
* Compilation: javac LookupCSV.java
* Execution: java LookupCSV file.csv keyField valField
* Dependencies: ST.java In.java StdIn.java StdOut.java
* Data files: http://algs4.cs.princeton.edu/35applications/DJIA.csv
* http://algs4.cs.princeton.edu/35applications/UPC.csv
* http://algs4.cs.princeton.edu/35applications/amino.csv
* http://algs4.cs.princeton.edu/35applications/elements.csv
* http://algs4.cs.princeton.edu/35applications/ip.csv
* http://algs4.cs.princeton.edu/35applications/morse.csv
*
* Reads in a set of key-value pairs from a two-column CSV file
* specified on the command line; then, reads in keys from standard
* input and prints out corresponding values.
*
* % java LookupCSV amino.csv 0 3 % java LookupCSV ip.csv 0 1
* TTA www.google.com
* Leucine 216.239.41.99
* ABC
* Not found % java LookupCSV ip.csv 1 0
* TCT 216.239.41.99
* Serine www.google.com
*
* % java LookupCSV amino.csv 3 0 % java LookupCSV DJIA.csv 0 1
* Glycine 29-Oct-29
* GGG 252.38
* 20-Oct-87
* 1738.74
*
*
******************************************************************************/
package edu.princeton.cs.algs4;
/**
* The {@code LookupCSV} class provides a data-driven client for reading in a
* key-value pairs from a file; then, printing the values corresponding to the
* keys found on standard input. Both keys and values are strings.
* The fields to serve as the key and value are taken as command-line arguments.
* <p>
* For additional documentation, see <a href="http://algs4.cs.princeton.edu/35applications">Section 3.5</a> of
* <i>Algorithms, 4th Edition</i> by Robert Sedgewick and Kevin Wayne.
*
* @author Robert Sedgewick
* @author Kevin Wayne
*/
public class LookupCSV {
// Do not instantiate.
private LookupCSV() { }
public static void main(String[] args) {
int keyField = Integer.parseInt(args[1]);
int valField = Integer.parseInt(args[2]);
// symbol table
ST<String, String> st = new ST<String, String>();
// read in the data from csv file
In in = new In(args[0]);
while (in.hasNextLine()) {
String line = in.readLine();
String[] tokens = line.split(",");
String key = tokens[keyField];
String val = tokens[valField];
st.put(key, val);
}
while (!StdIn.isEmpty()) {
String s = StdIn.readString();
if (st.contains(s)) StdOut.println(st.get(s));
else StdOut.println("Not found");
}
}
}
/******************************************************************************
* Copyright 2002-2016, Robert Sedgewick and Kevin Wayne.
*
* This file is part of algs4.jar, which accompanies the textbook
*
* Algorithms, 4th edition by Robert Sedgewick and Kevin Wayne,
* Addison-Wesley Professional, 2011, ISBN 0-321-57351-X.
* http://algs4.cs.princeton.edu
*
*
* algs4.jar is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* algs4.jar is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with algs4.jar. If not, see http://www.gnu.org/licenses.
******************************************************************************/