题1
描述
给定整数数组 A,每次 move 操作将会选择任意 A[i],并将其递增 1。
返回使 A 中的每个值都是唯一的最少操作次数。
示例 1:
输入:[1,2,2]
输出:1
解释:经过一次 move 操作,数组将变为 [1, 2, 3]。
示例 2:
输入:[3,2,1,2,1,7]
输出:6
解释:经过 6 次 move 操作,数组将变为 [3, 4, 1, 2, 5, 7]。
可以看出 5 次或 5 次以下的 move 操作是不能让数组的每个值唯一的。
提示:
0 <= A.length <= 40000
0 <= A[i] < 40000
题解1 排序法
思路:先用Arrays.sort()对数组进行排序,遍历排序后的数组,如果后一个元素小于前一个元素,则后一个元素+1,移动的步数等于当前位置变更后的值与数组原先值之差,所有差之和即为总的移动步数。
class Solution {public int minIncrementForUnique(int[] A) {Arrays.sort(A);int count = 0;for(int i = 1; i < A.length; i++){if(A[i] <= A[i-1]){int temp = A[i];A[i] = A[i-1] + 1;count += A[i] - temp;}}return count;}
}
题解2 计数法
思路:我拿到题目后的第一种思路是计数+暴力法,在小数据的测试用例下都可以得到正确答案,思路也没有问题,疑惑的是无法通过超级大且多的数据。后来看了官方的解答知道了答案。
虽然 A[i] 的范围为 [0, 40000),但我们有可能会将数据递增到 40000 的两倍 80000。这是因为在最坏情况下,数组 A 中有 40000 个 40000,这样要使得数组值唯一,需要将其递增为 [40000, 40001, …, 79999],因此用来统计的数组需要开到 80000。
当我们找到一个没有出现过的数的时候,将之前某个重复出现的数增加成这个没有出现过的数。注意,这里 「之前某个重复出现的数」 是可以任意选择的,它并不会影响最终的答案,因为将 P 增加到 X 并且将 Q 增加到 Y,与将 P 增加到 Y 并且将 Q 增加到 X 都需要进行 (X + Y) - (P + Q) 次操作。例如当数组 A 为 [1, 1, 1, 1, 3, 5] 时,我们发现有 3 个重复的 1,且没有出现过 2,4 和 6,因此一共需要进行 (2 + 4 + 6) - (1 + 1 + 1) = 9 次操作。
class Solution {public int minIncrementForUnique(int[] A) {int[] count = new int[80000];for (int x: A) count[x]++;int ans = 0, taken = 0;for (int x = 0; x < 80000; ++x) {if (count[x] >= 2) {taken += count[x] - 1;ans -= x * (count[x] - 1);}else if (taken > 0 && count[x] == 0) {taken--;ans += x;}}return ans;}
}
题2
描述
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
示例 1:
给定 nums = [3,2,2,3], val = 3,
函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。
你不需要考虑数组中超出新长度后面的元素。
示例 2:
给定 nums = [0,1,2,2,3,0,4,2], val = 2,
函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。
注意这五个元素可为任意顺序。
你不需要考虑数组中超出新长度后面的元素。
说明:
为什么返回数值是整数,但输出的答案是数组呢?
请注意,输入数组是以「引用」方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。
你可以想象内部操作如下:
// nums 是以“引用”方式传递的。也就是说,不对实参作任何拷贝
int len = removeElement(nums, val);
// 在函数里修改输入数组对于调用者是可见的。
// 根据你的函数返回的长度, 它会打印出数组中 该长度范围内 的所有元素。
for (int i = 0; i < len; i++) {
print(nums[i]);
}
题解
思路:类似于昨天的题2,用i指针来移动,用j指针来指向与val相同的元素位置,用i指针指向的元素向前覆盖j位置的值
class Solution {public int removeElement(int[] nums, int val) {// 使用双指针if (nums == null || nums.length == 0)return 0;int j = 0;for(int i = 0; i < nums.length; i++){if(nums[i] != val){nums[j] = nums[i];j++;}}return j;}
}