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)。

posted @ 2023-02-01 22:13  epiphanyy  阅读(12)  评论(0)    收藏  举报  来源