给你一个下标从 0 开始的数组 nums ,数组长度为 n 。
nums 的 不同元素数目差 数组可以用一个长度为 n 的数组 diff 表示,其中 diff[i] 等于前缀 nums[0, …, i] 中不同元素的数目 减去 后缀 nums[i + 1, …, n - 1] 中不同元素的数目。
返回 nums 的 不同元素数目差 数组。
注意 nums[i, …, j] 表示 nums 的一个从下标 i 开始到下标 j 结束的子数组(包含下标 i 和 j 对应元素)。特别需要说明的是,如果 i > j ,则 nums[i, …, j] 表示一个空子数组。
示例 1:
输入:nums = [1,2,3,4,5]
输出:[-3,-1,1,3,5]
解释:
对于 i = 0,前缀中有 1 个不同的元素,而在后缀中有 4 个不同的元素。因此,diff[0] = 1 - 4 = -3 。
对于 i = 1,前缀中有 2 个不同的元素,而在后缀中有 3 个不同的元素。因此,diff[1] = 2 - 3 = -1 。
对于 i = 2,前缀中有 3 个不同的元素,而在后缀中有 2 个不同的元素。因此,diff[2] = 3 - 2 = 1 。
对于 i = 3,前缀中有 4 个不同的元素,而在后缀中有 1 个不同的元素。因此,diff[3] = 4 - 1 = 3 。
对于 i = 4,前缀中有 5 个不同的元素,而在后缀中有 0 个不同的元素。因此,diff[4] = 5 - 0 = 5 。
示例 2:
输入:nums = [3,2,3,4,2]
输出:[-2,-1,0,2,3]
解释:
对于 i = 0,前缀中有 1 个不同的元素,而在后缀中有 3 个不同的元素。因此,diff[0] = 1 - 3 = -2 。
对于 i = 1,前缀中有 2 个不同的元素,而在后缀中有 3 个不同的元素。因此,diff[1] = 2 - 3 = -1 。
对于 i = 2,前缀中有 2 个不同的元素,而在后缀中有 2 个不同的元素。因此,diff[2] = 2 - 2 = 0 。
对于 i = 3,前缀中有 3 个不同的元素,而在后缀中有 1 个不同的元素。因此,diff[3] = 3 - 1 = 2 。
对于 i = 4,前缀中有 3 个不同的元素,而在后缀中有 0 个不同的元素。因此,diff[4] = 3 - 0 = 3 。
提示:
1 <= n == nums.length <= 50
1 <= nums[i] <= 50
解:
两张普通的hash表,先默认全在右侧,记录下当前数组中的不同元素的个数,与每个元素的个数。
再从头开始遍历,只需要每次判断当前元素的情况,在进行数值的改变。
int* distinctDifferenceArray(int* nums, int numsSize, int* returnSize){
if (numsSize == 0) {
return (int*)malloc(0);
}
int* res = (int*)malloc(numsSize*sizeof(int));
int ld=0,rd=0;
int lcnt[51],rcnt[51];
memset(lcnt,0,sizeof(lcnt));
memset(rcnt,0,sizeof(rcnt));
for (int i = 0;i < numsSize;i++) {
if(rcnt[nums[i]] == 0) {
rd++;
}
rcnt[nums[i]]++;
}
for(int i = 0;i < numsSize;i++) {
if(lcnt[nums[i]] == 0) {
ld++;
}
lcnt[nums[i]]++;
rcnt[nums[i]]--;
if(rcnt[nums[i]] <= 0) {
rd--;
}
res[i]=ld-rd;
}
*returnSize = numsSize;
return res;
}