题目描述:
有 n 个盒子。给你一个长度为 n 的二进制字符串 boxes ,其中 boxes[i] 的值为 ‘0’ 表示第 i 个盒子是 空 的,而 boxes[i] 的值为 ‘1’ 表示盒子里有 一个 小球。
在一步操作中,你可以将 一个 小球从某个盒子移动到一个与之相邻的盒子中。第 i 个盒子和第 j 个盒子相邻需满足 abs(i - j) == 1 。注意,操作执行后,某些盒子中可能会存在不止一个小球。
返回一个长度为 n 的数组 answer ,其中 answer[i] 是将所有小球移动到第 i 个盒子所需的 最小 操作数。
每个 answer[i] 都需要根据盒子的 初始状态 进行计算。
提示:
- n == boxes.length
- 1 <= n <= 2000
- boxes[i] 为 ‘0’ 或 ‘1’
示例 1:
输入:boxes = “110”
输出:[1,1,3]
解释:每个盒子对应的最小操作数如下:
- 第 1 个盒子:将一个小球从第 2 个盒子移动到第 1 个盒子,需要 1 步操作。
- 第 2 个盒子:将一个小球从第 1 个盒子移动到第 2 个盒子,需要 1 步操作。
- 第 3 个盒子:将一个小球从第 1 个盒子移动到第 3 个盒子,需要 2 步操作。将一个小球从第 2 个盒子移动到第 3 个盒子,需要 1 步操作。共计 3 步操作。
示例 2:
输入:boxes = “001011”
输出:[11,8,5,4,3,4]
代码如下:
class Solution {
public int[] minOperations(String boxes) {
int n = boxes.length();
char[] ch = boxes.toCharArray();
int[] nums = new int[n];
int ans = 0;
for (int i = 1; i < n; i++) {
if (ch[i] == '1') {
ans += i;
}
}
nums[0] = ans;
ans = 0;
for (int i = 0; i < n - 1; i++) {
if (ch[i] == '1') {
ans += (n - 1 - i);
}
}
nums[n - 1] = ans;
for (int i = 1; i < n - 1; i++) {
int a = 0, b = 0;
for (int j = 0; j < i; j++) {
if (ch[j] == '1') {
a += (i - j);
}
}
for (int k = i + 1; k < n; k++) {
if (ch[k] == '1') {
b += (k - i);
}
}
nums[i] = a + b;
}
return nums;
}
}
动态规划解法:
class Solution {
public int[] minOperations(String boxes) {
int n = boxes.length();
char[] ch = boxes.toCharArray();
int[] dp = new int[n];
int ans = 0;
int count = 0;
for (int i = 0; i < n; i++) {
if (ch[i] == '1') {
ans += i;
count++;
}
}
dp[0] = ans;
int left = 0, right = count;
for (int i = 1; i < n; i++) {
if (ch[i - 1] == '1') {
right--;
left++;
}
dp[i] = dp[i - 1] - right + left;
}
return dp;
}
}
执行结果: