【牛客-Shopee 2019校招部分编程题汇总】[编程题]Shopee的办公室（二）dp（python3）

题目描述

时间限制：C/C++ 1秒，其他语言2秒

空间限制：C/C++ 32M，其他语言64M

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

思路题解

dfs(超时)

原来想的用dfs，发现在x，y=30+的时候就已经20+位数了，已经超时。遂放弃。dfs代码：

count=0
x,y,n=map(int,input().split())
m=[[0 for i in range(max(x,y)+1)] for j in range(max(x,y)+1)]
m[x][y]=1
for i in range(n):
    xi,yi=map(int,input().split())
    m[xi][yi]=2
def dfs(m,i,j):
    global count
    if i>x or j>y or m[i][j]==2:
        return
    if m[i][j]==1:
        count+=1
        return
    dfs(m,i+1,j)
    dfs(m,i,j+1)
dfs(m,0,0)
print(count)

dp

终点dp-无法到达点dp(错误):
后来觉得用dp，原来想得比较简单，先算无boss的dp，再减去各个boss。
ans=ans-dp[xi][yi]*dp[x-xi][y-yi]，但是当boss一多且位置复杂时就难以分析。
代码就不贴了。

改进dp:
直接把无法到达的点也算入dp中，遂正确

count=0
x,y,n=map(int,input().split())
# dp[i][j]=dp[i-1][j]+dp[i][j-1]
dp=[[1 for i in range(y+1)] for j in range(x+1)]
for i in range(n):
    xi,yi=map(int,input().split())
    dp[xi][yi]=0
for i in range(0,x+1):
    for j in range(1,y+1):
        if dp[i][j]!=0:
            if i==0:
                if dp[i][j-1]==0:dp[i][j]=0
                else:dp[i][j]=dp[i][j-1]
            else:dp[i][j]=dp[i-1][j]+dp[i][j-1]
ans=dp[x][y]
print(ans)

举个栗子：