CodeForces 835C Star sky

题目链接：http://codeforces.com/contest/835/problem/C
题意：有一片星空这个星空里有n个星星，每个星星的位置为(x_i,y_i)，初始亮度为s_i，所有星星的最大亮度为c，没过一个时间，亮度加1，达到最大c，就置为0，现在有q次询问，每此询问t时刻，(x₁,y₁)(x₂,y₂)范围里所有星星的总亮度
解析：可以直接二维树状数组来做预处理，也可以用类似一个前缀和的东西来预处理出来，比如dp[i][j][k]表示从(1,1)到(i,j)这个区域亮度为k的星星有多少个，然后查询的时候减一下即可

#include <bits/stdc++.h>
using namespace std;
int dp[105][105][15];
int main(void)
{
    int n,q,c;
    scanf("%d %d %d",&n,&q,&c);
    for(int i=0;i<n;i++)
    {
        int x,y,c;
        scanf("%d %d %d",&x,&y,&c);
        dp[x][y][c]++;
    }
    for(int i=1;i<105;i++)
    {
        for(int j=1;j<105;j++)
        {
            for(int k=0;k<=c;k++)
                dp[i][j][k] += dp[i-1][j][k]+dp[i][j-1][k]-dp[i-1][j-1][k];
        }
    }
    while(q--)
    {
        int x1,y1,x2,y2,t;
        scanf("%d %d %d %d %d",&t,&x1,&y1,&x2,&y2);
        int ans = 0;
        for(int i=0;i<=c;i++)
        {
            int v = (t+i)%(c+1);
            int tmp = dp[x2][y2][i]+dp[x1-1][y1-1][i]-dp[x1-1][y1][i]-dp[x1][y1-1][i];
            ans += v*tmp;
        }
        printf("%d\n",ans);
    }
    return 0;
}

#include <bits/stdc++.h>
using namespace std;
int a[105][105][20];
int lowbit(int x)
{
    return x&(-x);
}
void update(int x,int y,int s)
{
    for(int i=x;i<105;i+=lowbit(i))
    {
        for(int j=y;j<105;j+=lowbit(j))
            a[i][j][s]++;
    }
}
int query(int x,int y,int s)
{
    int ans = 0;
    for(int i=x;i>0;i-=lowbit(i))
    {
        for(int j=y;j>0;j-=lowbit(j))
            ans+=a[i][j][s];
    }
    return ans;
}
int main()
{
    int n,q,c;
    cin>>n>>q>>c;
    while(n--)
    {
        int x,y,s;
        scanf("%d%d%d",&x,&y,&s);
        update(x,y,s);
    }
    while(q--)
    {
        int t,x1,y1,x2,y2;
        scanf("%d%d%d%d%d",&t,&x1,&y1,&x2,&y2);
        int ans = 0;
        for(int i=0;i<=c;i++)
        {
            int t1 = query(x2,y2,i);
            int t2 = query(x1-1,y2,i);
            int t3 = query(x2,y1-1,i);
            int t4 = query(x1-1,y1-1,i);
            ans += (i+t)%(c+1)*(t1-t2-t3+t4);
        }
        printf("%d\n",ans);
    }
    return 0;
}