Leetcode 1744 你能在你最喜欢的那天吃到你最喜欢的糖果吗？前缀和

最新推荐文章于 2024-09-26 17:00:00 发布

幸愉信奥

最新推荐文章于 2024-09-26 17:00:00 发布

阅读量119

点赞数

分类专栏：刷题记录本文章标签： Leetcode 前缀和

本文链接：https://blog.csdn.net/Edviv/article/details/117457466

版权

刷题记录本专栏收录该内容

69 篇文章 3 订阅

订阅专栏

题目链接
在这里插入图片描述
前缀和思想， $v [i]$ 表示前 $i - 1$ 天糖果数量总合，判断一下每天一颗糖果是否在第 $favoriteDay_i$ 吃到糖果，并且最多能吃的糖果数是否大于 $favoriteDay_i$ 前吃的糖果数量总合。

class Solution {
public:
    typedef long long ll;
    vector<bool> canEat(vector<int>& candiesCount, vector<vector<int>>& queries) {
        int n = candiesCount.size();
        vector<ll> v(n+1);
        ll res = 0;
        for(int i = 0; i < n; i++)
        {
           v[i] = res;
           res += candiesCount[i];
        }
        v[n] = res;
        vector<bool> ans;
        for(auto& x : queries)
        {
            int id = x[0]; ll day = x[1], eve = x[2];
            ll t = eve*(day+1);
           
            if(day < v[id+1] && t > v[id]) ans.push_back(true);
            else ans.push_back(false);
        }
        return ans;
    }
};