LeetCode 3096.得到更多分数的最少关卡数目(前缀和,建议练习!)

一. 题目描述

给你一个长度为 n 的二进制数组 possible 。

Alice 和 Bob 正在玩一个有 n 个关卡的游戏,游戏中有一些关卡是 困难 模式,其他的关卡是 简单 模式。如果 possible[i] == 0 ,那么第 i 个关卡是 困难 模式。一个玩家通过一个简单模式的关卡可以获得 1 分,通过困难模式的关卡将失去 1 分。

游戏的一开始,Alice 将从第 0 级开始 按顺序 完成一些关卡,然后 Bob 会完成剩下的所有关卡。

假设两名玩家都采取最优策略,目的是 最大化 自己的得分,Alice 想知道自己 最少 需要完成多少个关卡,才能获得比 Bob 更多的分数。

请你返回 Alice 获得比 Bob 更多的分数所需要完成的 最少 关卡数目,如果 无法 达成,那么返回 -1 。

注意,每个玩家都至少需要完成 1 个关卡。

示例 1:

输入:possible = [1,0,1,0]

输出:1

解释:

我们来看一下 Alice 可以完成的关卡数目:

  • 如果 Alice 只完成关卡 0 ,Bob 完成剩下的所有关卡,那么 Alice 获得 1 分,Bob 获得 -1 + 1 - 1 = -1 分。
  • 如果 Alice 完成到关卡 1 ,Bob 完成剩下的所有关卡,那么 Alice 获得 1 - 1 = 0 分,Bob 获得 1 - 1 = 0 分。
  • 如果 Alice 完成到关卡 2 ,Bob 完成剩下的所有关卡,那么 Alice 获得 1 - 1 + 1 = 1 分,Bob 获得 -1 分。

Alice 需要完成至少一个关卡获得更多的分数。

示例 2:

输入:possible = [1,1,1,1,1]

输出:3

解释:

我们来看一下 Alice 可以完成的关卡数目:

  • 如果 Alice 只完成关卡 0 ,Bob 完成剩下的所有关卡,那么 Alice 获得 1 分,Bob 获得 4 分。
  • 如果 Alice 完成到关卡 1 ,Bob 完成剩下的所有关卡,那么 Alice 获得 2 分,Bob 获得 3 分。
  • 如果 Alice 完成到关卡 2 ,Bob 完成剩下的所有关卡,那么 Alice 获得 3 分,Bob 获得 2 分。
  • 如果 Alice 完成到关卡 3 ,Bob 完成剩下的所有关卡,那么 Alice 获得 4 分,Bob 获得 1 分。

Alice 需要完成至少三个关卡获得更多的分数。

示例 3:

输入:possible = [0,0]

输出:-1

解释:

两名玩家只能各完成 1 个关卡,Alice 完成关卡 0 得到 -1 分,Bob 完成关卡 1 得到 -1 分。两名玩家得分相同,所以 Alice 无法得到更多分数。

提示:

  • 2 <= n == possible.length <= 105
  • possible[i] 要么是 0 要么是 1 。

二. 解题思路

        1. 首先我们理解一下题意,题目中说到如果possible[i] == 1,说明是简单题,通过得1分,如果possible[i] == 0, 为困难题,通过要扣除1分。Alice和Bob两个人做题,Alice先开始做题,Alice做完后Bob从Alice结束的地方开始做完剩余的题目,问做完后Alice 获得比 Bob 更多的分数所需要完成的 最少 关卡数目,没有则返回-1。

        2. 本题利用前缀和的思想,我们首先对原数组做一些操作:遍历possible数组,如果possible[i] == 1,我们不改变其值(得分为1);如果possible[i] == 0,我们将其改为-1(得分-1),以例一为例操作完之后为:[1, -1,1, -1]。然后我们计算possible数组的前缀和(直接在原数组上进行操作即可:possible[i]的值等于possible[i]初始值加possible[i - 1],操作完之后:[1,0,1,0])。

        3. 遍历前缀和数组possible,Alice的分数就等于possible[i],而Bob的分数等于possible[n - 1]减去Alice走过的关(即possible[n - 1] - possible[i]),最后比较二者大小,如果Alice > Bob 直接return i+1 即可。

三. 代码

class Solution {
public:
    int minimumLevels(vector<int>& possible) {
        int res = 0;
        int n = possible.size();
        for(int i = 0; i < n; i++) possible[i] = possible[i] == 1 ? 1 : -1;    // 重新赋值
        for(int i = 1; i < n; i++){        
            possible[i] += possible[i - 1];        // 计算前缀和
        }
        for(int i = 0; i < n - 1; i++){        // n-1是因为题中明确要求每个人必须至少做一道题
            int bob = 0;
            int Alice = 0;
            bob += possible[n - 1] - possible[i];    
            Alice += possible[i];
            if(Alice > bob){
                return i + 1;
            }
        }
        return -1;
    }
};

四. 总结:

        1. 本题主要考察数组的使用以及前缀和,对前缀和知识点不熟悉的同学可以先学习前缀和之后再做题。

        2. 对于本题,在进行赋值和求取前缀和的时候,我直接在原数组的基础上进行了操作,如果看不懂的同学可以将possible数组拷贝到一个新的数组中进行操作,这样操作可能易于理解一些,但是空间复杂度会上升。

        3. 复杂度分析:

                时间复杂度:O(n),其中 n 表示给定数组的长度。遍历两次 possible 数组即可。

                空间复杂度:O(1)。

喜欢的话给个关注吧!!

  • 7
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值