Tontti

#include <iostream>
#include <string>
#include <cstdint>
typedef long long ll;
using namespace std;


uint32_t sumtable[1600][1600] = {0};

inline uint32_t S(uint32_t x, uint32_t y) {
    if (x<1 || y<1) {
        return 0;
    }
    return sumtable[x-1][y-1];
}

int main() {
    ll height;
    ll width;
    ll trees;

    cin >> height;
    cin >> width;
    cin >> trees;

    string temps;

    if (trees==0) {
        std::cout << 0 << std::endl;
        return 0;
    }
    ll linesum;
    for (ll y = 0; y < height; y++) {
        linesum = 0;
        cin >> temps;
        for (ll x = 0; x < width; x++) {
            if (temps[x]=='*') {
                linesum++;
            }
            sumtable[x][y] = linesum;
            if (y!=0) {
                sumtable[x][y] += sumtable[x][y-1];
            }
        }
    }
    ll res = 0;
    for (uint32_t x = 0; x < width; x++) {
        for (uint32_t y = 0; y < height; y++) {
            for (uint32_t s = 2; s < min(width-x, height-y)+1; s++) {
                if (S(x+s,y+s)-S(x,y+s)-S(x+s,y)+S(x,y) == trees) {
                    res++;
                }
            }
        }
    }

    if (trees == 1) {
        res += sumtable[width-1][height-1];
    }

    std::cout << res << std::endl;
    return 0;
}

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 1

Verdict: ACCEPTED

Test 3

Group: 1

Verdict: ACCEPTED

Test 4

Group: 1

Verdict: ACCEPTED

Test 5

Group: 1

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

Test 10

Group: 2

Verdict: ACCEPTED

Test 11

Group: 3

Verdict: RUNTIME ERROR

Test 12

Group: 3

Verdict: RUNTIME ERROR

Test 13

Group: 3

Verdict: RUNTIME ERROR

Test 14

Group: 3

Verdict: RUNTIME ERROR

Test 15

Group: 3

Verdict: RUNTIME ERROR