Lehmät

use std::io::{self, BufReader, BufRead};
 
fn main() {
    let stdin = io::stdin();
    let buf = BufReader::new(stdin);
    let mut lines = buf.lines();
    let dim = lines.next().unwrap().unwrap();
    let mut split = dim.split(" ");
 
    let n: usize = split.next().unwrap().parse().unwrap();
    let _m: usize = split.next().unwrap().parse().unwrap();
 
    let mut start: (usize, usize) = (0, 0);
    let mut end: (usize, usize) = (0, 0);
    let mut cows = 0;
 
    for y in 0..n {
        let line = lines.next().unwrap().unwrap();
        if start == (0, 0) {
            if let Some(x) = line.find('*') {
                start = (y, x);
                end = (n, line.rfind('*').unwrap())
            }
        } else {
            if y > start.0 || y > end.0 {
                if line.find('*').is_none() {
                    end = (y - 1, end.1);
                    continue;
                }
 
                cows += line
                    .chars()
                    .enumerate()
                    .fold(0, |acc, (i, c)| if i > start.1 && i < end.1 && c == '@' { acc + 1 } else { acc + 0 });
            }
        }
    }
 
    println!("{cows}")
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: WRONG ANSWER

Test 3

Group: 1, 2

Verdict: ACCEPTED

Test 4

Group: 1, 2

Verdict: ACCEPTED

Test 5

Group: 1, 2

Verdict: WRONG ANSWER

Test 6

Group: 2

Verdict: ACCEPTED

Test 7

Group: 2

Verdict: ACCEPTED

Test 8

Group: 2

Verdict: ACCEPTED

Test 9

Group: 2

Verdict: ACCEPTED