【力扣杯 春季编程大赛】1. 宝石补给(简单)

174 阅读3分钟

一起养成写作习惯!这是我参与「掘金日新计划 · 4 月更文挑战」的第7天,点击查看活动详情

一、题目描述

欢迎各位勇者来到力扣新手村,在开始试炼之前,请各位勇者先进行「宝石补给」。

每位勇者初始都拥有一些能量宝石, gem[i] 表示第 i 位勇者的宝石数量。现在这些勇者们进行了一系列的赠送,operations[j] = [x, y] 表示在第 j 次的赠送中 第 x 位勇者将自己一半的宝石(按需向下取整)赠送给第 y 位勇者。

在完成所有的赠送后,请找到拥有最多宝石的勇者和拥有最少宝石的勇者,并返回他们二者的宝石数量之差

注意:

  • 赠送将按顺序逐步进行。

示例 1:

输入:gem = [3,1,2], operations = [[0,2],[2,1],[2,0]]

输出:2

解释:
第 1 次操作,勇者 0 将一半的宝石赠送给勇者 2, gem = [2,1,3]
第 2 次操作,勇者 2 将一半的宝石赠送给勇者 1, gem = [2,2,2]
第 3 次操作,勇者 2 将一半的宝石赠送给勇者 0, gem = [3,2,1]
返回 3 - 1 = 2

示例 2:

输入:gem = [100,0,50,100], operations = [[0,2],[0,1],[3,0],[3,0]]

输出:75

解释:
第 1 次操作,勇者 0 将一半的宝石赠送给勇者 2, gem = [50,0,100,100]
第 2 次操作,勇者 0 将一半的宝石赠送给勇者 1, gem = [25,25,100,100]
第 3 次操作,勇者 3 将一半的宝石赠送给勇者 0, gem = [75,25,100,50]
第 4 次操作,勇者 3 将一半的宝石赠送给勇者 0, gem = [100,25,100,25]
返回 100 - 25 = 75

示例 3:

输入:gem = [0,0,0,0], operations = [[1,2],[3,1],[1,2]]

输出:0

提示:

  • 2 <= gem.length <= 10^3
  • 0 <= gem[i] <= 10^3
  • 0 <= operations.length <= 10^4
  • operations[i].length == 2
  • 0 <= operations[i][0], operations[i][1] < gem.length

二、思路分析

题意很简单,要完成所有的赠送,也就是说需要遍历 operations 数组,赠送规则为:编号为 operations[i][0] 的勇士将自己一半的宝石送给编号为 operations[i][1] 的勇士。

清楚这个之后,我们就可以模拟这个赠送过程来写代码:

  • 遍历 operations 数组,当前元素为 operations[i] ,获取要赠送的勇士的编号 operations[i][0] ,获取被赠送的勇士的编号 operations[i][1]
  • 得知勇士编号后,就可以知道勇士的宝石数量 gem[勇士编号] ,也就可以获取要赠送的宝石的数量 Math.floor(gem[operations[i][0]] / 2) ,设为 give
  • 赠送宝石的勇士减去 give ,被赠送宝石的勇士加上 give
  • 重复这个过程,直到遍历完 operations 数组,此时赠送完毕;
  • 遍历宝石数组 gem ,获取其中最大值和最小值,两者相减就是题目要返回的值了。

三、AC 代码

/**
 * @param {number[]} gem
 * @param {number[][]} operations
 * @return {number}
 */
var giveGem = function(gem, operations) {
    for (let arr of operations) {
        const a = arr[0], b = arr[1];
        const give = Math.floor(gem[a] / 2);
        gem[a] -= give;
        gem[b] += give;
    }
    let max = -Infinity, min = Infinity;
    for (let x of gem) {
        max = Math.max(max, x);
        min = Math.min(min, x);
    }
    return max - min;
};