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

参考链接

1. https://leetcode-cn.com/problems/minimum-number-of-operations-to-move-all-balls-to-each-box/
2. https://leetcode-cn.com/problems/minimum-number-of-operations-to-move-all-balls-to-each-box/solution/dong-tai-gui-hua-shuang-bai-by-juicie-mnxf/

题目描述

有 n 个盒子。给你一个长度为 n 的二进制字符串 boxes ，其中 boxes[i] 的值为 ‘0’ 表示第 i 个盒子是 空 的，而 boxes[i] 的值为 ‘1’ 表示盒子里有 一个 小球。

在一步操作中，你可以将 一个 小球从某个盒子移动到一个与之相邻的盒子中。第 i 个盒子和第 j 个盒子相邻需满足 abs(i - j) == 1 。注意，操作执行后，某些盒子中可能会存在不止一个小球。

返回一个长度为 n 的数组 answer ，其中 answer[i] 是将所有小球移动到第 i 个盒子所需的 最小 操作数。

每个 answer[i] 都需要根据盒子的 初始状态 进行计算。

解题思路

复杂的动态规划

可以从i的左右两边分别讨论，先考虑将i-1处的’1’移到i处需要多少步，然后再考虑将i+1处的’1’移到i处需要多少步。

简洁的动态规划

先求第 0 个盒子，同时记录当前及其右边的 1 的数量，然后依次求 1 到 最后一个盒子，每次向右移动的时候，dp[i] 会比 dp[i-1] 的 minOperations 减少 right 个需要注意的是如果 boxes[i-1] 是 1 的话 ，会导致左边的 1 的数量增加，在 left 中补充上左边的 1，同时 当前及其右边的 1 的数量 right 会减少， 最终：dp[i] = dp[i-1] - right + left。

代码

复杂的动态规划

class Solution {
public:
    vector<int> minOperations(string boxes) {
        vector<int> left_table(boxes.size()); // i左边积累了多少个
        vector<int> right_table(boxes.size()); // i右边积累了多少个
        vector<int> left_dis(boxes.size()); // i左边积累的到i多少步
        vector<int> right_dis(boxes.size());// i右边积累的到i多少步
        vector<int> answer(boxes.size());

        left_table[0] = boxes[0] == '1' ? 1 : 0;
        
        for (int i = 1; i < boxes.size(); i ++)
        {
            if (boxes[i] == '1')
            {
                left_table[i] = left_table[i - 1] + 1;
            }
            else
            {
                left_table[i] = left_table[i - 1];
            }
            left_dis[i] = left_dis[i - 1] + left_table[i - 1];
        }
        answer[boxes.size()-1] = left_dis[boxes.size()-1];

        right_table[boxes.size() - 1] = boxes[boxes.size() - 1] == '1' ? 1 : 0;
        for (int i = boxes.size() - 2; i >= 0; i --)
        {
            if (boxes[i] == '1')
            {
                right_table[i] = right_table[i + 1] + 1;
            }
            else
            {
                right_table[i] = right_table[i + 1];
            }
            right_dis[i] = right_dis[i + 1] + right_table[i + 1];
            answer[i] = right_dis[i] + left_dis[i];
        }

        return answer;
    }
};

简洁的动态规划

class Solution {
public:
    vector<int> minOperations(string boxes) {
        vector<int> dp(boxes.size());
        int right = 0;

        for (int i = 0; i < boxes.size(); i ++)
        {
            if (boxes[i] == '1')
            {
                dp[0] += i;
                right ++;
            }
        }

        int left = 0;
        for (int i = 1; i < boxes.size(); i ++)
        {
            if (boxes[i - 1] == '1')
            {
                left ++;
                right --;
            }
            dp[i] = dp[i - 1] - right + left;
        }
        return dp;
    }
};