影响决策的因素有:时刻、这一时刻的位置和上一时刻的位置 。 由于n和k都不大,可以开三重循环来递推这三个量
状态方程dp[i][j]表示在时刻i打位置j的老鼠,此时消耗总能量的最小值,则可以写出状态转移方程:dp[i][j] = min(dp[i][j],dp[i+1][x]+abs(a[i][j]-a[i+1][x])); 时间按照从大到小倒着枚举。
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#include<vector>
using namespace std;
const int INF = 1000000000;
int n,k,a[30][15],dp[30][20];
int main(){
while(~scanf("%d%d",&n,&k)){
for(int i=1;i<=n;i++)
for(int j=1;j<=k;j++) scanf("%d",&a[i][j]);
memset(dp,0,sizeof(dp));
int x=0,y=1;
for(int i=n-1;i>=1;i--){
for(int j=1;j<=k;j++){
dp[i][j] = INF;
for(int x=1;x<=k;x++){
dp[i][j] = min(dp[i][j],dp[i+1][x]+abs(a[i][j]-a[i+1][x]));
}
}
}
int ans = INF;
for(int i=1;i<=k;i++) ans = min(ans,dp[1][i]);
printf("%d\n",ans);
}
return 0;
}