454. 四数相加 II
给定四个包含整数的数组列表 A , B , C , D ,计算有多少个元组 (i, j, k, l) ,使得 A[i] + B[j] + C[k] + D[l] = 0。
为了使问题简单化,所有的 A, B, C, D 具有相同的长度 N,且 0 ≤ N ≤ 500 。所有整数的范围在 -228 到 228 - 1 之间,最终结果不会超过 231 - 1 。
看到这个题目我以为会用递归,即先找到一个A[i],再使得接下来的问题变为,找到B[j]+C[k]+D[l]=-A[i],但显然这样子会有很多的麻烦。暴力n^4求解也大概率超时,随后看了题解。。。
```cpp
class Solution {
public:
int fourSumCount(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D) {
unordered_map<int, int> countAB;
for (int u: A) {
for (int v: B) {
++countAB[u + v];
}
}
int ans = 0;
for (int u: C) {
for (int v: D) {
if (countAB.count(-u - v)) {
ans += countAB[-u - v];
}
}
}
return ans;
}
};
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/4sum-ii/solution/si-shu-xiang-jia-ii-by-leetcode-solution/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
在这里学习到了一中遍历A+B+C+D的n^2算法,即使用乘法的交换律和结合律降低时间复杂度。
这段代码里的哈希表unordered_map用法要好好学习。
下面两篇文章可以参考c++的哈希表基本用法:
https://blog.csdn.net/qq_32172673/article/details/85165194
https://www.cnblogs.com/wanghui-garcia/p/11683668.html
523. 连续的子数组和
给定一个包含 非负数 的数组和一个目标 整数 k,编写一个函数来判断该数组是否含有连续的子数组,其大小至少为 2,且总和为 k 的倍数,即总和为 n*k,其中 n 也是一个整数。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/continuous-subarray-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
看到这个题目我想到了一个之前了解的数学规律,就是mod一个数的结果相同的话那么这些数的中间值%k就为0。结合刚才学到的unordered_map,很容易写出了以下的代码:
class Solution {
public:
bool checkSubarraySum(vector<int>& nums, int k) {
int sum=0;
unordered_map<int,int> mp;
mp[0]=-1;
int temp=0;
for(int i=0;i<nums.size();i++)
{
sum+=nums[i];
if(k!=0)
temp=sum%k;
if(!mp.count(temp))
mp[temp]=i;
if(mp.count(temp)&&(i-mp[temp])>1)
return true;
}
if(sum==0&&nums.size()>1)
return true;
return false;
}
};
这里需要注意的是对题目中特殊情况的考虑,如0+0可以整除任何数,还有就是k=0的特殊情况。
自己写出一个最优解的中等题目很开心嘻嘻嘻。