码迷,mamicode.com
首页 > 其他好文 > 详细

[每日一题] leetcode 403. 青蛙过河

时间:2021-04-30 12:10:08      阅读:0      评论:0      收藏:0      [点我收藏+]

标签:距离   public   oss   ret   bsp   遍历   class   ++   去重   

因为数据太大,但数据量少,因此可以用map先建立映射关系

再用二维set,其中set[i]表示第i个石块所能跳的距离数组

二维vector会超时,set去重就可以了

然后遍历每个石块,并求出其所能到达石块 所能跳的距离

最后判断第n - 1个石块是否有能跳的步数即可

class Solution {
public:
    set<int> V[2020];
    map<int, int> M;
    bool canCross(vector<int>& stones) {
        int n = stones.size();
        int cnt = 0;
        for(int i = 0; i < n; i++)
            M[stones[i]] = i;
        V[0].insert(1);
        set<int>::iterator it;
        for(int i = 0; i < n; i++)
        {
            for(it = V[i].begin(); it != V[i].end(); it++)
            {
                if(M.count(stones[i] + *it) != 0)
                {
                    V[M[stones[i] + *it]].insert(*it);
                    if(*it - 1 != 0)
                        V[M[stones[i] + *it]].insert(*it - 1);
                    V[M[stones[i] + *it]].insert(*it + 1);
                }
            }
        }
        return V[n - 1].size();
    }
};

 

[每日一题] leetcode 403. 青蛙过河

标签:距离   public   oss   ret   bsp   遍历   class   ++   去重   

原文地址:https://www.cnblogs.com/WTSRUVF/p/14718623.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!