11月--1293--小马过河

时间:2021-03-27 04:07:58

Description
cloverprince养着一只小马,活泼可爱,很喜欢在外面玩。有一天,小马驮着一袋盐从一条小河上经过,到了河对岸,他发现,身上的盐轻了不少,好像自己的力气变大了似的。从此,小马开始喜欢过河。每天都要经过小河,到河对岸的草地上去玩。
有一天,小马玩得很开心,不知不觉,天色已经晚了。小马想起了王子要他晚上早点回家,否则王子会生气的。于是,小马想用最快的速度过河。
河水有缓有急,不同地方深浅也不一样,因此,在河的不同地方,行走的速度是有区别的。小马把河划分成N*M个矩形区域,像一个矩阵,南北N个区,东西M个区。小马现在在河的南岸,要走到北岸去。小马可以从任何邻接南岸的区域进入河中;在河里,每次只能从当前区域走到北方、东北、西北三个相邻区域;只要走到最北边的任何一个区域,下一步就可以上岸。小马在每个区域中需要停留的时间不尽相同。 请你帮助小马计算一下它在河里至少需要停留多长时间才能过河。 

Input
包含多个测试用例,不超过5个,连续给出 
每个测试用例中:
第一行是两个正整数N,M,表示区域的个数。 
第2至第N+1行,每行M个正整数,是一个N行M列矩阵。每个元素表示如果进入这个区域,需要逗留的时间。
区域给出的顺序:逐行由北向南,每行由西向东。
接着是下一个测试用例。
读到N=M=0时结束。
数据限制:
0 < N,M <= 1000 
矩阵中每个元素不超过1000之间。

Output
对于每个测试用例,输出一行,仅包含一个整数,就是过河需要的最短时间。

Sample Input
5 5
1 2 3 4 5 
2 2 3 3 4 
5 4 1 2 8
9 2 3 6 3
1 1 6 2 2
0 0

Sample Output
7

Hint
顺次经过第5行第1列、第4行第2列、第3行第3列、第2行第2列、第1行第1列,在河里停留的时间总共是1+2+1+2+1=7。

矩阵从第一行到第N行表示由北向南,求小马从南岸到北岸的最短时间,可以使用动态规划来求解,从北向南计算小马走到每一行总最短逗留时间。初始状态即最北岸一行的最短逗留时间,第i+1行的最短逗留时间就是该区域停留时间,加上从该区域东北、北方、西北三个方向逗留时间中的最小值。最后得到第N行的最小值,就是小马从南到北过河的最少停留时间。

状态转移方程为:

dp[i][j] = min( dp[i-1][j-1] , dp[i-1][j] , dp[i-1][j+1]) + s[i][j];

需要注意的是单行多列的情况。


#include <iostream>
#include <stdio.h>
#include <string>
#include <string.h>
#include <vector>
#include <list>
#include <math.h>
#include <algorithm>

using namespace std;

int s[1001][1001],dp[1001][1001];

int min2(int a,int b)
{
    if(a<b)
        return a;
    else return b;
}

int min3(int a,int b,int c)
{
    if(a<b)
    {
        if(a<c)return a;
        else return c;
    }
    else 
    {
        if(b<c)return b;
        else return c;
    }
}

int main()
{
    int n, m , time;

    while(scanf("%d %d",&n,&m) && n!=0 && m!=0)
    {
        if( n == 0 && m==0 )break;
        int i,j;
        for( i = 0 ; i < n ; i++ )
            for( j = 0 ; j < m ; j++)
                scanf("%d",&s[i][j]);
        for( j = 0 ; j < m ; j++ )
            dp[0][j] = s[0][j];
        for( i = 1 ; i < n ; i++ )
        {
            if( m > 1 )
            {
                dp[i][0] = min2( dp[i-1][0] , dp[i-1][1] ) + s[i][0];
                dp[i][m-1] = min2( dp[i-1][m-2] , dp[i-1][m-1] ) + s[i][m-1];
                for( j = 1 ; j < (m-1) ; j++ )
                    dp[i][j] = min3( dp[i-1][j-1] , dp[i-1][j] , dp[i-1][j+1]) + s[i][j];
            }
            else
            {
                    dp[i][0] = dp[i-1][0] + s[i][0];
            }
        }
        time = 9999999;
        for( j = 0 ; j < m ; j++ )
            if( dp[n-1][j] < time ) time = dp[n-1][j];
        printf("%d\n",time);
    }
    
    return 1;    
}