中位数贪心|前缀和_距离和ret=l+r_1

数据按排序后,处于中间位置的数值(数据个数奇数取正中间,偶数取中间两个的平均值)

lc462

class Solution {

public:

int minMoves2(vector<int>& nums) {

int m = nums.size() / 2;

++ranges::nth_element(nums, nums.begin() + m);++

++int mid = nums[m];++

int ans = 0;

for (int x : nums) {

++ans += abs(x - mid);++

}

return ans;

}

};

lc2033

class Solution {

/*

输入:grid = [[2,4],[6,8]], x = 2

输出:4

*/

public:

int minOperations(vector<vector<int>>& grid, int x)

{

vector<int> a;

for(auto& g:grid)

{

for(auto& e:g)

a.push_back(e);

}

int n=a.size();

int m=n/2;

++ranges::nth_element(a,a.begin()+m);
int mid=a[m];
++

int ret=0;

for(auto& b:a)

{

int d=abs(b-mid);

if(d%x!=0)

return -1;

ret+=(d/x);

}

return ret;

}

};

lc2602

排序+前缀和+二分

++right = sum[n] - sum[j] - q * (n - j); // 绿色面积++

使数组元素 全部等于某个元素的最小操作数

从q*n优化为(q+n)*log n

class Solution {

public:

vector<long long> minOperations(vector<int>& nums, vector<int>& queries) {

ranges::sort(nums);

int n = nums.size();

vector<long long> sum(n + 1); // 前缀和

for (int i = 0; i < n; i++) {

sum[i + 1] = sum[i] + nums[i];

}

int m = queries.size();

vector<long long> ans(m);

for (int i = 0; i < m; i++) {

int q = queries[i];

long long j = ranges::lower_bound(nums, q) - nums.begin();

long long left = q * j - sum[j]; // 蓝色面积

long long ++right = sum[n] - sum[j] - q * (n - j); // 绿色面积++

ans[i] = left + right;

}

return ans;

}

};

lower_bound 返回有序容器中++第一个≥目标值的元素迭代器++

upper_bound 返回第一个>目标值的元素迭代器(均基于二分查找,仅匹配条件不同)

lc2615

hash分组+前缀和求距离和

++ret[q] = left + right;++

class Solution {

typedef long long ll;

public:

vector<long long> distance(vector<int>& nums) {

int n=nums.size();

unordered_map<int,vector<int>> hash;

vector<ll> ret(n);

for(int i=0;i<n;i++)

{

hash[nums[i]].push_back(i);

}

for(auto& [_,b]:hash)

{

int m=b.size();

if(m==1)

{

ret[b[0]]=0;

continue;

}

vector<ll> p(m+1);

for(int i=1;i<=m;i++)

p[i]=p[i-1]+b[i-1];

for(int i=0;i<m;i++)

{

int q=b[i];

ll left = (ll)q*i-p[i]; // blue

ll right = p[m] - p[i] - (ll)q * (m - i); //green

++ret[q] = left + right;++

}

}

return ret;

}

};

相关推荐
MM_MS6 小时前
Halcon变量控制类型、数据类型转换、字符串格式化、元组操作
开发语言·人工智能·深度学习·算法·目标检测·计算机视觉·视觉检测
独自破碎E6 小时前
【二分法】寻找峰值
算法
mit6.8247 小时前
位运算|拆分贪心
算法
ghie90907 小时前
基于MATLAB的TLBO算法优化实现与改进
开发语言·算法·matlab
恋爱绝缘体17 小时前
2020重学C++重构你的C++知识体系
java·开发语言·c++·算法·junit
wuk9987 小时前
VSC优化算法MATLAB实现
开发语言·算法·matlab
Z1Jxxx7 小时前
加密算法加密算法
开发语言·c++·算法
乌萨奇也要立志学C++8 小时前
【洛谷】递归初阶 三道经典递归算法题(汉诺塔 / 占卜 DIY/FBI 树)详解
数据结构·c++·算法
vyuvyucd8 小时前
C++引用:高效编程的别名利器
算法
鱼跃鹰飞8 小时前
Leetcode1891:割绳子
数据结构·算法