LeetCode/cpp/number-of-laser-beams-in-a-bank.cpp

#include <algorithm>
#include <string>
#include <vector>

class Solution {
  public:
    int numberOfBeams(const std::vector<std::string> &bank) {
        int beams = 0;

        int last_row = 0;
        for (const auto &row : bank) {
            if (auto current_row = std::count(row.begin(), row.end(), '1');
                current_row != 0) {
                beams += last_row * current_row;
                last_row = current_row;
            }
        }

        return beams;
    }
};