LeLeetCode 403. 青蛙过河

题目描述

LeLeetCode 403. 青蛙过河_第1张图片

解题思路

bfs暴力查找+小剪枝

首先特判一下特殊情况,当第二个与第一个的距离差大于1的话,直接返回false
对于每个石子的下一跳都有三种情况,k-1、k、k+1。其中这个K代表当前这个石子是由多长的步数跳过来的。
假设当前位置为pos,它是由k长度跳过来的。vis[pos][k]表示这个状态是否出现过。
需要注意的是:一定要判断vis[pos][k]是否之前出现,如果 出现过则不再进行搜索。
举个栗子:pos - 1 的位置跳了k-1步到了pos则pos出现(pos,k)这种状态(三种情况中的一种);
pos - 2 的位置跳了k步到了pos,则pos也会出现(pos,k)这种状态(三种状态中的一种);
所以,如果不对重复的状态进行剪枝的话会导致多余的搜索次数,因此在这里需要一个小的剪枝。

代码实现

struct node
{
    int k;
    int x;
};
int dx[] = {1, 0, -1};

class Solution
{
public:
    bool canCross(vector<int>& stones)
    {
        queue que;
        map<int, int >mpstones;
        mapint, int>, int >mppos;
        int n = stones.size();

        for(int i = 0; i < n; i++)
            mpstones[stones[i]] = i;
        if(stones[1]-stones[0] > 1) return false;
        node start;
        start.k = stones[1]-stones[0];
        start.x = stones[1];
        que.push(start);
        pair<int, int> pa = make_pair(start.x, start.k);
        mppos[pa] = 1;
        while(!que.empty())
        {
            node xp = que.front();
            que.pop();
            node sd;
            if(xp.x == stones[n-1])
                return true;
            for(int i = 0; i < 3; i++)
            {
                sd.k = xp.k + dx[i];
                if(sd.k <= 0) continue;
                int newx = xp.x+sd.k;
                if(mpstones[newx] == 0) continue;
                sd.x = newx;
                int xPos = mpstones[newx];
                pair<int, int> sa = make_pair(xPos, sd.k);
                if(mppos[sa]) continue;
                mppos[sa] = 1;
                que.push(sd);
            }
        }
        return false;
    }
};

你可能感兴趣的:(LeetCode)