题目描述
现在有一个城市销售经理,需要从公司出发,去拜访市内的商家,已知他的位置以及商家的位置,但是由于城市道路交通的原因,他只能在左右中选择一个方向,在上下中选择一个方向,现在问他有多少种方案到达商家地址。
给定一个地图map及它的长宽n和m,其中1代表经理位置,2代表商家位置,-1代表不能经过的地区,0代表可以经过的地区,请返回方案数,保证一定存在合法路径。保证矩阵的长宽都小于等于10。
测试样例:
[[0,1,0],[2,0,0]],2,3
返回:2
public int countPath(int[][] map, int n, int m) {
int mX = 0, mY = 0;//经理坐标
int sX = 0, sY = 0;//商店坐标
for (int i = 0; i < n; i++) {
for (int j = 0; j < m; j++) {
if (map[i][j] == 1) {
mX = i;
mY = j;
continue;
}
if (map[i][j] == 2) {
sX = i;
sY = j;
}
}
}
//根据经理和商家的相对位置判断向左还是向右
int diffX = mX < sX ? 1 : -1;
int diffY = mY < sY ? 1 : -1;
//初始化经理坐标同一列,遇见-1之前全部置1,之后全部置为0
for (int i = mX + diffX; i != sX + diffX; i += diffX) {
map[i][mY] = map[i][mY] == -1 ? 0 : map[i - diffX][mY];
}
for (int i = mY + diffY; i != sY + diffY; i += diffY) {
map[mX][i] = map[mX][i] == -1 ? 0 : map[mX][i-diffY];
}
for (int i = mX + diffX; i != sX+diffX ; i+=diffX) {
for (int j = mY + diffY; j != sY + diffY ; j+=diffY) {
map[i][j] = map[i][j] == -1 ? 0 : map[i-diffX][j] + map[i][j-diffY];
}
}
return map[sX][sY];
}