Leetcode—2731.移动机器人【中等】

2023每日刷题(二十二)

Leetcode—2731.移动机器人

Leetcode—2731.移动机器人【中等】_第1张图片

算法思路

参考自灵茶山艾府
Leetcode—2731.移动机器人【中等】_第2张图片

实现代码

class Solution {
public:
    const int MOD = 1e9 + 7;
    int sumDistance(vector<int>& nums, string s, int d) {
        int n = nums.size();
        vector<long long>arr(n, 0);
        long long ans = 0;
        long long sum = 0;
        for(int i = 0; i < n; i++) {
            arr[i] = (long long)nums[i] + d * ((s[i] & 2) - 1);
        }
        sort(arr.begin(), arr.end());
        sum = arr[0];
        for(int i = 1; i < n; i++) {
            ans = (ans + arr[i] * i - sum) % MOD;
            sum += arr[i];
        }
        return ans;
    }
};

运行结果

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

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