我们可以直接使用二重循环,枚举给定的数组 words 中的 words[i] 和 words[j] 是否可以匹配。
由于题目规定了数组 words 中包含的字符串互不相同,因此在枚举时,只要保证 i<j,那么每个字符串最多匹配一次。
C++:
class Solution {
public:
int maximumNumberOfStringPairs(vector<string>& words) {
int n = words.size();
int ans = 0;
for (int i = 0; i < n; ++i) {
for (int j = i + 1; j < n; ++j) {
if (words[i][0] == words[j][1] && words[i][1] == words[j][0]) {
++ans;
}
}
}
return ans;
}
};
python:
class Solution:
def maximumNumberOfStringPairs(self, words: List[str]) -> int:
n = len(words)
ans = 0
for i in range(n):