哈希第六天

leetcode第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 。所有整数的范围在 -2^28 到 2^28 - 1 之间，最终结果不会超过 2^31 - 1 。

思路

对比昨天的【两数之和】，该题给定4个数组（昨天给定一个数组），找到A[i] + B[j] + C[k] + D[l] = 0的元组个数（i, j, k, I）。

思路和昨天类似，使用unordered_map<int, int> m, key 存储A、B数组中各元素之和，value 存储和出现的次数，然后再遍历C、D数组，查找map 中是否有 0 - c - d的key存在即可。

步骤：

1. 首先定义 一个unordered_map，key放a和b两数之和，value 放a和b两数之和出现的次数。
2. 遍历大A和大B数组，统计两个数组元素之和，和出现的次数，放到map中。
3. 定义int变量count，用来统计a+b+c+d = 0 出现的次数。
4. 再遍历大C和大D数组，找到如果 0-(c+d) 在map中出现过的话，就用count把map中key对应的value也就是出现次数统计出来。
5. 最后返回统计值 count 就可以了。

C++代码

class solution
{
public:
	int fourSumCount(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D)
	{
		unordered_map<int, int> m;		//key 保存sum， value保存和出现的次数
		//遍历大A和大B数组，统计两个数组元素之和，和出现的次数，放到map中。
		for (int a : A)		//c++11新特性：for的区间遍历
		{
			for (int b : B)
			{
				m[a + b]++;
			}
		}
		int count = 0;		//统计a+b+c+d = 0 出现的次数。
		//再遍历大C和大D数组，找到如果 0-(c+d) 在map中出现过的话，就用count把map中key对应的value也就是出现次数统计出来。
		for (int c : C)
		{
			for (int d : D)
			{
				if (m.find(0 - c - d) != m.end())
				{
					//统计次数，value存储的是sum = 0 - c - d = a + b
					count += m[0 - c - d];
				}
			}
		}
		//最后返回统计值 count 就可以了。
		return count;
	}
};

如果要返回包含下标的元组而不是元组个数要如何处理？

上例中返回 [(0, 0, 0, 1), (1, 1, 0, 0)] 而不是返回2

class solution
{
public:
	vector<vector<int>> fourSumCount(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D)
	{
		//key 保存sum， value元素和为sum的元素下标， map.second.size() 即为和出现的次数
		unordered_map<int, vector<vector<int>>> m;		
		//遍历大A和大B数组，统计两个数组元素之和，和出现的次数，放到map中。
		for (int i = 0; i < A.size(); i++)		
		{
			for (int j = 0; j < B.size(); j++)
			{
				auto iter = m.find(A[i] + B[j]);
				if (iter != m.end())
				{//因为unordered_map 键值不可重复，所以m 中已有sum = A[i] + B[j]时在其下标数组中添加当前下标即可。
					iter->second.push_back(vector<int>(i, j));
				}
				else
				{//没有则新加一个和（key） 为sum = A[i] + B[j]的对组
					m.insert(pair<int, vector<vector<int>>>(A[i] + B[j], vector<vector<int>>(1, vector<int>(i, j))));
				}
				
			}
		}
		
		//因为要返回下标元组，所以需要一个vector
		vector<vector<int>> result;
		//再遍历大C和大D数组，找到如果 0-(c+d) 在map中出现过的话，就用count把map中key对应的value也就是出现次数统计出来。
		for (int i = 0; i < C.size(); i++)
		{
			for (int j = 0; j < D.size(); j++)
			{
				if (m.find(0 - nums3[i] - nums4[j]) != m.end())
				{
					//将满足条件的下标添加到对应数组中,m[0 - c - d]是一个vector<vector<int>> 二维数组(一行多列的二维数组)
					for (int k = 0; k < m[0 - C[i] - D[j]].size(); k++)
					{
						m[0 - C[i] - D[j]][k].push_back(i);
						m[0 - C[i] - D[j]][k].push_back(j);
						//再添加到最终结果中
						result.push_back(m[0 - C[i] - D[j]][k]);
					}
				}
			}
		}
		return result;
	}
};

原文：哈希表：其实需要哈希的地方都能找到map的身影