![Codeforces 835C - Star sky - [二维前缀和] Codeforces 835C - Star sky - [二维前缀和]](https://image.shishitao.com:8440/aHR0cHM6Ly9ia3FzaW1nLmlrYWZhbi5jb20vdXBsb2FkL2NoYXRncHQtcy5wbmc%2FIQ%3D%3D.png?!?w=700)
题目链接:http://codeforces.com/problemset/problem/835/C
题意:
在天空上划定一个直角坐标系,有 $n$ 颗星星,每颗星星都有坐标 $(x_i,y_i)$,星星初始亮度为 $s_i$,所有星星的亮度有个上限 $c$。
在时刻 $0$,每颗星星都是初始亮度 $s_i$,然后每过一个单位时间,星星亮度都增加 $1$,如果亮度一旦超过 $c$ 就立刻变为 $0$。
现在有 $q$ 次观察天空的机会,观察时刻为 $t_i$,观察的视野为左下角为 $(x_{1i},y_{1i})$,右上角为 $(x_{2i},y_{2i})$ 的平行于坐标轴的长方形。要求出每次观察时,在视野中的星星的亮度和。
题解:
首先星星亮度是周期变化的,周期长度是 $c+1$,所以我们只需要求出周期内每个时刻星星的亮度。
然后我们再用一个二维前缀和,就能 $O(1)$ 查出每个时刻任意长方形范围的星星亮度和。
注意一点,可能多个星星重叠在一个坐标上。
AC代码:
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+;
const int C=;
const int X=, Y=;
int n,q,c;
struct Star{
int x,y,s;
}star[N];
int sum[C][X][Y];
int main()
{
ios::sync_with_stdio();
cin.tie(), cout.tie(); cin>>n>>q>>c;
for(int i=;i<=n;i++) cin>>star[i].x>>star[i].y>>star[i].s; memset(sum,,sizeof(sum));
for(int t=;t<=c;t++)
{
for(int i=;i<=n;i++)
sum[t][star[i].x][star[i].y]+=(star[i].s+t)%(c+); for(int x=;x<=;x++)
for(int y=;y<=;y++)
sum[t][x][y]+=sum[t][x-][y]+sum[t][x][y-]-sum[t][x-][y-];
} for(int i=,t,x1,y1,x2,y2;i<=q;i++)
{
cin>>t>>x1>>y1>>x2>>y2;
cout<<sum[t%(c+)][x2][y2]-sum[t%(c+)][x2][y1-]-sum[t%(c+)][x1-][y2]+sum[t%(c+)][x1-][y1-]<<"\n";
}
}