[编程题] Shopee的办公室(二)
时间限制:1秒
空间限制:32768K
shopee的办公室非常大,小虾同学的位置坐落在右上角,而大门却在左下角,可以把所有位置抽象为一个网格(门口的坐标为0,0),小虾同学很聪明,每次只向下,或者向左走,因为这样最容易接近目的地,但是小虾同学不想让自己的boss们看到自己经常在他们面前出没,或者迟到被发现。他决定研究一下如果他不通过boss们的位置,他可以有多少种走法?
输入描述:
第一行 x,y,n (0<x<=30, 0<y<=30, 0<=n<= 20) 表示x,y小虾的座位坐标,n 表示boss的数量( n <= 20)
接下来有n行, 表示boss们的坐标(0<xi<= x, 0<yi<=y,不会和小虾位置重合)
x1, y1
x2, y2
……
xn, yn
输出描述:
输出小虾有多少种走法
输入例子1:
3 3 2
1 1
2 2
输出例子1:
4
题目链接(牛客网)
思路:
这题坑人的地方很多,一开始以为直接暴力搜索就好了,不出所料超时了,正确的姿势是动态规划,因为只能向下或向左,所以到每一个坐标(x,y)的走法等于(x+1,y)的走法加上(x,y+1)的走法。原始坐标的那一排和那一列的坐标走法都是1,因为只能向左或向下,记得如果有boss的坐标是0,而且一旦遇到boss,那一排或那一列后面的坐标走法全部为0,因为遇上boss就过不去了。结果很大,要用大数类BigInteger。
代码:
package cn.diao.main;
import java.math.BigInteger;
import java.util.Arrays;
import java.util.Scanner;
public class Main
{
static int cnt=0;
public static void main(String[] args) throws Exception
{
Scanner sc=new Scanner(System.in);
int x=sc.nextInt();
int y=sc.nextInt();
int n=sc.nextInt();
int[][] arr=new int[x+1][y+1];
//数据很大,要使用大数类储存
BigInteger[][] dp=new BigInteger[x+1][y+1];
for(int i=1;i<=n;i++)
{
int xx=sc.nextInt();
int yy=sc.nextInt();
arr[xx][yy]=1;
//有boss的格子直接初始化为0
dp[xx][yy]=BigInteger.ZERO;
}
//给小虾只能往左走的坐标赋值为1,因为他只能往左走,所以只有一种方法
for(int i=x-1;i>=0;i--)
{
//当坐标点上没有boss,才能赋值为1
if(arr[i][y]==0)
dp[i][y]=BigInteger.ONE;
//当坐标点有boss时,因为只能往左走,之前的点都被boss挡住了,所以后面所有的坐标都赋值为0
else
{
for(int ii=i-1;ii>=0;ii--)
dp[ii][y]=BigInteger.ZERO;
break ;
}
}
//给小虾只能往下走的坐标赋值为1,因为他只能往下走,所以只有一种方法
for(int j=y-1;j>=0;j--)
{
//当坐标点上没有boss,才能赋值为1
if(arr[x][j]==0)
dp[x][j]=BigInteger.ONE;
//当坐标点有boss时,因为只能往下走,之前的点都被boss挡住了,所以后面所有的坐标都赋值为0
else
{
for(int jj=j-1;jj>=0;jj--)
dp[x][jj]=BigInteger.ZERO;
break ;
}
}
//到每一个坐标dp[x][y]的走法等于dp[x+1][y]+dp[x][y+1]
for(int i=x-1;i>=0;i--)
for(int j=y-1;j>=0;j--)
{
if(arr[i][j]==0)
dp[i][j]=dp[i+1][j].add(dp[i][j+1]);
}
System.out.println(dp[0][0].toString());
}
}