文章目錄

• • 1. 題目
  • 2. 解題

1. 題目

給你一個下標(biāo)從 0 開始、由 n 個整數(shù)組成的數(shù)組 arr 。

arr 中兩個元素的 間隔 定義為它們下標(biāo)之間的 絕對差 。更正式地，arr[i] 和 arr[j] 之間的間隔是 |i - j| 。

返回一個長度為 n 的數(shù)組 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] 解釋： - 下標(biāo) 0 ：另一個 2 在下標(biāo) 4 ，|0 - 4| = 4 - 下標(biāo) 1 ：另一個 1 在下標(biāo) 3 ，|1 - 3| = 2 - 下標(biāo) 2 ：另兩個 3 在下標(biāo) 5 和 6 ，|2 - 5| + |2 - 6| = 7 - 下標(biāo) 3 ：另一個 1 在下標(biāo) 1 ，|3 - 1| = 2 - 下標(biāo) 4 ：另一個 2 在下標(biāo) 0 ，|4 - 0| = 4 - 下標(biāo) 5 ：另兩個 3 在下標(biāo) 2 和 6 ，|5 - 2| + |5 - 6| = 4 - 下標(biāo) 6 ：另兩個 3 在下標(biāo) 2 和 5 ，|6 - 2| + |6 - 5| = 5示例 2： 輸入：arr = [10,5,10,10] 輸出：[5,0,3,4] 解釋： - 下標(biāo) 0 ：另兩個 10 在下標(biāo) 2 和 3 ，|0 - 2| + |0 - 3| = 5 - 下標(biāo) 1 ：只有這一個 5 在數(shù)組中，所以到相同元素的間隔之和是 0 - 下標(biāo) 2 ：另兩個 10 在下標(biāo) 0 和 3 ，|2 - 0| + |2 - 3| = 3 - 下標(biāo) 3 ：另兩個 10 在下標(biāo) 0 和 2 ，|3 - 0| + |3 - 2| = 4提示： n == arr.length 1 <= n <= 10^5 1 <= arr[i] <= 10^5

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/intervals-between-identical-elements
著作權(quán)歸領(lǐng)扣網(wǎng)絡(luò)所有。商業(yè)轉(zhuǎn)載請聯(lián)系官方授權(quán)，非商業(yè)轉(zhuǎn)載請注明出處。

2. 解題

• 按照數(shù)字分組
• 對每組數(shù)字的 下標(biāo)求前綴和，因為對 i 位置前面的可以拆成 i-i前，后面的可以拆成 i后-i
• 利用前綴和獲取同符號的區(qū)間的和

class Solution { public:vector<long long> getDistances(vector<int>& arr) {unordered_map<int,vector<long long>> pos; // 相同數(shù)字的位置for(int i = 0; i < arr.size(); ++i)pos[arr[i]].push_back(i); // 數(shù)字的各個位置vector<long long> ans(arr.size(), 0);for(auto& p : pos){vector<long long> pp = p.second;for(int i = 1; i < pp.size(); ++i)pp[i] += pp[i-1]; // 各個位置的前綴和int lnum = 0, rnum = p.second.size();for(int i = 0; i < p.second.size(); ++i){lnum = i; // i 左邊有的同數(shù)字的 個數(shù)rnum = p.second.size()-i-1; // 右側(cè)有的個數(shù)long long idx = p.second[i]; // 對應(yīng)原來數(shù)組中的位置ans[idx] = pp.back()-pp[i] - idx*rnum + idx*lnum - (i>0 ? pp[i-1] : 0);// a1,a2,... i, ... b1, b2 后半段的都可以拆成 b1-i, b2-i, 個數(shù) rnum個// 前半段 都可以拆成 i-a1, i-a2,... 個數(shù) lnum個// a1, a2, ... 的和用前綴和直接獲取// b1, b2, ... 的和用前綴和做差獲取}}return ans;} };

340 ms 131.3 MB C++

---

我的CSDN博客地址 https://michael.blog.csdn.net/

長按或掃碼關(guān)注我的公眾號（Michael阿明），一起加油、一起學(xué)習(xí)進(jìn)步！

總結(jié)

以上是生活随笔為你收集整理的LeetCode 2121. 相同元素的间隔之和（前缀和）的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。