Tontti

#include <iostream>
#include <string>
#include <algorithm>
 
using namespace std;
 
int korkeus, leveys, puut;
int summataulukko[2000][2000];
int vakio;
 
void readInput(int x, int y);
int lasku(int i, int j, int korkeus, int leveys, int puut);
int summataulukkolaske(int x1, int y1, int x2, int y2);
 
int main()
{
	cin >> korkeus >> leveys >> puut;
 
	readInput(leveys, korkeus);
 
	int lkm = 0;
	
	for (int j = 0; j < korkeus; j++)
	{
		for (int i = 0; i < leveys; i++)
		{
			lkm = lkm + lasku(i, j, korkeus, leveys, puut);
		}
	}
 
	cout << lkm;
	cin.get();
	cin.get();
}
 
void readInput(int x, int y)
{
 
	for (int j = 0; j < y; j++)
	{
		for (int i = 0; i < x; i++)
		{
			char a;
			cin >> a;
			vakio = 0;
			if (a == '*')
			{
				vakio = 1;
			}
 
			if (i == 0 && j == 0)
			{
				summataulukko[i][j] = vakio;
			}
			else if (i == 0)
			{
				summataulukko[i][j] = summataulukko[i][j - 1] + vakio;
			}
			else if (j == 0)
			{
				summataulukko[i][j] = summataulukko[i - 1][j] + vakio;
			}
			else
			{
				summataulukko[i][j] = summataulukko[i - 1][j] + summataulukko[i][j - 1] + vakio - summataulukko[i - 1][j - 1];
			}
		}
	}
}
 
int lasku(int i, int j, int korkeus, int leveys, int puut)
{
	int maksimi = min(korkeus - j, leveys - i);
	int pituus = 0;
	int nykyisetpuut = 0;
	int lasku = 0;
 
	while (nykyisetpuut <= puut && pituus < maksimi)
	{
		int lkm2 = summataulukkolaske(i + pituus, j + pituus, i, j);
		if (lkm2 == puut)
			lasku += 1;
 
		nykyisetpuut = lkm2;
 
		pituus++;
	}
 
	return lasku;
}
 
int summataulukkolaske(int x1, int y1, int x2, int y2)
{
	int i = x2 - 1;
	int j = y2 - 1;
	int vastaus = 0;
 
	if (i < 0 && j < 0)
	{
		vastaus = summataulukko[x1][y1];
	}
	else if (i < 0)
	{
		vastaus = summataulukko[x1][y1] - summataulukko[x1][j];
	}
	else if (j < 0)
	{
		vastaus = summataulukko[x1][y1] - summataulukko[i][y1];
	}
	else
	{
		vastaus = summataulukko[x1][y1] - summataulukko[i][y1] - summataulukko[x1][j] + summataulukko[i][j];
	}
 
	return vastaus;
}
 

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: 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