题意:一个时间点在数轴的不同位置有龙珠,但每个时间点只能且必须收集一个龙珠,在不同的位置收集龙珠会损失不同的能量,移动距离也会损失能量,求m段时间后损失的最小能量。
分析:就是一个数塔的dp模型,跟免费馅饼还有多时段多课问题一样。
dp[i][j]:第i段时间第j的位置的最小能量损失;
转移:dp[i][j]=min(dp[i+1][k]+|a[i][j].pos-a[i+1][k].pos|)+a[i][j].value,三重循环,交C++超时了,交G++过了
代码:
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
#define INF 1000000007
using namespace std;
int t,m,n,x;
//long long b[100][1005];
struct node{
int pos;
long long value;
};
node a[100][1005];
long long dp[100][1005];
int main()
{
scanf("%d",&t);
while(t--){
memset(a,0,sizeof(a));
// memset(b,0,sizeof(b));
memset(dp,0,sizeof(dp));
// long long mx=-1;
scanf("%d%d%d",&m,&n,&x);
for(int i=1;i<=m;i++){
for(int j=0;j<n;j++){
scanf("%d",&a[i][j].pos);
// mx=max(mx,a[i][j].pos);
}
}
a[0][0].pos=x,a[0][0].value=0;
for(int i=0;i<=m;i++){
for(int j=0;j<n;j++){
dp[i][j]=INF;
}
}
for(int i=1;i<=m;i++){
for(int j=0;j<n;j++){
scanf("%I64d",&a[i][j].value);
// dp[i][j]=a[i][j].value;
}
}
for(int i=0;i<n;i++) dp[m][i]=a[m][i].value;
for(int i=m-1;i>=0;i--){
for(int j=0;j<n;j++){
// if(b[i][j]||i==0){
// cout<<i<<" "<<j<<": ";
for(int k=0;k<n;k++){
// if(b[i+1][k]){
// cout<<dp[i+1][k]<<" "<<a[i+1][k].pos<<" "<<a[i+1][k].value;
dp[i][j]=min(dp[i][j],a[i][j].value+dp[i+1][k]+abs(a[i+1][k].pos-a[i][j].pos));//cout<<" "<<dp[i][j]<<endl;
// }
}
// }
}
}
printf("%I64d\n",dp[0][0]);
}
}