JAVA动态规划之特殊路径

题目描述：

给定m行n列的网格，有一个机器人从（0，0）出发，每一步可以向下或向右移动一步

在A[i][j]==1的位置设有障碍

n>=1,m>=1

public class Solution1_1 {
    public static int uniquePathsWithObstacles(int[][] A){
        int m=A.length;
        int n=A[0].length;
        if(m==0||n==0)
            return 0;

        int[][] f=new int [m][n];//构造辅助空间

        int i,j;
        for(i=0;i<m;i++)//构造边缘
            f[i][0]=1;
        for(j=1;j<n;j++)
            f[0][j]=1;//边缘
        Print(f);//输出一下辅助空间，便于观察

        for(i=0;i<m;i++){//构造边缘
            for(j=0;j<n;j++){
                if(A[i][j]==1){//障碍在i j处，则辅助空间该处为0
                    f[i][j]=0;
                    if(i==0){//如果障碍在0行，则该行之后所有均访问不到
                        for(int k=j;k<n;k++)
                            f[0][k]=0;
                    }
                    if(j==0){//如果障碍在0列，则该列之后所有列第一项均访问不到
                        for(int k=i;k<m;k++)
                            f[k][0]=0;
                    }
                }
            }
        }
        Print(f);//输出一下辅助空间，便于观察

        for(i=1;i<m;i++){
            for(j=1;j<n;j++){
                if(A[i][j]==1)//障碍在i j处，则辅助空间该处为0
                    f[i][j]=0;
                else {
                    f[i][j]=f[i-1][j]+f[i][j-1];
                }
            }
        }
        Print(f);//输出一下辅助空间，便于观察

        return f[m-1][n-1];
    }

    public static void Print(int[][] A)
    {
        int m=A.length;int n=A[0].length;
        for(int i=0;i<m;i++){//输出一下辅助空间，便于观察
            for(int j=0;j<n;j++)
                System.out.print(A[i][j]+" ");
            System.out.println();
        }
        System.out.println();
    }

    public static void main(String[] args) {
        int[][] A={{0,1,0,0},{0,0,1,0},{0,0,0,0}};//1为有障碍
        System.out.print(uniquePathsWithObstacles(A));
    }
}