CSES - Datatähti 2016 alku - Results
Submission details
Task:Tontti
Sender:rxz9000
Submission time:2015-10-11 22:15:18 +0300
Language:C++
Status:READY
Result:47
Feedback
groupverdictscore
#1ACCEPTED14
#2ACCEPTED33
#30
Test results
testverdicttimegroup
#1ACCEPTED0.06 s1details
#2ACCEPTED0.06 s1details
#3ACCEPTED0.05 s1details
#4ACCEPTED0.06 s1details
#5ACCEPTED0.06 s1details
#6ACCEPTED0.07 s2details
#7ACCEPTED0.06 s2details
#8ACCEPTED0.06 s2details
#9ACCEPTED0.07 s2details
#10ACCEPTED0.07 s2details
#11--3details
#12ACCEPTED0.53 s3details
#13ACCEPTED0.17 s3details
#14ACCEPTED0.19 s3details
#15ACCEPTED0.21 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;
	}

	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]
				+ (forest[i - 1].at(j - 1) == '*' ? 1 : 0);
		}
	}

	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