POJ 1195 Mobile phones 二维线段树

最新推荐文章于 2018-11-20 21:22:21 发布

原创最新推荐文章于 2018-11-20 21:22:21 发布 · 284 阅读

0 ·

CC 4.0 BY-SA版权

线段树同时被 2 个专栏收录

20 篇文章

订阅专栏

树套树

6 篇文章

订阅专栏

本文详细介绍了如何使用二维线段树解决特定矩阵操作问题，包括矩阵元素的更新及子矩阵求和等操作。通过具体代码实现了矩阵的初始化、元素更新及查询等功能，并对关键步骤进行了注释说明。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：

http://poj.org/problem?id=1195

题意：

给出一个s*s的矩阵，有以下四种操作：

0 S 把一个S*S的矩阵全初始化为0，这个操作仅在最初执行一次
1 X Y A 把(X, Y)处的元素加上A
2 L B R T 查询以(L, R)为左上角、(B, T)为右下角的矩阵内元素的和
3 结束操作

思路：

二维线段树，很裸的题目了。。。不裸的话，有二维线段树写起来很麻烦，而且空间很容易被卡，感觉有点鸡肋啊。。。

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

using namespace std;

const int N = 1000 + 26, INF = 0x3f3f3f3f;

int n;
int sum[N<<2][N<<2];
void push_upy(int ky, int kx)
{
    sum[kx][ky] = sum[kx][ky<<1] + sum[kx][ky<<1|1];
}
void push_upx(int ky, int kx)
{
    sum[kx][ky] = sum[kx<<1][ky] + sum[kx<<1|1][ky];
}
void updatey(int y, int L, int R, int ky, int kx, int val, int f)
{
    if(L == R)
    {
        if(f) sum[kx][ky] += val;
        else push_upx(ky, kx);
        return;
    }
    int mid = (L + R) >> 1;
    if(y <= mid) updatey(y, L, mid, ky << 1, kx, val, f);
    else updatey(y, mid + 1, R, ky << 1|1, kx, val, f);
    push_upy(ky, kx);
}
void updatex(int x, int y, int L, int R, int kx, int val)
{
    if(L == R)
    {
        updatey(y, 1, n, 1, kx, val, 1); return;
    }
    int mid = (L + R) >> 1;
    if(x <= mid) updatex(x, y, L, mid, kx << 1, val);
    else updatex(x, y, mid + 1, R, kx << 1|1, val);
    updatey(y, 1, n, 1, kx, val, 0);
}
int queryy(int ly, int ry, int L, int R, int ky, int kx)
{
    if(ly <= L && R <= ry) return sum[kx][ky];
    int mid = (L + R) >> 1, ans = 0;
    if(ly <= mid) ans += queryy(ly, ry, L, mid, ky << 1, kx);
    if(ry > mid) ans += queryy(ly, ry, mid + 1, R, ky << 1|1, kx);
    return ans;
}
int queryx(int lx, int rx, int ly, int ry, int L, int R, int kx)
{
    if(lx <= L && R <= rx) return queryy(ly, ry, 1, n, 1, kx);
    int mid = (L + R) >> 1, ans = 0;
    if(lx <= mid) ans += queryx(lx, rx, ly, ry, L, mid, kx << 1);
    if(rx > mid) ans += queryx(lx, rx, ly, ry, mid + 1, R, kx << 1|1);
    return ans;
}
int main()
{
    int op;
    //memset(sum, 0, sizeof sum);
    while(scanf("%d", &op), op != 3)
    {
        if(op == 0) scanf("%d", &n);
        else if(op == 1)
        {
            int x, y, val;
            scanf("%d%d%d", &x, &y, &val);
            x++, y++;
            updatex(x, y, 1, n, 1, val);
        }
        else if(op == 2)
        {
            int l, b, r, t;
            scanf("%d%d%d%d", &l, &b, &r, &t);
            l++, b++, r++, t++;
            printf("%d\n", queryx(l, r, b, t, 1, n, 1));
        }
    }
    return 0;
}