基本上，您需要创建某种数据结构，您需要在其中跟踪关键字、链接到的标题数量以及与之关联的标题 ID；像这样：
[canon, {2, [ title1, title2 ] } ]
如果是这种情况，这至少有两个原因是不好的：
- 使用计数作为键是不好的，因为计数会在其他元素中重复
- 您可以通过返回给定关键字的标题 ID 列表的大小或长度来简单地计算计数。
我提出的解决方案有两个方面：使用一个类来捕获您的关键字和标题 ID 映射，然后是一组这些映射对象。
类
public class KeywordMapping implements Comparable<KeywordMapping> {
private final String keyword;
private TreeSet<String> titleIds = new TreeSet<String>();
private int frequency;
public KeywordMapping(String keyword, String titleId) {
this.keyword = keyword;
titleIds.add(titleId);
}
public String getKeyword () {
return keyword;
}
public int getTitleCount () {
return titleIds.size();
}
public void addTitleId (String titleId) {
titleIds.add(titleId);
}
public String getFirstTitleId () {
return titleIds.first();
}
public void incrementFrequency(String keyword) {
if (this.keyword.equals(keyword)) {
frequency++;
}
}
public int getFrequency() {
return frequency;
}
public boolean containsTitle(String titleId) {
return titleIds.contains(titleId);
}
@Override
public int hashCode () {
final int prime = 31;
int result = 1;
result = prime * result + ( (keyword == null) ? 0 : keyword.hashCode());
result = prime * result + ( (titleIds == null) ? 0 : titleIds.hashCode());
return result;
}
@Override
public boolean equals (Object obj) {
if (this == obj) return true;
if (obj == null) return false;
if (getClass() != obj.getClass()) return false;
KeywordMapping other = (KeywordMapping) obj;
if (keyword == null) {
if (other.keyword != null) return false;
} else if (!keyword.equals(other.keyword)) return false;
if (titleIds == null) {
if (other.titleIds != null) return false;
} else if (!titleIds.equals(other.titleIds)) return false;
return true;
}
@Override
public String toString () {
return "KeywordMapping [keyword=" + keyword + ", titleIds=" + titleIds + ", frequency=" + frequency
+ "]\n";
}
@Override
public int compareTo (KeywordMapping o) {
return COMPARATOR.compare(this, o);
}
private static final Comparator<KeywordMapping> COMPARATOR =
Comparator.comparing( (KeywordMapping keyMapping) -> keyMapping.keyword);
}
这个解决方案没问题。我宁愿使用 Builder Pattern 来构建它并使此类不可变，以便仅在将所有标题 ID 添加到类中时才构建此类的每个实例。但是，让我解释一下这个实现的一些细节。我使用TreeSet 来存储标题ID，以便它们按自然顺序（按字母顺序）排列。这对于存储标题 ID 并不重要，但稍后会很重要。最后，如果您需要将这些对象插入到可排序的集合中，该类必须实现 Comparable<T>。
关键字映射的TreeMap
TreeMap<String, KeywordMapping> mappings = new TreeMap<>();
必须使用树形图来按键按自然顺序排列条目。
我创建了这个简单的 main 方法来演示这个实现
public static void main (String[] args) {
List<String> titles = List.of("title2: canon camera canon",
"title3: wireless mouse",
"title1: canon photo printer",
"title4: wireless charger", "title5: mouse trap");
MasterMap mappings = new MasterMap(); // A class wrapping TreeMap<String, KeywordMapping> mappings (the code is shown later on)
for (String title : titles) {
String[] tokens = title.split(": ");
Set<String> keywords = new HashSet<>(Arrays.asList(tokens[1].split(" ")));
for (String keyword : keywords) {
KeywordMapping km = mappings.get(keyword); // don't create duplicate keyword mapping objects
if (km == null) {
km = new KeywordMapping(keyword, tokens[0]);
}
km.addTitleId(tokens[0]);
km.incrementFrequency(keyword);
mappings.put(keyword, km); // update existing or new keyword mapping entry
}
}
System.out.println(mappings);
System.out.println("First entry: " + mappings.firstMapping());
System.out.println("First title in first entry: " + mappings.firstMapping().getValue().getFirstTitleId());
}
这个程序的输出是：
camera=KeywordMapping [keyword=camera, titleIds=[title2], frequency=1]
canon=KeywordMapping [keyword=canon, titleIds=[title1, title2], frequency=2]
charger=KeywordMapping [keyword=charger, titleIds=[title4], frequency=1]
mouse=KeywordMapping [keyword=mouse, titleIds=[title3, title5], frequency=2]
photo=KeywordMapping [keyword=photo, titleIds=[title1], frequency=1]
printer=KeywordMapping [keyword=printer, titleIds=[title1], frequency=1]
trap=KeywordMapping [keyword=trap, titleIds=[title5], frequency=1]
wireless=KeywordMapping [keyword=wireless, titleIds=[title3, title4], frequency=2]
First entry: camera=KeywordMapping [keyword=camera, titleIds=[title2], frequency=1]
First title in first entry: title2
请注意，即使原始标题列表不是按自然顺序排列的，在插入KeywordMapping 对象后，支持标题 ID 列表的集合也会按顺序排列它们。这与打印树形图中的条目的顺序相同。
最后，每个标题的频率计数仅增加一次，因为标题 ID 列表由 Set 而不是 List 支持，并且 Java 中的集合不允许重复。因此，对于title2: canon camera canon 这样的情况，关键字canon 只计算一次，因为对于该标题，关键字集只有{cannon, camera}。
包装 TreeMap 以增加功能
public class MasterMap {
private TreeMap<String, KeywordMapping> mappings = new TreeMap<>();
public KeywordMapping put(String key, KeywordMapping value) {
return mappings.put(key, value);
}
public KeywordMapping get(String key) {
return mappings.get(key);
}
public KeywordMapping firstMapping() {
return mappings.firstEntry().getValue();
}
public String getKeywordForTitle (String titleId) {
List<KeywordMapping> keywords = new ArrayList<>();
Collection<KeywordMapping> values = mappings.values();
Iterator<KeywordMapping> iter = values.iterator();
while (iter.hasNext()) {
KeywordMapping value = iter.next();
if (value.containsTitle(titleId)) {
keywords.add(value);
}
}
KeywordMapping temp = keywords.stream()
.max(Comparator.comparingInt(KeywordMapping::getFrequency)
.thenComparing(KeywordMapping::getKeyword).reversed())
.get();
return temp.getKeyword();
}
}
为了满足最后一个要求，我决定将TreeMap 包装到一个类中，以便我可以添加方法来执行最后一个操作：根据以下条件返回与给定标题关联的关键字：返回频率最高的关键字或按字母顺序排列的第一个关键字（如果频率相同）。实现的函数不会以任何方式操作KeywordMapping 对象或包装的树映射。它的作用是使用Comparator 来根据该条件查找匹配的关键字。
添加更改后（之前的代码已更新），将“title3”传递给函数会产生以下输出：
Keyword for title3: mouse
这是因为与“title3”{mouse, wireless} 关联的两个关键字具有相同的频率 (2)，但“mouse”是集合中按字母顺序排列的第一个关键字。