URAL 1029 Ministry

URAL 1029

思路：

dp+记录路径

状态：dp[i][j]表示到(i,j)这个位置为止的最少花费

初始状态：dp[1][i]=a[1][i](1<=i<=m)

状态转移：dp[i][j]=a[i][j]+max(dp[i-1][j],dp[i][j-1],dp[i][j+1])（注意扫的方向不同）

数组记录上一次的坐标，最后递归输出

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
#define mem(a,b) memset(a,b,sizeof(a))
 
const int N=;
const int INF=0x3f3f3f3f;
int a[N][N];
int dp[N][N];
pair<int,int> pre[N][N];
void dfs(int x,int y){
    if(x==-||y==-)return ;
    dfs(pre[x][y].first,pre[x][y].second);
    cout<<y<<' ';
}
int main(){
    ios::sync_with_stdio(false);
    cin.tie();
    int n,m;
    mem(a,INF);
    mem(dp,INF);
    cin>>n>>m;
    for(int i=;i<=n;i++){
        for(int j=;j<=m;j++){
            cin>>a[i][j];
        }
    }
    for(int i=;i<=n;i++)for(int j=;j<=m;j++)pre[i][j].first=-,pre[i][j].second=-;
    for(int i=;i<=m;i++)dp[][i]=a[][i];
    for(int i=;i<=n;i++){
        for(int j=;j<=m;j++){
            if(dp[i][j]>a[i][j]+dp[i-][j]){
                dp[i][j]=a[i][j]+dp[i-][j];
                pre[i][j].first=i-;
                pre[i][j].second=j;
            }
            if(dp[i][j]>a[i][j]+dp[i][j-]){
                dp[i][j]=a[i][j]+dp[i][j-];
                pre[i][j].first=i;
                pre[i][j].second=j-;
            }
        }
        for(int j=m;j>=;j--){
            if(dp[i][j]>a[i][j]+dp[i-][j]){
                dp[i][j]=a[i][j]+dp[i-][j];
                pre[i][j].first=i-;
                pre[i][j].second=j;
            }
            if(dp[i][j]>a[i][j]+dp[i][j+]){
                dp[i][j]=a[i][j]+dp[i][j+];
                pre[i][j].first=i;
                pre[i][j].second=j+;
            }
        }
    }
    int ans=INF,id=;
    for(int i=;i<=m;i++){
        if(dp[n][i]<ans){
            ans=dp[n][i];
            id=i;
        }
    }
    int x=n,y=id;
    dfs(x,y);
    cout<<endl;
    return ;
}