subList bugfix

twitter · Jun 28, 2012 · ae9727e · ae9727e
1 parent cbeaf53
commit ae9727e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/examples/java/RandomWalk.java b/examples/java/RandomWalk.java
@@ -66,7 +66,9 @@ public static void main(String[] args) {
 
     // Sort neighbors (or nodes) in descending number of visits and take the top 10 neighbors
     List<Integer> topNeighbors = Ordering.natural().onResultOf(Functions.forMap(neighbors)).reverse()
-            .immutableSortedCopy(neighbors.keySet()).subList(0, 10);
+            .immutableSortedCopy(neighbors.keySet());
+
+    if (topNeighbors.size() > 10) topNeighbors = topNeighbors.subList(0, 10);
 
     // Print the top 10 neighbors (and paths)
     System.out.printf("%8s%10s\t%s\n", "NodeID", "#Visits", "Top 2 Paths with counts");