题目地址:
https://leetcode.com/problems/sentence-similarity/
给定一个列表,列表中每个位置都是一个长为 2 2 2的字符串数组,规定数组内的字符串是”相似的“。再给定两个字符串数组,问这两个数组是否对应位置的字符串都相似。注意,规定”相似关系“不具有传递性,只具有对称性。如果两个数组长度不等则直接返回false;并且如果对应位置上的字符串相等,也视为相似,不必含在列表里。
import java.util.*;
public class Solution {
class Pair {
String s1, s2;
public Pair(String s1, String s2) {
// 为了保证含相同字符串的两个Pair就相等,这里要规定s1 < s2;
if (s1.compareTo(s2) <= 0) {
this.s1 = s1;
this.s2 = s2;
} else {
this.s1 = s2;
this.s2 = s1;
}
}
@Override
public boolean equals(Object p) {
Pair pair = (Pair) p;
return s1.equals(pair.s1) && s2.equals(pair.s2);
}
@Override
public int hashCode() {
return Objects.hash(s1, s2);
}
}
public boolean areSentencesSimilar(String[] words1, String[] words2, List<List<String>> pairs) {
if (words1.length != words2.length) {
return false;
}
Map<String, Integer> map = new HashMap<>();
Set<Pair> set = new HashSet<>();
for (List<String> pair : pairs) {
set.add(new Pair(pair.get(0), pair.get(1)));
}
for (int i = 0; i < words1.length; i++) {
// 特判,如果相等则符合条件,略过
if (words1[i].equals(words2[i])) {
continue;
}
// 不相似则返回false
if (!set.contains(new Pair(words1[i], words2[i]))) {
return false;
}
}
return true;
}
}
时空复杂度 O ( n l ) O(nl) O(nl), n n n为单词个数, l l l为单词最长长度。
注意:
在Pair类里面,不能只把equals改写为return (s1.equals(pair.s1) && s2.equals(pair.s2)) || (s1.equals(pair.s2) && s2.equals(pair.s1));
,因为哈希表在存pair的时候会先算哈希值,哈希值不等就直接判定为不等了,不会再调用equals,这回造成判定两个单词是否相似的时候,两个单词的顺序会影响判断。比较好的方法是初始化pair的时候就直接强行规定s1 < s2
。