Lehmät

def IsOutOfBounds(x, y):
    if (x < 0) or (y < 0) or (x >= width) or (y >= height):
        return True
    return False
 
 
dimensionsInput = input()
height = int(dimensionsInput.split(" ")[0])
width = int(dimensionsInput.split(" ")[1])
 
fieldMap = []
for i in range(0, height):
    fieldMap.append(input().strip())
 
leftEdge, rightEdge, topEdge, bottomEdge = -1, -1, -1, -1
# Find the top left corner of the pen
found = False
for y in range(height):
    if found: break
    for x in range(width):
        if fieldMap[y][x] == "*":
            leftEdge = x
            topEdge = y
            found = True
            break
# Find the right and bottom edge
rightEdge = leftEdge
bottomEdge = topEdge
while not IsOutOfBounds(rightEdge + 1, topEdge):
    if(fieldMap[topEdge][rightEdge + 1] == "*"):
        rightEdge += 1
    else:
        break
while not IsOutOfBounds(leftEdge, bottomEdge + 1):
    if(fieldMap[bottomEdge + 1][leftEdge] == "*"):
        bottomEdge += 1
    else:
        break
 
# Count the cows
count = 0
for y in range(topEdge + 1, bottomEdge):
    for x in range(leftEdge + 1, rightEdge):
        if fieldMap[y][x] == "@":
            count += 1
 
print(count)
 
 

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: ACCEPTED

Test 3

Group: 1, 2

Verdict: ACCEPTED

Test 4

Group: 1, 2

Verdict: ACCEPTED

Test 5

Group: 1, 2

Verdict: ACCEPTED

Test 6

Group: 2

Verdict: ACCEPTED

Test 7

Group: 2

Verdict: ACCEPTED

Test 8

Group: 2

Verdict: ACCEPTED

Test 9

Group: 2

Verdict: ACCEPTED