CSES - Datatähti 2016 alku - Results
Submission details
Task:Tontti
Sender:rxz9000
Submission time:2015-10-11 22:52:34 +0300
Language:C++
Status:READY
Result:47
Feedback
groupverdictscore
#1ACCEPTED14
#2ACCEPTED33
#30
Test results
testverdicttimegroup
#1ACCEPTED0.06 s1details
#2ACCEPTED0.06 s1details
#3ACCEPTED0.06 s1details
#4ACCEPTED0.05 s1details
#5ACCEPTED0.05 s1details
#6ACCEPTED0.07 s2details
#7ACCEPTED0.06 s2details
#8ACCEPTED0.06 s2details
#9ACCEPTED0.06 s2details
#10ACCEPTED0.06 s2details
#11--3details
#12ACCEPTED0.54 s3details
#13ACCEPTED0.18 s3details
#14ACCEPTED0.22 s3details
#15ACCEPTED0.22 s3details

Code

#include <iostream>
#include <vector>
#include <string>
#include <algorithm>
#include <cmath>
using namespace std;

int main() {
	ios::sync_with_stdio(false);

	int height;
	cin >> height;

	int width;
	cin >> width;

	int idealAmountOfTrees;
	cin >> idealAmountOfTrees;

	vector<string> forest;
	forest.resize(height);

	for (int i = 0; i < height; i++) {
		cin >> forest[i];
	}

	vector<vector<int> > sums;
	sums.resize(height + 1);
	for (int i = 0; i <= height; i++) {
		sums[i].resize(width + 1);
	}


	for (int i = 0; i <= height; i++) {
		sums[i][0] = 0;
	}

	for (int j = 1; j <= width; j++) {
		sums[0][j] = 0;
	}

	int amountOfTrees = 0;
	for (int i = 1; i <= height; i++) {
		for (int j = 1; j <= width; j++) {
			sums[i][j] = sums[i - 1][j] - sums[i - 1][j - 1] + sums[i][j - 1];
			if (forest[i - 1].at(j - 1) == '*') {
				sums[i][j]++;
				amountOfTrees++;
			}
		}
	}

	if (amountOfTrees < idealAmountOfTrees) {
		cout << "0" << endl;
	} else {

		int waysOfChoosing = 0;
		const int firstMSL = (int)ceil(sqrt(idealAmountOfTrees));

		for (int i = 0; i < height; i++) {
			for (int j = 0; j < width; j++) {
				int minSl = firstMSL;
				int maxSl = min(height - i, width - j);
				int midSl;
				while (minSl <= maxSl) {
					midSl = (minSl + maxSl) / 2;
					int treesInSquare = sums[i + midSl][j + midSl] - sums[i + midSl][j] - sums[i][j + midSl] + sums[i][j];
					if (treesInSquare == idealAmountOfTrees) {
						waysOfChoosing++;
						int l1 = midSl - 1;
						while (l1 >= minSl && (sums[i + l1][j + l1] - sums[i + l1][j] - sums[i][j + l1] + sums[i][j]) == idealAmountOfTrees) {
							waysOfChoosing++;
							l1--;
						}
						int l2 = midSl + 1;
						while (l2 <= maxSl && (sums[i + l2][j + l2] - sums[i + l2][j] - sums[i][j + l2] + sums[i][j]) == idealAmountOfTrees) {
							waysOfChoosing++;
							l2++;
						}
						break;
					}
					else if (treesInSquare < idealAmountOfTrees) {
						minSl = midSl + 1;
					}
					else {
						maxSl = midSl - 1;
					}
				}
			}
		}

	cout << waysOfChoosing << endl;

	}
	return 0;
}

Test details

Test 1

Group: 1

Verdict: ACCEPTED

input
10 10 1
......*...
.......*..
*..*....*.
*....*....
...

correct output
94

user output
94

Test 2

Group: 1

Verdict: ACCEPTED

input
10 10 5
**********
**********
**********
**********
...

correct output
0

user output
0

Test 3

Group: 1

Verdict: ACCEPTED

input
10 10 10
**...*...*
*..*.**.*.
...**.*..*
*...**.*..
...

correct output
4

user output
4

Test 4

Group: 1

Verdict: ACCEPTED

input
10 10 5
****......
*.*.**..**
....*.*..*
...*.***..
...

correct output
16

user output
16

Test 5

Group: 1

Verdict: ACCEPTED

input
10 10 2
**.***..*.
...*.*....
.***.*...*
***.***..*
...

correct output
30

user output
30

Test 6

Group: 2

Verdict: ACCEPTED

input
500 500 1
.................................

correct output
9552040

user output
9552040

Test 7

Group: 2

Verdict: ACCEPTED

input
500 500 5
.................................

correct output
1536063

user output
1536063

Test 8

Group: 2

Verdict: ACCEPTED

input
500 500 25000
**...*...**..*.*..*.**.*..*.*....

correct output
288

user output
288

Test 9

Group: 2

Verdict: ACCEPTED

input
500 500 12500
**.**.*..*...*.**...*.***........

correct output
786

user output
786

Test 10

Group: 2

Verdict: ACCEPTED

input
500 500 5000
.*.*.**..*.*.**.**..*..**...*....

correct output
1763

user output
1763

Test 11

Group: 3

Verdict:

input
2000 2000 1
.................................

correct output
489611392

user output
(empty)

Test 12

Group: 3

Verdict: ACCEPTED

input
2000 2000 5
.................................

correct output
120725884

user output
120725884

Test 13

Group: 3

Verdict: ACCEPTED

input
2000 2000 400000
..*..**.**.**.*.***...**.*..**...

correct output
1849

user output
1849

Test 14

Group: 3

Verdict: ACCEPTED

input
2000 2000 200000
***.*....*.*..*....**..*..*.*....

correct output
2665

user output
2665

Test 15

Group: 3

Verdict: ACCEPTED

input
2000 2000 80000
**.**...*.***.**....**.*....*....

correct output
5587

user output
5587