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 binarysearch(int maksimi, int x1, int y1, int x2, int y2);
 
int main()
{
	ios_base::sync_with_stdio(0);
 
	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;
	int maxhyppy = 0;
	int minhyppy = 0;
	bool maxfound = false;
 
	for (int b = maksimi / 2; b >= 1; b /= 2)
	{
		while (pituus + b < maksimi && summataulukkolaske(i + pituus + b, j + pituus + b, i, j) <= puut) pituus += b;
	}
	if (summataulukkolaske(i + pituus, j + pituus, i, j) == puut)
	{
		maxhyppy = pituus;
		maxfound = true;
	}
	//cout << pituus << '\n';
 
 
	for (int b = maksimi / 2; b >= 1; b /= 2)
	{
		while (pituus - b >= 0 && summataulukkolaske(i+pituus-b, j+pituus-b, i, j) >= puut) pituus -= b;
	}
	if (summataulukkolaske(i + pituus, j + pituus, i, j) == puut)
	{
		minhyppy = pituus;
	}
	//cout << pituus << '\n';
 
	/*while (nykyisetpuut <= puut && pituus < maksimi)
	{
		int lkm2 = summataulukkolaske(i + pituus, j + pituus, i, j);
		if (lkm2 == puut)
			lasku += 1;
 
		nykyisetpuut = lkm2;
 
		pituus++;
	}
	int x1 = binarysearch(maksimi, i, j, i, j);
	int y1 = x1;
 
	int max = x1;
	while (summataulukkolaske(x1 + 1, y1 + 1, i, j) == puut)
	{
		x1 += 1;
		y1 += 1;
		max += 1;
	}
	int min = x1;
	while (summataulukkolaske(x1 - 1, y1 - 1, i, j) == puut)
	{
		x1 -= 1;
		y1 -= 1;
		min -= 1;
	}*/
	if (maxfound)
		return maxhyppy - minhyppy + 1;
	else
		return 0;
}
 
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;
}
 
/*int binarysearch(int maksimi, int x1, int y1, int x2, int y2)
{
	int laskutulos = summataulukkolaske(x1, y1, x2, y2);
	if (laskutulos == puut)
	{
		return x1, y1;
	}
	else
	{
		if (laskutulos < puut)
		{
			x1 = x2 + int((maksimi - x1) / 2);
			y1 = y2 + int((maksimi - x1) / 2);
		}
		if (laskutulos > puut)
		{
			x1 = x2 - int((maksimi - int(x1 / 2)) / 2);
			y1 = y2 - int((maksimi - int(y1 / 2)) / 2);
		}
 
		binarysearch(maksimi, x1, y1, x2, y2);
	}
}*/
 

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

Test 12

Group: 3

Verdict: ACCEPTED

Test 13

Group: 3

Verdict: ACCEPTED

Test 14

Group: 3

Verdict: ACCEPTED

Test 15

Group: 3

Verdict: ACCEPTED