Niitty

#include <cmath>
#include <iostream>
#include <set>
#include <string>
#include <unordered_map>
#include <vector>

using namespace std;

int main() {
	set<char> meadowFlowers;
	int areas = 0;

	int n;
	string s;
	cin >> n;

	vector<vector<char>> meadow(n, vector<char>(n, 'A'));

	for (int i = 0; i < n; i++) {
		cin >> s;
		for (int j = 0; j < n; j++) {
			meadow.at(i).at(j) = s.at(j);
			meadowFlowers.insert(s.at(j));
		}
	}

	unordered_map<char, int> flowerMap;
	vector<char> flowerVector(meadowFlowers.begin(), meadowFlowers.end());
	const int flowerNum = flowerVector.size();

	// Map flowers to index
	for (int i = 0; i < flowerNum; i++) {
		flowerMap[flowerVector.at(i)] = i;
	}
	vector<vector<vector<int>>> meadowPrefixSums(
		 flowerNum, vector<vector<int>>(n + 1, vector<int>(n + 1, 0)));

	// Calculate prefix sums for each flower type
	for (int f = 0; f < flowerNum; ++f) {
		char flowerType = flowerVector.at(f);
		for (int i = 1; i <= n; i++) {
			for (int j = 1; j <= n; j++) {
				meadowPrefixSums.at(f).at(i).at(j) =
					 meadowPrefixSums.at(f).at(i - 1).at(j) +
					 meadowPrefixSums.at(f).at(i).at(j - 1) -
					 meadowPrefixSums.at(f).at(i - 1).at(j - 1) +
					 (meadow.at(i - 1).at(j - 1) == flowerType ? 1 : 0);
			}
		}
	}

	// Iterate through every possible window size
	for (int winHeight = 1; winHeight <= n; winHeight++) {
		for (int winWidth = 1; winWidth <= n; winWidth++) {

			// Now slide the window over the grid
			for (int i = 1; i <= n - winHeight + 1; i++) {
				for (int j = 1; j <= n - winWidth + 1; j++) {
					int k = i + winHeight - 1;
					int l = j + winWidth - 1;

					vector<int> sum(flowerNum, 0);
					int found = 0;

					// Calculate the sum of each flower type in the current window
					for (int f = 0; f < flowerNum; f++) {
						int flowerCount =
							 meadowPrefixSums.at(f).at(k).at(l) -
							 meadowPrefixSums.at(f).at(i - 1).at(l) -
							 meadowPrefixSums.at(f).at(k).at(j - 1) +
							 meadowPrefixSums.at(f).at(i - 1).at(j - 1);
						sum.at(f) = flowerCount;
						if (sum.at(f) > 0) {
							++found;
						}
					}

					// If all flower types are found in the current window, increment
					// areas
					if (found == flowerNum) {
						++areas;
					}
				}
			}
		}
	}

	cout << areas;
}

Test 1

Group: 1, 2, 3, 4, 5, 6

Verdict: ACCEPTED

Test 2

Group: 1, 2, 3, 4, 5, 6

Verdict: ACCEPTED

Test 3

Group: 1, 2, 3, 4, 5, 6

Verdict: ACCEPTED

Test 4

Group: 1, 2, 3, 4, 5, 6

Verdict: ACCEPTED

Test 5

Group: 1, 2, 3, 4, 5, 6

Verdict: ACCEPTED

Test 6

Group: 2, 3, 4, 5, 6

Verdict: ACCEPTED

Test 7

Group: 2, 3, 4, 5, 6

Verdict: ACCEPTED

Test 8

Group: 2, 3, 4, 5, 6

Verdict: ACCEPTED

Test 9

Group: 2, 3, 4, 5, 6

Verdict: ACCEPTED

Test 10

Group: 3, 4, 5, 6

Verdict: ACCEPTED

Test 11

Group: 3, 4, 5, 6

Verdict: ACCEPTED

Test 12

Group: 3, 4, 5, 6

Verdict: ACCEPTED

Test 13

Group: 3, 4, 5, 6

Verdict: ACCEPTED

Test 14

Group: 4, 5, 6

Verdict: TIME LIMIT EXCEEDED

Test 15

Group: 4, 5, 6

Verdict: TIME LIMIT EXCEEDED

Test 16

Group: 4, 5, 6

Verdict: ACCEPTED

Test 17

Group: 4, 5, 6

Verdict: TIME LIMIT EXCEEDED

Test 18

Group: 5, 6

Verdict: TIME LIMIT EXCEEDED

Test 19

Group: 5, 6

Verdict: TIME LIMIT EXCEEDED

Test 20

Group: 5, 6

Verdict: TIME LIMIT EXCEEDED

Test 21

Group: 5, 6

Verdict: TIME LIMIT EXCEEDED

Test 22

Group: 6

Verdict: TIME LIMIT EXCEEDED

Test 23

Group: 6

Verdict: TIME LIMIT EXCEEDED

Test 24

Group: 6

Verdict: TIME LIMIT EXCEEDED

Test 25

Group: 6

Verdict: TIME LIMIT EXCEEDED