D - Unidirectional TSP UVA - 116

D - Unidirectional TSP

  UVA - 116 

#include <iostream>
#include<string.h>
#include<stdio.h>
#include<algorithm>
using namespace std;
int n,m;
int INF=0x3f3f3f;
int dp[15][105],a[15][105];

int main()
{
    //freopen("E:\\file.txt","r",stdin);
    while(~scanf("%d%d",&m,&n))
    {
        int next[15][105]; //不知道为什么next拿到主函数外面就编译错误
        for(int i=0;i<m;i++)
    {
        for(int j=0;j<n;j++)
        {
            scanf("%d",&a[i][j]);
        }
    }
    int ans=INF,first=0;
    memset(next,0,sizeof(next));
    memset(dp,0,sizeof(dp));
    for(int j=n-1;j>=0;j--)  //以列为标准更新数据，最后一步的状态由它的前一步状态决定
    {
        for(int i=0;i<m;i++)
        {
        if(j==n-1)  dp[i][j]=a[i][j];  //如果是在最后一列 那我们知道它只有一步需要走
        else
        {
            dp[i][j]=INF;    //在其他列的值由它的下一步决定
            int rows[3]={i,i-1,i+1};
            if(i==0)      rows[1]=m-1;
            if(i==m-1)    rows[2]=0;  //i=m-1 就是它的最后一行
            sort(rows,rows+3); //从当前的状态到下一步有三种情况可以走
            for(int k=0;k<3;k++)
            {
                int v=dp[rows[k]][j+1]+a[i][j];
                if(v<dp[i][j])
                {
                dp[i][j]=v;
                next[i][j]=rows[k];
                }
            }

        }
         if(j==0&&dp[i][j]<ans)
        {
            ans=dp[i][j]; //dp[i][j] 为最终状态
            first=i;
        }
        }


    }
     cout<<first+1;
     int i=first;
     for(int j=1;j<n;j++)
     {
         cout<<' '<<next[i][j-1]+1;
         i=next[i][j-1];
     }
    cout<<endl<<ans<<endl;
    }


    return 0;
}