【每日一题】【DFS】【试除法求约数】【大剪枝】清楚姐姐跳格子 牛客周赛 Round 54 D题 C++

牛客周赛 Round 54 D题

清楚姐姐跳格子

题目背景

牛客周赛 Round 54

题目描述

样例 #1

样例输入 #1

5
2 3 1 5 4

样例输出 #1

2

做题思路

首先知道$a_i\ge 1$，所以必定有一个正整数因子1
至少可以从第一格一步步的到达第$n$格，所以答案肯定是有的。

假定$vi{s}_i$为从第$i$格到第$n$格的最少步数。
那么初始化为

for(int i=1;i<=n;i++)cin >> a[i] , vis[i] = n-i;

后面考虑假如一直往右跳的可能。

这时候从前往后遍历的话会导致前面用了后面的$vis$但该$vis$并不是真正的从第$i$格到第$n$格的最少步数。

所以考虑从后往前遍历，这样遍历到前面用后面的$vis$就是一直往右跳真正的从第$i$格到第$n$格的最少步数。

这时就产生了第一个“状态转移“方程vis[i] = min(vis[i],vis[i+k]+1)
其中$k$为$a_i$的一个正整数因子。

现在再考虑又往右又往左的情况。
实际上就是假如该点的$vi{s}_i$变了，那么依附于其(或者“状态转移“方程出现$vi{s}_i$)的$vi{s}_j$全都得变。

简单来说就是当第$i$个格子的$vi{s}_i$(为从第$i$格到第$n$格的最少步数)变少了，那么如果存在一个$j$并且第$j$个格子能跳到第$i$个格子那么该格子的$vi{s}_j$就可能变少。

如果$vi{s}_j$变少，只可能变成$vi{s}_i+1$,因为现在是$vi{s}_i$变少了，更新完$vi{s}_i$后需要检查$vi{s}_j$需不需要更新

其状态转移方程为$vi{s}_j=\min (vi{s}_i+1,vi{s}_j)$

同理如果$vi{s}_j$变少也需要重复以上操作。

总结：
本题核心为vis[i] = min(vis[i],vis[i+k]+1)
因为该方程不满足无后效性，所以使用dfs进行关联更新

代码

#include <iostream>
#include <algorithm>
#include <vector>
#include <map>
#include <queue>
#include <tuple>
#include <cstring>
#include <cmath>
#define int long long
using namespace  std;
const int N = 1e5+10;
const int INF = 0x3f3f3f3f3f3f3f3f;
vector<vector<int>>connect(N,vector<int>());
int vis[N] , n , a[N];
void dfs(int x){
    for(auto j :connect[x]){
        if(vis[j] > vis[x]+1){
            vis[j] = vis[x] + 1;
            dfs(j);
        }
    }
}
signed main(){
    cin >> n;
    for(int i=1;i<=n;i++)cin >> a[i] , vis[i] = n-i;

    for(int i=n-1;i>=1;i--){
        int x = a[i] , u = i;
        for(int j=1;j<=x/j;j++){
            if(j > n)break;//注意重点剪枝，大于n必定跳出格子
            if(x%j == 0){
                int k1 = j , k2 = x/j;

                if(k1 + u <= n && k1 + u >= 1 && vis[k1+u] + 1 < vis[u]){
                    vis[u] = vis[k1 + u] + 1;
                    connect[u+k1].push_back(i);
                }

                if(u - k1 >= 1 && u - k1 <= n){
                    vis[u] = min(vis[u],vis[u - k1] + 1);
                    connect[u-k1].push_back(i);
                }

                if(k2 + u <= n && k2 + u >= 1 && vis[k2+u] + 1 < vis[u]){
                    vis[u] = vis[k2 + u] + 1;
                    connect[k2+u].push_back(i);
                }

                if(u - k2 >= 1 && u - k2 <= n){
                    vis[u] = min(vis[u - k2] + 1,vis[u]);
                    connect[u-k2].push_back(i);
                }

            }
        }
        dfs(i);
        //for(int i=1;i<=n;i++)cout << vis[i] << " \n"[i==n];
    }

    cout << vis[1];
    return 0;
}