Leetcode—1769. 移动所有球到每个盒子所需的最小操作数【中等】

2024每日刷题(一零九)

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

Leetcode—1769. 移动所有球到每个盒子所需的最小操作数【中等】_第1张图片

实现代码

class Solution {
public:
    vector<int> minOperations(string boxes) {
        int n = boxes.size();
        vector<int> ans;
        for(int i = 0; i < n; i++) {
            int res = 0;
            for(int j = 0; j < n; j++) {
                if(j != i && boxes[j] - '0' == 1) {
                    res += abs(i - j);
                }
            }
            ans.push_back(res);
        }
        return ans;
    }
};

运行结果

Leetcode—1769. 移动所有球到每个盒子所需的最小操作数【中等】_第2张图片

巧解实现代码

class Solution {
public:
    vector<int> minOperations(string boxes) {
        int n = boxes.size();
        vector<int> ans(n);
        // 记录第i个盒子之前的盒子向第i个盒子移动
        for(int i = 0, count = 0, moves = 0; i < n; i++) {
            ans[i] += moves;
            count += boxes[i] - '0';
            moves += count;
        }
        // 记录第i个盒子之后的盒子向第i个盒子移动
        for(int i = n - 1, count = 0, moves = 0; i >= 0; i--) {
            ans[i] += moves;
            count += boxes[i] - '0';
            moves += count;
        }
        return ans;
    }
};

运行结果

Leetcode—1769. 移动所有球到每个盒子所需的最小操作数【中等】_第3张图片
之后我会持续更新,如果喜欢我的文章,请记得一键三连哦,点赞关注收藏,你的每一个赞每一份关注每一次收藏都将是我前进路上的无限动力 !!!↖(▔▽▔)↗感谢支持!

你可能感兴趣的:(LeetCode刷题,leetcode,算法,职场和发展,c++,字符串,数组,经验分享)