Map、groupby、值列表

Question

给定一张有序地图，<string,list<sting>>：

k1 [v1,v2,v5]
k2 [v1]
k3 [v4,v5]
k4 [v2]

还有一组字符串[buckets]：

v2 v4 v5

输出是一张地图：

v2 [k1,k4]
v4 [k3]
v5 [k1, k3]

顺序在输出中很重要。 有没有比简单的暴力破解更好的方法，编程语言是java？？

Answer 1

只需遍历地图并将值收集到一个SortedMap 中，并持有SortedSets（例如，TreeMap 的TreeSets），将排序和复制过程留给底层数据结构：

Map<String, List<String> input = ...;
Set<String> buckets = ...;
SortedMap<String, TreeSet> output = new TreeMap<String, TreeSet>();

for (Map.Entry<String, List<String> entry : input) {
    List<String> possibleBuckets = entry.getValue();
    for (String possibleBucket : possibleBuckets) {
        if (buckets.contains(possibleBucket) {
            Set<String> set = output.get(possibleBucket);
            if (set == null) {
                set = new TreeSet<String>();
                output.put (possibleBucket, set);
            }
            set.add (entry.getKey());
        } 
    }
}