2598. 执行操作后的最大 MEX——模运算

题目

给你一个下标从 0 开始的整数数组 nums 和一个整数 value 。

在一步操作中，你可以对 nums 中的任一元素加上或减去 value 。

例如，如果 nums = [1,2,3] 且 value = 2 ，你可以选择 nums[0] 减去 value ，得到 nums = [-1,2,3] 。
数组的 MEX (minimum excluded) 是指其中数组中缺失的最小非负整数。

例如，[-1,2,3] 的 MEX 是 0 ，而 [1,0,3] 的 MEX 是 2 。
返回在执行上述操作 任意次 后，nums 的最大 MEX 。

示例 1：
输入：nums = [1,-10,7,13,6,8], value = 5
输出：4
解释：执行下述操作可以得到这一结果：

• nums[1] 加上 value 两次，nums = [1,0,7,13,6,8]
• nums[2] 减去 value 一次，nums = [1,0,2,13,6,8]
• nums[3] 减去 value 两次，nums = [1,0,2,3,6,8]
  nums 的 MEX 是 4 。可以证明 4 是可以取到的最大 MEX 。

分析

模运算性质

代码

class Solution {
public:
    int findSmallestInteger(vector<int>& nums, int value) {
        unordered_map<int, int> um;
        int n = nums.size();
        for (int i = 0; i < n; i++) {
            um[(nums[i] % value + value) % value]++; // 为了正负均可以
        }
        int res = 0;
        while(um[res % value]-- > 0) {res++;}
        return res;
    }
};