1769.移动所有球到每个盒子所需的最小操作数

1769.移动所有球到每个盒子所需的最小操作数

Category	Difficulty	Likes	Dislikes
algorithms	Medium (85.22%)	65	-

Tags
Companies
有 n 个盒子。给你一个长度为 n 的二进制字符串 boxes ，其中 boxes[i] 的值为 ‘0’ 表示第 i 个盒子是 空 的，而 boxes[i] 的值为 ‘1’ 表示盒子里有 一个 小球。
在一步操作中，你可以将 一个 小球从某个盒子移动到一个与之相邻的盒子中。第 i 个盒子和第 j 个盒子相邻需满足 abs(i - j) == 1 。注意，操作执行后，某些盒子中可能会存在不止一个小球。
返回一个长度为 n 的数组 answer ，其中 answer[i] 是将所有小球移动到第 i 个盒子所需的 最小 操作数。
每个 answer[i] 都需要根据盒子的 初始状态 进行计算。

示例 1：
输入：boxes = “110” 输出：[1,1,3] 解释：每个盒子对应的最小操作数如下： 1) 第 1 个盒子：将一个小球从第 2 个盒子移动到第 1 个盒子，需要 1 步操作。 2) 第 2 个盒子：将一个小球从第 1 个盒子移动到第 2 个盒子，需要 1 步操作。 3) 第 3 个盒子：将一个小球从第 1 个盒子移动到第 3 个盒子，需要 2 步操作。将一个小球从第 2 个盒子移动到第 3 个盒子，需要 1 步操作。共计 3 步操作。
示例 2：
输入：boxes = “001011” 输出：[11,8,5,4,3,4]

提示：

• n == boxes.length
• 1 <= n <= 2000
• boxes[i] 为 ‘0’ 或 ‘1’

---

Discussion | Solution

思路

记录所有为 1 的下标，统计到当前下标的所有为 1 下标元素的和即可

解答

/*
 * @lc app=leetcode.cn id=1769 lang=javascript
 *
 * [1769] 移动所有球到每个盒子所需的最小操作数
 */

// @lc code=start
/**
 * @param {string} boxes
 * @return {number[]}
 */
var minOperations = function (boxes) {
  const index = [];
  for (let i = 0; i < boxes.length; i++) {
    const box = boxes[i];
    if (box === "1") index.push(i);
  }
  return boxes.split("").map((_, i) => {
    return index.reduce((sum, num) => sum + Math.abs(num - i), 0);
  });
};
// @lc code=end