LeetCode 2341. 数组能形成多少数对
给你一个下标从 0 开始的整数数组 nums 。在一步操作中,你可以执行以下步骤:
从 nums 选出 两个 相等的 整数
从 nums 中移除这两个整数,形成一个 数对
请你在 nums 上多次执行此操作直到无法继续执行。
返回一个下标从 0 开始、长度为 2 的整数数组 answer 作为答案,其中 answer[0] 是形成的数对数目,answer[1] 是对 nums 尽可能执行上述操作后剩下的整数数目。
示例 1:
输入:nums = [1,3,2,1,3,2,2]
输出:[3,1]
解释:
nums[0] 和 nums[3] 形成一个数对,并从 nums 中移除,nums = [3,2,3,2,2] 。
nums[0] 和 nums[2] 形成一个数对,并从 nums 中移除,nums = [2,2,2] 。
nums[0] 和 nums[1] 形成一个数对,并从 nums 中移除,nums = [2] 。
无法形成更多数对。总共形成 3 个数对,nums 中剩下 1 个数字。
1 <= nums.length <= 100
0 <= nums[i] <= 100
解法一:由于0 <= nums[i] <= 100,因此可用长度为101的哈希表统计每个数出现过多少次,然后遍历一遍哈希表即可:
class Solution {
public:
vector<int> numberOfPairs(vector<int>& nums) {
vector<int> hashTbl(101, 0);
for (int i : nums) {
++hashTbl[i];
}
int ans = 0;
for (int i : hashTbl) {
ans += i / 2;
}
return {ans, static_cast<int>(nums.size() - 2 * ans)};
}
};
如果输入数组长度为n,每个数组元素范围为m,此算法时间复杂度为O(n+m),空间复杂度为O(m)。
解法二:如果每个数组的元素范围不定,可以用map存下每个数字的出现次数:
class Solution {
public:
vector<int> numberOfPairs(vector<int>& nums) {
unordered_map<int, int> numToCount;
for (int i : nums) {
++numToCount[i];
}
int ans = 0;
for (pair<const int, int> &numAndCount : numToCount) {
ans += numAndCount.second / 2;
}
return {ans, static_cast<int>(nums.size() - 2 * ans)};
}
};
如果输入数组长度为n,此算法时间复杂度为O(n),空间复杂度为O(n)。
浙公网安备 33010602011771号