如何使用新函数扩展 java HashSet 类？

Question

我正在制作一个安卓应用程序。我想扩展 HashSet 类并向它添加一个新函数，该函数从集合中选择一个随机元素。其他一切都应该像普通的 HashSet 一样。

所以我想到了这门课：

public class RandomHashSet<E> extends HashSet<E> {
    private Random mRand = new Random();

}

我想添加一个在 HashSet 上执行此代码的函数：

if (set.size() == 0) {
    return null; 
}
int index = mRand.nextInt(set.size());
Iterator<Object> iter = set.iterator();
for (int i = 0; i < index; i++) {
    iter.next();
}
return iter.next();

但是，我不确定如何在类中实现这个函数，特别是如何调用实例化的集合以及如何对其进行迭代。

在HashSet source code 我看到他们初始化了一个名为'map'的变量：

map = new HashMap<>();

例如HashSet add()函数是这样实现的：

public boolean add(E e) {
        return map.put(e, PRESENT)==null;
}

我也可以在我的函数中以某种方式使用“地图”吗（私有）？

Answer 1

import java.util.HashSet;
import java.util.Iterator;
import java.util.Random;

public class RandomHashSet<E> extends HashSet<E> {
    private Random mRand = new Random();

    public E randomElement() {
        if (size() == 0) {
            return null;
        }
        int index = mRand.nextInt(size());
        Iterator<E> iter = iterator();
        for (int i = 0; i < index; i++) {
            iter.next();
        }
        return iter.next();
    }
}