一开始想到的方法是暴力递归,遍历当前石板序号N的约数,若对约数k,有N+k到M的解决方案,计算其步数,然后与当前最小步数比较找出最小步数。果不其然,暴力算法的时间复杂度很大,当数据量达到100000级别时,就会超时。
看评论区大佬的动态规划算法,恍然大悟!
dp数组记录到达每一块石板的最小步数,初始化为INT_MAX,若最后最小步数为INT_MAX,则视为不可达,dp[N]=0,i从N开始遍历到M,找到当前石板序号的约数为j,动态规划的转移方程为:
dp[i+j]=min(dp[i]+1,dp[i+j]);
dp[N]=0;
代码如下:
#include<iostream>
#include<vector>
using namespace std;
#define INT_MAX 0x7fffffff
int main(){
int n,m;
cin>>n>>m;
vector<int> dp(m+1,INT_MAX);
dp[n]=0;
for(int i=n;i<m;i++){
if(dp[i]==INT_MAX) continue;
for(int j=2;j*j<=i;j++){
if(j*(i/j)==i){
if(i+j<m+1){
dp[i+j]=min(dp[i]+1,dp[i+j]);
}
if(j*j!=i&&i+i/j<m+1){
dp[i+i/j]=min(dp[i]+1,dp[i+i/j]);
}
}
}
}
if(dp[m]==INT_MAX)dp[m]=-1;
cout << dp[m] << endl;
return 0;
}