OlympiadCode/luogu_py/P5079.py

# input parse
rows, columns = input().split(" ")
rows = int(rows)
columns = int(columns)

matrix = [["." for _ in range(columns + 2)] for _ in range(rows)]

for row in range(rows):
    user_input = input().split(" ")
    # 确保用户输入的数据长度与列数匹配
    if len(user_input) != columns:
        print("Error: Incorrect number of columns entered.")
        continue
    # 在第一个点和最后一个点之间插入用户输入
    matrix[row][1:-1] = user_input


# Subtask 1 (30 points)：矩阵中仅有数字1

# Subtask 2 (30 points)：矩阵中不含有数字 1 和 4

# Subtask 3 (40 points)：无特殊性质。

count1 = 0
count2 = 0
count3 = 0
result = ""
number_dict = {
    "050": "1",
    "434": "2",
    "335": "3",
    "315": "4",
    "534": "6",
    "115": "7",
    "535": "8",
    "435": "9",
    "525": "0",
}

first_row = 0
for column in range(columns):
    for row in range(rows):
        if matrix[row][column] == "#":
            count1 += 1
        if matrix[row][column + 1] == "#":
            count2 += 1
        if matrix[row][column + 2] == "#":
            count3 += 1
    number = str(count1) + str(count2) + str(count3)
    count1 = 0
    count2 = 0
    count3 = 0
    if number in number_dict.keys():
        if number == "434":
            for row in range(rows):
                try:
                    if matrix[row][column] == "#" and matrix[row + 1][column] == ".":
                        result += "2"
                        break
                    elif matrix[row][column] == "#" and matrix[row + 1][column] == "#":
                        result += "5"
                        break
                    else:
                        pass
                except IndexError:
                    pass
        else:
            result += number_dict[number]

print(result)