Ai
1 Star 2 Fork 5

LilithSangreal/LeetCode-Solutions

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
3sum-with-multiplicity.cpp 977 Bytes
一键复制 编辑 原始数据 按行查看 历史
kamyu 提交于 2018-10-14 13:57 +08:00 . Create 3sum-with-multiplicity.cpp
// 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);
}
};
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/LilithSangreal/LeetCode-Solutions.git
git@gitee.com:LilithSangreal/LeetCode-Solutions.git
LilithSangreal
LeetCode-Solutions
LeetCode-Solutions
master

搜索帮助