力扣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;
      }
  };
相关推荐
岁忧1 分钟前
(LeetCode 面试经典 150 题) 200. 岛屿数量(深度优先搜索dfs || 广度优先搜索bfs)
java·c++·leetcode·面试·go·深度优先
一枝小雨9 分钟前
【OJ】C++ vector类OJ题
数据结构·c++·算法·leetcode·oj题
Tisfy20 分钟前
LeetCode 3516.找到最近的人:计算绝对值大小
数学·算法·leetcode·题解
自信的小螺丝钉21 分钟前
Leetcode 206. 反转链表 迭代/递归
算法·leetcode·链表
黑色的山岗在沉睡1 小时前
LeetCode 189. 轮转数组
java·算法·leetcode
墨染点香1 小时前
LeetCode 刷题【65. 有效数字】
算法·leetcode·职场和发展
Tisfy1 小时前
LeetCode 3027.人员站位的方案数 II:简单一个排序O(n^2)——ASCII图解
leetcode·题解·思维·排序·hard
源代码•宸2 小时前
Leetcode—2749. 得到整数零需要执行的最少操作数【中等】(__builtin_popcountl)
c++·经验分享·算法·leetcode·位运算
用户4822137167752 小时前
深度学习——AlexNet网络结构
算法