《数据结构与算法实战-周强-2.2》——大炮打蚊子

题目

现在，我们用大炮来打蚊子：蚊子分布在一个M×N格的二维平面上，每只蚊子占据一格。向该平面的任意位置发射炮弹，炮弹的杀伤范围如下示意：

O
OXO
O

其中，X为炮弹落点中心，O为紧靠中心的四个有杀伤力的格子范围。若蚊子被炮弹命中（位于X格），一击毙命，若仅被杀伤（位于O格），则损失一半的生命力。也就是说，一次命中或者两次杀伤均可消灭蚊子。现在给出蚊子的分布情况以及连续k发炮弹的落点，给出每炮消灭的蚊子数。
输入格式:

第一行为两个不超过20的正整数M和N，中间空一格，表示二维平面有M行、N列。

接下来M行，每行有N个0或者#字符，其中#表示所在格子有蚊子。

接下来一行，包含一个不超过400的正整数k，表示发射炮弹的数量。

最后k行，每行包括一发炮弹的整数坐标x和y（0≤x<M，0≤y<N），之间用一个空格间隔。
输出格式:

对应输入的k发炮弹，输出共有k行，第i行即第i发炮弹消灭的蚊子数。
输入样例:

5 6
00#00#
000###
00#000
000000
00#000
2
1 2
1 4

输出样例:

0
2

分析

用二维列表去存储矩阵，然后要对炮战的落地点就行判断，不同区域的处理办法不同。

代码

MN=input().split(" ")
M,N=int(MN[0]),int(MN[1])
ls=list([i for i in range(N)] for j in range(M))
for i in range(M):
    tmp=input()   # string类型
    for j in range(N):
        if tmp[j]=='#':
            ls[i][j]=2
        else:
            ls[i][j]=0
k=eval(input())
lk=[]

def Hurt(ls,x,y,flag=0):    # flag 是默认参数
    if flag==1: # flag==1 表示的是炮弹周围
        if ls[x][y] == 2:
            ls[x][y]=1
            return 0
        elif ls[x][y] == 1:
            ls[x][y]=0
            return 1
        else:
            return 0
    else:   # flag==0 表示的是炮弹中心
        if ls[x][y] == 2 or ls[x][y] == 1 :
            ls[x][y]=0
            return 1
        else:
            return 0

def Pao(ls,x,y):
    sum=0   # 下面是计算中间区域，四个邻边和四个顶点
    if x==M-1 and y==N-1:
        sum += Hurt(ls, x, y-1,1)+Hurt(ls,x-1,y,1)+Hurt(ls,x,y)
    elif x==0 and y-1>=0 and y+1<N:
        sum+=Hurt(ls,x,y-1,1)+Hurt(ls,x,y+1,1)+Hurt(ls,x+1,y,1)+Hurt(ls,x,y)
    elif x==M-1 and y-1>=0 and y+1<N:
        sum+=Hurt(ls,x,y-1,1)+Hurt(ls,x,y+1,1)+Hurt(ls,x-1,y,1)+Hurt(ls,x,y)
    elif y==0 and x-1>=0 and x+1<M:
        sum+=Hurt(ls,x-1,y,1)+Hurt(ls,x+1,y,1)+Hurt(ls,x,y+1,1)+Hurt(ls,x,y)
    elif y==N-1 and x-1>=0 and x+1<M:
        sum+=Hurt(ls,x,y-1,1)+Hurt(ls,x-1,y,1)+Hurt(ls,x+1,y,1)+Hurt(ls,x,y)
    elif x==0 and y==0:
        sum+=Hurt(ls,x+1,y,1)+Hurt(ls,x,y+1,1)+Hurt(ls,x,y)
    elif x==0 and y==N-1:
        sum+=Hurt(ls,x,y-1,1)+Hurt(ls,x+1,y,1)+Hurt(ls,x,y)
    elif x==M-1 and y==0:
        sum+=Hurt(ls,x-1,y,1)+Hurt(ls,x,y+1,1)+Hurt(ls,x,y)
    else:
        sum += Hurt(ls, x - 1, y,1) + Hurt(ls, x, y - 1,1) + Hurt(ls, x + 1, y,1) + Hurt(ls, x, y + 1,1)+Hurt(ls,x,y)
    return sum

for i in range(k):
    lk.append(input())
for i in range(k):
    tmp=lk[i].split(" ")
    x=int(tmp[0])
    y=int(tmp[1])
    print(Pao(ls, x, y))