力扣1703.得到连续K个1的最少相邻交换次数

力扣1703.得到连续K个1的最少相邻交换次数

  • 中位数贪心 + 前缀和

    • 设pi = qi - i ,则问题变为所有pi到x的距离和的最小值

    • 同时x应取pi的中位数

    • 因此枚举i作为最左端的1,求最小值即可

    • 推出距离和公式:s[i] + s[i+k] - s[i+k/2]2 - p[i+k/2](k%2)

cpp 复制代码
  class Solution {
  public:
      int minMoves(vector<int>& nums, int k) {
          vector<int> p;
          for(int i=0;i<nums.size();i++)
              if(nums[i])
                  p.push_back(i-p.size());
          int m = p.size(),s[m+1];
          s[0] = 0;
          for(int i=1;i<=m;i++)
              s[i] = s[i-1] + p[i-1];
          int ans = INT_MAX;
          for(int i=0;i<=m-k;i++)
              ans = min(ans,s[i] + s[i+k] - s[i+k/2]*2 - p[i+k/2]*(k%2));
          return ans;
      }
  };
相关推荐
Fanxt_Ja1 小时前
【LeetCode】算法详解#15 ---环形链表II
数据结构·算法·leetcode·链表
侃侃_天下1 小时前
最终的信号类
开发语言·c++·算法
茉莉玫瑰花茶1 小时前
算法 --- 字符串
算法
博笙困了1 小时前
AcWing学习——差分
c++·算法
NAGNIP2 小时前
认识 Unsloth 框架:大模型高效微调的利器
算法
NAGNIP2 小时前
大模型微调框架之LLaMA Factory
算法
echoarts2 小时前
Rayon Rust中的数据并行库入门教程
开发语言·其他·算法·rust
Python技术极客2 小时前
一款超好用的 Python 交互式可视化工具,强烈推荐~
算法
徐小夕2 小时前
花了一天时间,开源了一套精美且支持复杂操作的表格编辑器tablejs
前端·算法·github
小刘鸭地下城2 小时前
深入浅出链表:从基础概念到核心操作全面解析
算法