面试题 17.26. 稀疏相似度
思路:
暴力法显然是可以,但是如何简化呢?思维转变一下,不是把两个数组比较,看有没有重复的。而是每次的数字出现,都去检查一下是否出现过,在哪个数组里出现的。
public List<String> computeSimilarities(int[][] docs) {
int n = docs.length;
List<String> ans = new ArrayList<>();
int[][] intersection = new int[n][n];
Map<Integer, List<Integer>> map = new HashMap<>();
for (int i = 0; i < n; i++) {
for (int j = 0; j < docs[i].length; j++) {
List<Integer> list = map.computeIfAbsent(docs[i][j], key -> new ArrayList<>());
for (int k : list) ++intersection[k][i];
list.add(i);
}
//因为要求K<I所以放循环这里是没有问题的
for (int k = 0; k < i; k++) {//k < docs.length也可以
if (intersection[k][i] == 0) continue;
ans.add(k + "," + i + ": " + String.format("%.4f",
(double) intersection[k][i] / (docs[k].length + docs[i].length - intersection[k][i])));
}
}
return ans;
}