给你一个下标从 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 。
输入: 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 。
输入: 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 <= 105
-109 <= nums[i] <= 109
impl Solution {
pub fn find_smallest_integer(nums: Vec<i32>, value: i32) -> i32 {
let mut count = vec![0; value as usize];
for &num in &nums {
count[num.rem_euclid(value) as usize] += 1;
}
for num in 0..=nums.len() as i32 {
if count[num.rem_euclid(value) as usize] == 0 {
return num;
}
count[num.rem_euclid(value) as usize] -= 1;
}
unreachable!();
}
}