CodeForces-1601B Frog Traveler

Frog Traveler

dp + bfs

感觉又像搜索

从后往前进行状态转移，像 \(bfs\) 一样保证当前搜索的是消耗次数最少的

状态转移因为是一个连续的区间，因此考虑当前能上升到的最大距离，只有能更新这个最大值，才进行状态转，保证每个位置只被访问一次

时间复杂度 \(O(n)\)

#include <iostream>
#include <cstdio>
#include <vector>
#include <queue>
using namespace std;
#define pii pair<int, int>
const int maxn = 3e5 + 10;
int dep[maxn];
int a[maxn], b[maxn];
pii pre[maxn];

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int n;
    cin >> n;
    for(int i=1; i<=n; i++) cin >> a[i];
    for(int i=1; i<=n; i++) cin >> b[i];
    int l = n;
    queue<int>q;
    q.push(n);
    dep[n] = 1;
    while(q.size())
    {
        int now = q.front();
        q.pop();
        int tp = now - a[now];
        if(tp <= 0)
        {
            pre[0] = {0, now};
            dep[0] = dep[now] + 1;
            break;
        }
        while(l >= tp)
        {
            int nex = l + b[l];
            l--;
            if(dep[nex]) continue;
            dep[nex] = dep[now] + 1;
            pre[nex] = {l + 1, now};
            q.push(nex);
        }
    }
    if(dep[0] == 0) cout << "-1\n";
    else
    {
        cout << dep[0] - 1 << "\n";
        int now = 0;
        vector<int>ans;
        for(int i=1; i<dep[0]; i++)
        {
            ans.push_back(pre[now].first);
            now = pre[now].second;
        }
        for(int i=ans.size()-1; i>=0; i--)
            cout << ans[i] << " ";
        cout << "\n";
    }
    return 0;
}