力扣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;
      }
  };
相关推荐
SsummerC1 小时前
【leetcode100】零钱兑换Ⅱ
数据结构·python·算法·leetcode·动态规划
好易学·数据结构2 小时前
可视化图解算法:二叉树的最大深度(高度)
数据结构·算法·二叉树·最大高度·最大深度·二叉树高度·二叉树深度
程序员-King.2 小时前
day47—双指针-平方数之和(LeetCode-633)
算法·leetcode
阳洞洞2 小时前
leetcode 1035. Uncrossed Lines
算法·leetcode·动态规划·子序列问题
小鹿鹿啊3 小时前
C语言编程--15.四数之和
c语言·数据结构·算法
rigidwill6663 小时前
LeetCode hot 100—最长有效括号
数据结构·c++·算法·leetcode·职场和发展
wuqingshun3141594 小时前
蓝桥杯17. 机器人塔
c++·算法·职场和发展·蓝桥杯·深度优先
图灵科竞社资讯组4 小时前
图论基础:图存+记忆化搜索
算法·图论
chuxinweihui5 小时前
数据结构——栈与队列
c语言·开发语言·数据结构·学习·算法·链表