Task: | Forest density |
Sender: | ashum-ta |
Submission time: | 2024-09-23 16:45:41 +0300 |
Language: | Python3 (PyPy3) |
Status: | READY |
Result: | ACCEPTED |
test | verdict | time | |
---|---|---|---|
#1 | ACCEPTED | 0.04 s | details |
#2 | ACCEPTED | 0.72 s | details |
#3 | ACCEPTED | 0.66 s | details |
Code
def solve():# Input readingn, q = map(int, input().split())grid = [input().strip() for _ in range(n)]# Initialize the prefix sum array with an extra row and column of zeros for easy calculationsprefix = [[0] * (n + 1) for _ in range(n + 1)]# Build the prefix sum arrayfor i in range(1, n + 1):for j in range(1, n + 1):# Convert grid values to 1 if it's a tree ('*'), otherwise 0prefix[i][j] = (1 if grid[i-1][j-1] == '*' else 0) \+ prefix[i-1][j] + prefix[i][j-1] - prefix[i-1][j-1]# Process each queryfor _ in range(q):y1, x1, y2, x2 = map(int, input().split())# Use the inclusion-exclusion principle to calculate the number of trees in the rectangleresult = prefix[y2][x2] - prefix[y1-1][x2] - prefix[y2][x1-1] + prefix[y1-1][x1-1]print(result)# Example usage:solve()
Test details
Test 1
Verdict: ACCEPTED
input |
---|
10 100 **.*.*.**. *.**.*..*. .*****.**. **....***. ... |
correct output |
---|
10 14 5 7 8 ... |
user output |
---|
10 14 5 7 8 ... Truncated |
Test 2
Verdict: ACCEPTED
input |
---|
1000 200000 **.**.****..**.***..**.***.**.... |
correct output |
---|
41079 2824 15631 1548 8483 ... |
user output |
---|
41079 2824 15631 1548 8483 ... Truncated |
Test 3
Verdict: ACCEPTED
input |
---|
1000 200000 ******************************... |
correct output |
---|
1000000 1000000 1000000 1000000 1000000 ... |
user output |
---|
1000000 1000000 1000000 1000000 1000000 ... Truncated |