5965:相同元素的间隔之和

This commit is contained in:
轩辕龙儿 2021-12-26 14:45:25 +08:00
parent 2197f49403
commit 283672d2d5
2 changed files with 135 additions and 0 deletions

View File

@ -0,0 +1,90 @@
//给你一个下标从 0 开始 n 个整数组成的数组 arr
//
// arr 中两个元素的 间隔 定义为它们下标之间的 绝对差 更正式地arr[i] arr[j] 之间的间隔是 |i - j|
//
// 返回一个长度为 n 的数组 intervals 其中 intervals[i] arr[i] arr 中每个相同元素 arr[i] 的值相同
// 间隔之和
//
// 注意|x| x 的绝对值
//
//
//
// 示例 1
//
// 输入arr = [2,1,3,1,2,3,3]
//输出[4,2,7,2,4,4,5]
//解释
//- 下标 0 另一个 2 在下标 4 |0 - 4| = 4
//- 下标 1 另一个 1 在下标 3 |1 - 3| = 2
//- 下标 2 另两个 3 在下标 5 6 |2 - 5| + |2 - 6| = 7
//- 下标 3 另一个 1 在下标 1 |3 - 1| = 2
//- 下标 4 另一个 2 在下标 0 |4 - 0| = 4
//- 下标 5 另两个 3 在下标 2 6 |5 - 2| + |5 - 6| = 4
//- 下标 6 另两个 3 在下标 2 5 |6 - 2| + |6 - 5| = 5
//
//
// 示例 2
//
// 输入arr = [10,5,10,10]
//输出[5,0,3,4]
//解释
//- 下标 0 另两个 10 在下标 2 3 |0 - 2| + |0 - 3| = 5
//- 下标 1 只有这一个 5 在数组中所以到相同元素的间隔之和是 0
//- 下标 2 另两个 10 在下标 0 3 |2 - 0| + |2 - 3| = 3
//- 下标 3 另两个 10 在下标 0 2 |3 - 0| + |3 - 2| = 4
//
//
//
//
// 提示
//
//
// n == arr.length
// 1 <= n <= 10
// 1 <= arr[i] <= 10
//
// 👍 9 👎 0
package leetcode.editor.cn;
import java.util.*;
//5965:相同元素的间隔之和
class IntervalsBetweenIdenticalElements {
public static void main(String[] args) {
//测试代码
Solution solution = new IntervalsBetweenIdenticalElements().new Solution();
}
//力扣代码
//leetcode submit region begin(Prohibit modification and deletion)
class Solution {
public long[] getDistances(int[] arr) {
Map<Integer, List<Integer>> map = new HashMap<>();
for (int i = 0; i < arr.length; i++) {
List<Integer> list = map.getOrDefault(arr[i], new ArrayList<>());
list.add(i);
map.put(arr[i], list);
}
long[] arrs = new long[arr.length];
for (int key : map.keySet()) {
List<Integer> list = map.get(key);
if (list.size() == 1) {
arrs[list.get(0)] = 0;
} else {
long[] temp = new long[list.size() + 1];
Collections.sort(list);
for (int i = 0; i < list.size(); i++) {
temp[i + 1] = temp[i] + list.get(i);
}
for (int i = 0; i < list.size(); i++) {
arrs[list.get(i)] = temp[temp.length - 1] - temp[i + 1] - temp[i] - ((long) temp.length - 2 * i - 2) * list.get(i);
}
}
}
return arrs;
}
}
//leetcode submit region end(Prohibit modification and deletion)
}

View File

@ -0,0 +1,45 @@
<p>给你一个下标从 <strong>0</strong> 开始、由 <code>n</code> 个整数组成的数组 <code>arr</code></p>
<p><code>arr</code> 中两个元素的 <strong>间隔</strong> 定义为它们下标之间的 <strong>绝对差</strong> 。更正式地,<code>arr[i]</code><code>arr[j]</code> 之间的间隔是 <code>|i - j|</code></p>
<p>返回一个长度为 <code>n</code> 的数组&nbsp;<code>intervals</code> ,其中 <code>intervals[i]</code><em> </em><code>arr[i]</code><em> </em><em> </em><code>arr</code><em> </em>中每个相同元素(与 <code>arr[i]</code> 的值相同)的 <strong>间隔之和</strong> <em></em></p>
<p><strong>注意:</strong><code>|x|</code><code>x</code> 的绝对值。</p>
<p>&nbsp;</p>
<p><strong>示例 1</strong></p>
<pre><strong>输入:</strong>arr = [2,1,3,1,2,3,3]
<strong>输出:</strong>[4,2,7,2,4,4,5]
<strong>解释:</strong>
- 下标 0 :另一个 2 在下标 4 |0 - 4| = 4
- 下标 1 :另一个 1 在下标 3 |1 - 3| = 2
- 下标 2 :另两个 3 在下标 5 和 6 |2 - 5| + |2 - 6| = 7
- 下标 3 :另一个 1 在下标 1 |3 - 1| = 2
- 下标 4 :另一个 2 在下标 0 |4 - 0| = 4
- 下标 5 :另两个 3 在下标 2 和 6 |5 - 2| + |5 - 6| = 4
- 下标 6 :另两个 3 在下标 2 和 5 |6 - 2| + |6 - 5| = 5
</pre>
<p><strong>示例 2</strong></p>
<pre><strong>输入:</strong>arr = [10,5,10,10]
<strong>输出:</strong>[5,0,3,4]
<strong>解释:</strong>
- 下标 0 :另两个 10 在下标 2 和 3 |0 - 2| + |0 - 3| = 5
- 下标 1 :只有这一个 5 在数组中,所以到相同元素的间隔之和是 0
- 下标 2 :另两个 10 在下标 0 和 3 |2 - 0| + |2 - 3| = 3
- 下标 3 :另两个 10 在下标 0 和 2 |3 - 0| + |3 - 2| = 4
</pre>
<p>&nbsp;</p>
<p><strong>提示:</strong></p>
<ul>
<li><code>n == arr.length</code></li>
<li><code>1 &lt;= n &lt;= 10<sup>5</sup></code></li>
<li><code>1 &lt;= arr[i] &lt;= 10<sup>5</sup></code></li>
</ul>
<div><li>👍 9</li><li>👎 0</li></div>