题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=4540
思路:比较简单的一道dp题,设dp[i][j]为第i个时刻砸第j只地鼠所需最小的能量,dp[i][j]=min(dp[i][j], dp[i-1][e]),0<=e<k;
用记忆化搜索写出来。
#include <cstdio>
#include <cstring>
#include <vector>
#include <cmath>
#include <algorithm>
using namespace std;
const int inf = 0x3f3f3f3f;
int a[30][30];
int dp[30][30];
int n, k, u;
int dfs(int cur, int pos)
{
if(cur >= n) return 0;
int &res = dp[cur][pos];
if(res != inf) return res;
for(int i=0; i<k; i++)
{
res = min(res, dfs(cur+1, i)+abs(a[cur+1][i] - a[cur][pos]));
}
return res;
}
int main()
{
while(~scanf("%d%d", &n, &k))
{
for(int i=1; i<=n; i++)
{
for(int j=0; j<k; j++)
{
scanf("%d", &a[i][j]);
}
}
memset(dp, 0x3f, sizeof(dp));
int ans = inf;
for(int i=0; i<k; i++)
ans = min(ans, dfs(1,i));
printf("%d\n", ans);
}
return 0;
}