依舊二分,鏈接如下3488. 距離最小相等元素查詢
看題目是個循環數組,記得當時做過一道什么題也是循環數組,就想著直接數組復制一下,然后跟上一道題一樣,用hashmap來存儲value的值以及value對應下標的vector。
和靈神的思路類似,都是比較左右元素,不過我沒有使用哨兵。
然后比較vector對應的idx到left元素的下標差以及到right元素的下標差,這里很容易出錯。昨天debug了好久找不到問題在哪,后來發現是把nums數組的長度和下標所在的vector數組的長度搞混到一起了。最好還是用靈神的做法,我的這種做法太笨重了,容易出錯。
C++代碼如下
class Solution {
public:vector<int> solveQueries(vector<int>& nums, vector<int>& queries) {int m = nums.size();vector<int> numss = nums;numss.insert(numss.end(),nums.begin(),nums.end());unordered_map<int,vector<int>> hmap;for(int i = 0 ; i < numss.size();i++)hmap[numss[i]].push_back(i);vector<int> ans;for( int i = 0; i<queries.size();i++){auto it = hmap.find(numss[queries[i]]);if(it != hmap.end()&& it->second.size()!=2){auto& s = it->second;int n = s.size()/2;int r = *upper_bound(s.begin(),s.end(),queries[i])-queries[i];int l ;auto l_ptr = lower_bound(s.begin(),s.end(),queries[i]);if(l_ptr == s.begin())l = abs(*--(l_ptr+n)-m-*l_ptr);elsel = abs(*(--l_ptr)-queries[i]);ans.push_back(min(r,l));}elseans.push_back(-1);}return ans;}
};