LeetCode 5294. 你能从盒子里获得的最大糖果数

最新推荐文章于 2020-09-19 08:14:53 发布

爱学习的人啊

最新推荐文章于 2020-09-19 08:14:53 发布

阅读量173

点赞数

分类专栏： LeetCode C++ 文章标签： leetcode 算法数据结构

本文链接：https://blog.csdn.net/weixin_44565518/article/details/103657406

版权

C++ 同时被 2 个专栏收录

29 篇文章 0 订阅

订阅专栏

LeetCode

11 篇文章 0 订阅

订阅专栏

LeetCode周赛题第四题hard难度，但是感觉比前面两道还简单。。。

class Solution {
public:
    int maxCandies(vector<int>& status, vector<int>& candies, vector<vector<int>>& keys, vector<vector<int>>& containedBoxes, vector<int>& initialBoxes) {
    queue<int> xiangzi;//存得到的箱子
    for(int i=0;i<initialBoxes.size();++i)
    {
        xiangzi.push(initialBoxes[i]);
    }    
    if(xiangzi.empty()) return 0;
    vector<int> yaoshi;//存获得的钥匙
    int tangguo=0;//总的糖果数
    while(!xiangzi.empty())
    {
        int xiangziSize=xiangzi.size();//首先遍历得到的所有箱子
        int box=1000;
        for(int i=0;i<xiangziSize;++i)
        {
            box = xiangzi.front();
            if(status[box] == 1)//1、开着的箱子不用钥匙
            {
                xiangzi.pop();
                break;
            }
            else
            {
                if(find(yaoshi.begin(),yaoshi.end(),box) != yaoshi.end())//2、锁着的箱子但是有钥匙
                {
                    xiangzi.pop();
                    status[box]=1;//同时将状态置为1
                    break;
                }
                else//如果没有钥匙，先出队再入队，将这个箱子放到队尾
                {
                    xiangzi.pop();
                    xiangzi.push(box);
                }
            }
        }
        //到这儿的三种情况1、箱子是开的2、箱子是锁的但有钥匙3、箱子有锁但是没钥匙，所以前两种情况status==1，后者==0
        if(status[box] == 1)
        {
            tangguo+=candies[box];//计算糖果的总数
            for(int i=0;i<keys[box].size();++i)
            yaoshi.push_back(keys[box][i]);//获得钥匙
            for(int i=0;i<containedBoxes[box].size();++i)
            xiangzi.push(containedBoxes[box][i]);//获得的箱子
        }
        else
        {
            xiangzi.pop();//如果status==0，则出队
        }
    }
    return tangguo;
    }
};