访问树集中的元素

Question

我在这里尝试一些非常基本的java，但遇到了一些令人头疼的问题。本质上，我需要将文件中的某些元素读入某种类型的数组或列表中，对它们进行排序，消除重复项，然后返回前三个元素。 TreeSet 似乎非常适合，因为它可以进行排序并删除重复项。我的问题是我对如何仅返回前三个元素感到困惑。迭代器似乎贯穿整个集合。使用手动迭代器创建一个 while 循环来包含一个保存迭代器循环的 while 循环似乎令人困惑并且不太可能成功。这里的答案是我需要迭代树集并将每个元素放入数组列表中，以便我可以访问前三个元素吗？我的意思是，这似乎可行，但看起来非常复杂。尖端？

Answer 1

使用 Guava 你可以这样做

return Lists.newArrayList(Iterables.limit(treeSet, 3));

Answer 2

嗯。显而易见的事情有什么问题吗？

ArrayList<MyType> buffer = new ArrayList<MyType>(3); 

for( MyType elt: myTreeSet ) {

    buffer.add(elt);
    if( buffer.size() == 3 ) break;
}

或者

ArrayList<MyType> buffer = new ArrayList<MyType>(3);
Iterator<MyType> iter = myTreeSet.iterator();

while( iter.hasNext() && buffer.size() < 3 ) buffer.add(iter.next());

如果您更喜欢“脱糖”版本？

Answer 3

字符串示例：

TreeSet<String> treeSet = new TreeSet<String>(); 

// you populate treeSet with data

String[] stringArray = new String[NUMBER_OF_NEEDED_RECORDS];
for(int i =0; i < NUMBER_OF_NEEDED_RECORDS; i++) {
    stringArray[i] = treeSet.pollFirst();
}

Answer 4

我会使用（希望您使用 Java 1.6）：

Arrays.copyOf(myTreeSet.toArray(), Math.min(3, myTreeset.size()));

编辑：使用我添加的 Math.min() 大小进行防弹