拜访-美团校招原题

来源:互联网 发布:中美贸易顺差数据 编辑:程序博客网 时间:2024/04/27 01:12

题目描述

现在有一个城市销售经理,需要从公司出发,去拜访市内的商家,已知他的位置以及商家的位置,但是由于城市道路交通的原因,他只能在左右中选择一个方向,在上下中选择一个方向,现在问他有多少种方案到达商家地址。

给定一个地图map及它的长宽nm,其中1代表经理位置,2代表商家位置,-1代表不能经过的地区,0代表可以经过的地区,请返回方案数,保证一定存在合法路径。保证矩阵的长宽都小于等于10。

测试样例:
[[0,1,0],[2,0,0]],2,3
返回:2

解题思路:利用动态规划思想,假设1在左上角,2在右下角,则经理只能向右或者向下走。先初始化1所在的列和行,遇到-1之前全部初始化为1,之后全为0.然后再计算出走到每个点的方法数,那么也就知道了到目标的方法数。

class Visit {public:    int countPath(vector<vector<int> > map, int n, int m)     {        // write code here        int startx = -1, starty = -1;        int endx = -1, endy = -1;        for (int i = 0; i < n; i++)        {            for (int j = 0; j < m; j++)            {                if (map[i][j] == 1)                {                    startx = i;                    starty = j;                }                if (map[i][j] == 2)                {                    endx = i;                    endy = j;                }            }            if (startx != -1 && endx != -1)                break;        }                int diffx = startx > endx ? -1 : 1;        int diffy = starty > endy ? -1 : 1;        for (int i = startx + diffx; i != endx + diffx; i += diffx)        map[i][starty] = map[i][starty] == -1 ? 0 : map[i - diffx][starty];        for (int i = starty + diffy; i != diffy + endy; i += diffy)        map[startx][i] = map[startx][i] == -1 ? 0 : map[startx][i - diffy];                for (int i = startx + diffx; i != endx + diffx; i += diffx)        {            for (int j = starty + diffy; j != endy + diffy; j += diffy)            {                map[i][j] = map[i][j] == -1 ? 0 : map[i][j - diffy] + map[i - diffx][j];                   }        }                return map[endx][endy];    }};



1 0