LeLeetCode 403. 青蛙過河

畫船聽雨發表於2018-07-10

題目描述

這裡寫圖片描述

解題思路

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<node> que;
        map<int, int >mpstones;
        map<pair<int, 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;
    }
};

相關文章