HDU 5092

http://acm.hdu.edu.cn/showproblem.php?pid=5092

卡读题，实质是每行取一个点，从上到下找一条路径权值和最小，点可以到达的地方是周围八个格子

类似数塔的dp，需要记录路径，当前行由上一行顶上的三个格子转移而来

#include <iostream>
#include <cstdio>
#include <queue>
#include <cstring>
#include <algorithm>
#include <stack>

using namespace std;
int n,m;
int a[][],dp[][],pre[][];
int dx[]={-,-,-};
int dy[]={,,-};

struct node{
    int x,y;
};

int main(){
    int T;
    scanf("%d",&T);
    for(int cas=;cas<=T;cas++){
        scanf("%d%d",&n,&m);
        for(int i=;i<=n;i++){
            for(int j=;j<=m;j++){
                scanf("%d",&a[i][j]);
            }
        }
        for(int i=;i<;i++)
            for(int j=;j<;j++)
                dp[i][j]=0xfffffff;
        for(int i=;i<=m;i++)
            dp[][i]=a[][i];
        memset(pre,-,sizeof(pre));
        for(int i=;i<=n;i++){
            for(int j=;j<=m;j++){
                if(j->= && j+<=m){
                    //dp[i][j]=min(dp[i][j],a[i][j]+min(dp[i-1][j],min(dp[i-1][j-1],dp[i-1][j+1])));
                    int t=min(dp[i-][j],min(dp[i-][j-],dp[i-][j+]));
                    if(a[i][j]+t<dp[i][j]){
                        dp[i][j]=a[i][j]+t;
                        if(dp[i-][j+]<=dp[i-][j-] && dp[i-][j+]<=dp[i-][j])
                            pre[i][j]=;
                        else if(dp[i-][j]<=dp[i-][j-] && dp[i-][j]<=dp[i-][j+])
                            pre[i][j]=;
                        else pre[i][j]=;
                    }
                }
                else if(j->=){
                    //dp[i][j]=min(dp[i][j],a[i][j]+min(dp[i-1][j],dp[i-1][j-1]));
                    int t=min(dp[i-][j],dp[i-][j-]);
                    if(a[i][j]+t<dp[i][j]){
                        dp[i][j]=a[i][j]+t;
                        if(dp[i-][j]<=dp[i-][j-])
                            pre[i][j]=;
                        else pre[i][j]=;
                    }
                }
                else if(j+<=m){
                    //dp[i][j]=min(dp[i][j],a[i][j]+min(dp[i-1][j],dp[i-1][j+1]));
                    int t=min(dp[i-][j],dp[i-][j+]);
                    if(a[i][j]+t<dp[i][j]){
                        dp[i][j]=a[i][j]+t;
                        if(dp[i-][j+]<=dp[i-][j])
                            pre[i][j]=;
                        else pre[i][j]=;
                    }
                }
            }
        }
        int ans=0xfffffff;
        int res;
        for(int i=;i<=m;i++){
            //ans=min(ans,dp[n][i]);
            if(ans>=dp[n][i]){
                ans=dp[n][i];
                res=i;
            }
        }
        printf("Case %d\n",cas);
        stack <node> s;
        node st;
        st.x=n;st.y=res;
        s.push(st);
        while(){
            if(st.x==)break;
            int xx=st.x+dx[pre[st.x][st.y]];
            int yy=st.y+dy[pre[st.x][st.y]];
            st.x=xx;st.y=yy;
            s.push(st);
        }
        for(int i=;i<=n;i++){
            if(i==)printf("%d",s.top().y);
            else printf(" %d",s.top().y);
            s.pop();
        }
        putchar('\n');
    }
    return ;
}