題意
給定一個石頭的序列,青蛙從0開始跳,第一次只能跳一步,不能落入水中。
現在,假如我從上一個石頭跳到這個石頭是k步,那么,我們下一次只能跳k - 1或k或k + 1步,問:青蛙最后能否調到最后一個石頭上。
思路
算法1
爆搜,本來以為會T,但是實際上中間會有很多情況會落入水中,所以bfs能過。
算法2
dp,我們可以這樣考慮:我們當前在位置i,是從位置j經過k步到達位置i的。那么我們下一次能夠達到stonesi+k或stonesi+k?1或stonesi+k+1。
于是,我們的狀態表示為:d[i],從i之前的某一位置j經過k步跳過來的,那么,很明顯我們的d[i]有很多種情況,我們可以再加一維j表示從j經過d[i,j]步跳過來的,也可以用一個set來存我們經過多少步到i的。
即,我們定義我們的d[i]為unordered_map<int, set<int>> pos,key為stones[i],value為跳到這個位置的所有可能步數集合。
那么,我們接下來需要做的就是:假設我們當前在位置i,我們取出pos[stones[i]]對應的集合tk。對于tk內的每一個步數k,那么我們下一次就可以走到的位置newpos為stones[i] + k或stones[i] + k - 1或stones[i] + k + 1。然后判斷newpos是否為一個石頭,如果是,那么我們就將可行的解加入pos即可。
最后,只需要判斷pos[stones[n - 1]].size()>0即可(即是否存在到終點的可行步數)
代碼
//algorithm 1#define PII pair<int, int>#define mp make_pairclass Solution {public:    bool canCross(vector<int>& stones) {        set<PII> vis;        set<int> has;        for (auto x : stones) has.insert(x);        int ll = stones[stones.size() - 1];        queue<PII> q;        q.push(mp(1, 1));        while (!q.empty()) {            PII t = q.front(); q.pop();            if (t.first == ll) return true;            if (vis.find(t) != vis.end()) continue;            else vis.insert(t);            int x = t.first, k = t.second;            if (has.find(x) != has.end()) {                if (x + k <= ll && has.find(x + k) != has.end()) q.push(mp(x + k, k));                if (x + k - 1 <= ll && has.find(x + k - 1) != has.end()) q.push(mp(x + k - 1, k - 1));                if (x + k + 1 <= ll && has.find(x + k + 1) != has.end()) q.push(mp(x + k + 1, k + 1));            }        }        return false;    }};//algorithm 2class Solution {public:    bool canCross(vector<int>& stones) {        unordered_map<int, set<int>> pos;        set<int> tmp;        for (auto x : stones) pos[x] = tmp;        for (int i = 0; i < stones.size(); i++) {            set<int> sk = pos[stones[i]];            for (auto k : sk) {                for (int step = k - 1; step <= k + 1; step++) {                if (step > 0 && pos[stones[i] + step] != NULL) pos[stones[i] + step].insert(stones[i] + step);                }            }        }        return pos[stones.size() - 1].size() > 0;    }};