NYOJ172 小珂的图表

 

小珂的图表

时间限制：1000 ms  |  内存限制：65535 KB

难度：2

 

描述

小珂最近有一个麻烦,就是要统计一下指定区域中有几个方格被染黑了.表格的布局及表格各个位置的坐标如下所示.

有三种操作命令,

BLACK x,y,l 表示把以坐标（x,y）为左上角顶点，（x+l-1,y+l-1）为右下角顶点的矩形染黑。

WHITE x,y,l 表示吧指定区域染白。

TEST x,y,l 表示计算指定区域的黑块的个数。

说明：如果 x,y,x+l-1 ,y+l-1超出图表的范围，就只计算图表内部的。

 

输入

第一行有一个整数n(0<n<100)，表示有n条命令，随后的n行有n个指令。

输出

遇到TEST命令，把结果输出并换行。

样例输入

5
BLACK 1 1 2
BLACK 2 2 2
TEST 1 1 3
WHITE 2 1 1
TEST 1 1 3

样例输出

7
6

来源

POJ

 

#include <cstdio>
#include <iostream>
#include <cstring>

using namespace std;

int map[105][105];
int col[105][105];

int lowbit(int i)
{
    return i & (-i);
}

void add(int x, int y, int color)
{
    if(color == col[x][y])
        return ;
    col[x][y] = color;
    for(int i = x; i <= 100; i += lowbit(i))
        for(int j = y; j <= 100; j += lowbit(j))
            map[i][j] += color;
}

int getSum(int x, int y)
{
    int tot = 0;
    for(int i = x; i > 0; i -= lowbit(i))
        for(int j = y; j > 0; j -= lowbit(j))
            tot += map[i][j];
    return tot;
}

int main()
{
    int T, x, y, x1, y1, len;
    char cmd[10];
    memset(map, 0, sizeof(map));
    memset(col, -1, sizeof(col));  // 初始标记每个格子都是白色 
    
    scanf("%d", &T);
    while(T--)
    {
        scanf("%s%d%d%d", cmd, &x, &y, &len);
        
        x1 = x+len-1 > 100 ? 100 : x+len-1;
        y1 = y+len-1 > 100 ? 100 : y+len-1;
        
        if(cmd[0] == 'B')
        {
            for(int i = x; i <= x1; ++i)
                for(int j = y; j <= y1; ++j)
                    add(i, j, 1);
        }
        else if(cmd[0] == 'W')
        {
            for(int i = x; i <= x1; ++i)
                for(int j = y; j <= y1; ++j)
                    add(i, j, -1);
        }
        else
        {
            printf("%d\n", getSum(x1,y1)+getSum(x-1,y-1)-getSum(x-1,y1)-getSum(x1,y-1));
        }
    }
    return 0;
}