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

示例 2：

输入：nums = [1,-10,7,13,6,8], value = 7
输出：2
解释：执行下述操作可以得到这一结果：
- nums[2] 减去 value 一次，nums = [1,-10,0,13,6,8]
nums 的 MEX 是 2 。可以证明 2 是可以取到的最大 MEX 。

提示：

• 1 <= nums.length, value <= 10^5
• -10^9 <= nums[i] <= 10^9

分析：题目实际上是问，对数组进行若干次操作后，能得到的最小未出现的非负整数的最大值。对于 n 个数组成的数组，进行题目给出的操作后，最好的情况是得到一个 [0,1,2,...,n-1] 的整数序列，此时答案为 n。

遍历原数组，求出 nums[i] 的所有余数，并记录 每个余数的最小出现次数 × value + 该余数 的值，这些值组成的数组就是能得到的所有非负整数，此时未出现的非负整数即为答案。

int findSmallestInteger(int* nums, int numsSize, int value) {int ans=0;int flag[numsSize+value+5],cnt[numsSize+value+5];for(int i=0;i<numsSize+value;++i)flag[i]=cnt[i]=0;for(int i=0;i<numsSize;++i){nums[i]=(nums[i]%value+value)%value;int idx=nums[i]+value*cnt[nums[i]];if(idx<numsSize)flag[idx]=1,cnt[nums[i]]++;}for(int i=0;i<numsSize;++i)if(!flag[i])return i;return numsSize;
}