代码拉取完成,页面将自动刷新
// Time: O(n^2), n is the number of disctinct A[i]
// Space: O(n)
class Solution {
public:
int threeSumMulti(vector<int>& A, int target) {
unordered_map<int, uint64_t> count;
for (const auto& a : A) {
++count[a];
}
uint64_t result = 0;
for (const auto& kvp1 : count) {
for (const auto& kvp2 : count) {
int i = kvp1.first, j = kvp2.first, k = target - i - j;
if (!count.count(k)) {
continue;
}
if (i == j && j == k) {
result += count[i] * (count[i] - 1) * (count[i] - 2) / 6;
} else if (i == j && j != k) {
result += count[i] * (count[i] - 1) / 2 * count[k];
} else if (i < j && j < k) {
result += count[i] * count[j] * count[k];
}
}
}
return result % static_cast<int>(1e9 + 7);
}
};
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。