Tontti

#include <iostream>
#include <math.h>
#include <stdio.h>
 
using namespace std;
 
int main() {
	cin.sync_with_stdio(false);
	int i, j, n, m, k, x, q, w, o, p;
	int sz, s;
	cin >> n >> m >> k;
	char input[n][m];
	int a[n*m];
	int f[n*m];
	for (i = 0; i < n; i++)
		cin >> input[i];
	sz = (int)ceil(sqrt((double)k));
	x = 0;
	for (i = 0; i < n; i++)
		for (j = 0; j < m; j++) {
			a[i*m+j] = !(input[i][j]=='.');
			f[i*m+j] = 0;
			if(k<2)
				x += a[i*m+j]^k^1;
		}
	int c = max(sz,2);
	int g = 0, h = 0, d, z, first = 1, scale = 1;
	for (s = c; s <= min(n,m); s++) {
		z = s - 1;
		first = 1;
		for (i = z; i < n; i++) {
			if (i > 0 && i < n) {
				q = i * m;
				w = q - s * m;
				for (o = 0; o < s; o++, q++, w++)
					h += (a[q] - a[w]);
				g = h;
			}
			scale = 0;
			for (j = z; j < m; j++) {
				d = i*m+j;
				int t = f[d];
				if (first) {
					first = 0;
					if (s == c) {
						for (q = 0; q < s; q++) {
							for (w = 0; w < s; w++) {
								t += a[q*m+w];
							}
						}
						g = h = t;
					} else {
						int iz = (i-z)*m, jz = j-z, iq = i * m, jq = j;
						for (q = 0; q < z; q++) {
							t += a[iz+jq];
							t += a[iq+jz];
							iq-=m;jq--;
						}
						g = h = t+a[iz+jz];
					}
					scale = 1;
				} else if (scale) {
					q = j;
					w = q - s;
					p = (i-z)*m;
					for (o = 0; o < s; o++, p+=m) {
						g += (a[p+q] - a[p+w]);
					}
				} else
					scale = 1;
				f[d] = t = g;
				x += (t == k);
			}
		}
	}
	cout << x << "\n";
	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: TIME LIMIT EXCEEDED

Test 7

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 8

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 9

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 10

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 11

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 12

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 13

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 14

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 15

Group: 3

Verdict: TIME LIMIT EXCEEDED