Leetcode每日一题 —— 2615. 等值距离和
- 内容介绍
- 文章标签
- 相关推荐
2615. 等值距离和 - 力扣（LeetCode）
2615. 等值距离和 - 给你一个下标从 0 开始的整数数组 nums 。现有一个长度等于 nums.length 的数组 arr 。对于满足 nums[j] == nums[i] 且 j != i 的所有 j ，arr[i] 等于所有 |i - j| 之和。如果不存在这样的 j ，则令 arr[i] 等于 0 。 返回数组 arr 。 示例 1： 输入：nums = [1,3,1,1,2] 输出：[5,0,3,4,0] 解释： i = 0 ，nums[0] == nums[2]...
思路
也就是对于每个下标 i 对应的数字 nums[i]，要找到与其相等的所有其他数字 nums[j]，累加 abs(i-j)。
题目输入规模可能很大，时间复杂度要控制在 O(n\log n) 以下。
咱们在意的是某个数字 nums[i] 在数组内出现的所有下标，但如果直接按数字分组维护列表，扫描所耗费的时间从总体上看来是难以接受的。
既要涉及求和，而且还需要尽量是线性时间复杂度，能猜到可能能用到前缀和思想。
对于一个下标 i，我可以计算前缀中所有 i-j 之和，再计算其后缀中所有 j-i 之和，这个位置的前缀加后缀就是最终结果 arr[i] 了。
2615. 等值距离和 - 力扣（LeetCode）
2615. 等值距离和 - 给你一个下标从 0 开始的整数数组 nums 。现有一个长度等于 nums.length 的数组 arr 。对于满足 nums[j] == nums[i] 且 j != i 的所有 j ，arr[i] 等于所有 |i - j| 之和。如果不存在这样的 j ，则令 arr[i] 等于 0 。 返回数组 arr 。 示例 1： 输入：nums = [1,3,1,1,2] 输出：[5,0,3,4,0] 解释： i = 0 ，nums[0] == nums[2]...
思路
也就是对于每个下标 i 对应的数字 nums[i]，要找到与其相等的所有其他数字 nums[j]，累加 abs(i-j)。
题目输入规模可能很大，时间复杂度要控制在 O(n\log n) 以下。
咱们在意的是某个数字 nums[i] 在数组内出现的所有下标，但如果直接按数字分组维护列表，扫描所耗费的时间从总体上看来是难以接受的。
既要涉及求和，而且还需要尽量是线性时间复杂度，能猜到可能能用到前缀和思想。
对于一个下标 i，我可以计算前缀中所有 i-j 之和，再计算其后缀中所有 j-i 之和，这个位置的前缀加后缀就是最终结果 arr[i] 了。