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

源代码•宸

已于 2023-11-08 22:15:35 修改

阅读量246

点赞数 4

分类专栏： LeetCode刷题文章标签： leetcode 算法职场和发展 c++ 经验分享前缀和

于 2023-11-07 22:52:36 首次发布

本文链接：https://blog.csdn.net/qq_44631615/article/details/134278068

版权

LeetCode刷题专栏收录该内容

372 篇文章 5 订阅

订阅专栏

2023每日刷题（二十二）

Leetcode—2731.移动机器人

在这里插入图片描述

算法思路

参考自灵茶山艾府
在这里插入图片描述

实现代码

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;
    }
};