[TYVJ P1716/BZOJ 3132 上帝造题的七分钟] 二维树状数组区间修改、区间查询

原创 2017年01月03日 03:09:32

[TYVJ P1716/BZOJ 3132 上帝造题的七分钟] 二维树状数组区间修改、区间查询

知识点:data structure binary index tree

1. 题目链接

[TYVJ P1716 上帝造题的七分钟]
类似题目还有[POJ 2155 Matrix]。只不过这个题目就是一个二维的区间翻转,单点查询。一个树状数组就可以做了,但是可以套上这题的模板做,如果你不闲麻烦的话。。

2. 题意描述

有一个nm的矩阵,有2种操作(操作次数不超过200000次)。
操作一:二维区间修改,选一个子矩阵【以(a,b),(c,d)为顶点】,进行区间加上delta
操作二:二维区间求和,选一个子矩阵【以(a,b),(c,d)为顶点】,求这个子矩阵的所有元素之和。

3. 解题思路

BZOJ 3132这个题目好像找不到了?就在TYVJ上找到这个题目了。
二维树状数组区间修改、区间查询的裸题。但是这个题目有点剧毒,可能是卡常特别严重?TLE了很多次,加上超级输入挂,依旧TLE,非得再加一个输出挂。
二维树状数组区间修改,区间更新原理:
跟一维树状数组区间修改,区间更新的原理类似。也是用差分增量数据来表示一个前缀和。然后用前缀和描述区间。
对于原矩阵A,用矩阵dij表示(1,1)(n,m)的增量,那么子矩阵(1,1)(x,y)的和为:

sum=i=1xj=1y[di,j(xi+1)(yi+1)]=i=1xj=1y{di,j[(x+1)(y+1)(x+1)j(y+1)i+ij]}=i=1xj=1y[(x+1)(y+1)dij(x+1)jdij(y+1)idij+ijdij]

因此,用四个树状数组分别维护:dij,(idij),(jdij),(ijdij)
最后,贴一个二维树状数组的模板。

4. 实现代码

#include <bits/stdc++.h>
using namespace std;

typedef long long LL;
const int MAXN = 2048 + 1;

int n, m;
// 输入挂
namespace IO {
    const int MT = 10 * 1024 * 1024;  /// 10MB 请注意输入数据的大小!!!
    char IO_BUF[MT];
    int IO_PTR, IO_SZ;
    /// 要记得把这一行添加到main函数第一行!!!
    void begin() {
        IO_PTR = 0;
        IO_SZ = fread (IO_BUF, 1, MT, stdin);
    }
    template<typename T>
    inline bool scan_d (T & t) {
        while (IO_PTR < IO_SZ && IO_BUF[IO_PTR] != '-' && (IO_BUF[IO_PTR] < '0' || IO_BUF[IO_PTR] > '9'))
            IO_PTR ++;
        if (IO_PTR >= IO_SZ) return false;
        bool sgn = false;
        if (IO_BUF[IO_PTR] == '-') sgn = true, IO_PTR ++;
        for (t = 0; IO_PTR < IO_SZ && '0' <= IO_BUF[IO_PTR] && IO_BUF[IO_PTR] <= '9'; IO_PTR ++)
            t = t * 10 + IO_BUF[IO_PTR] - '0';
        if (sgn) t = -t;
        return true;
    }
    inline bool scan_s (char s[]) {
        while (IO_PTR < IO_SZ && (IO_BUF[IO_PTR] == ' ' || IO_BUF[IO_PTR] == '\n') ) IO_PTR ++;
        if (IO_PTR >= IO_SZ) return false;
        int len = 0;
        while (IO_PTR < IO_SZ && IO_BUF[IO_PTR] != ' ' && IO_BUF[IO_PTR] != '\n')
            s[len ++] = IO_BUF[IO_PTR], IO_PTR ++;
        s[len] = '\0';
        return true;
    }
    template<typename T>
    void print(T x) {
        static char s[33], *s1; s1 = s;
        if (!x) *s1++ = '0';
        if (x < 0) putchar('-'), x = -x;
        while(x) *s1++ = (x % 10 + '0'), x /= 10;
        while(s1-- != s) putchar(*s1);
    }
    template<typename T>
    void println(T x) {
        print(x); putchar('\n');
    }
};

template<class T>
struct BIT_2D {
    struct Core {
        T C[4][MAXN][MAXN];
        void I() { memset(C, 0, sizeof(C)); }
        inline T lowbit(const T& x) { return x & (-x); }
        void update(const int& x0, const int& y0, const T& val) {
            for(int x = x0; x <= n; x += x & (-x)) {
                for(int y = y0; y <= m; y += y & (-y)) {
                    C[0][x][y] += val;
                    C[1][x][y] += val * x0;
                    C[2][x][y] += val * y0;
                    C[3][x][y] += val * x0 * y0;
                }
            }
        }
        T query(const int& x0, const int& y0) {
            T ret = 0;
            for(int x = x0; x > 0; x -= x & (-x)) {
                for(int y = y0; y > 0; y -= y & (-y)) {
                    ret += C[0][x][y] * (x0 + 1) * (y0 + 1);
                    ret -= C[1][x][y] * (y0 + 1);
                    ret -= C[2][x][y] * (x0 + 1);
                    ret += C[3][x][y];
                }
            }
            return ret;
        }
    };
    Core core;
    void update(const int& x0, const int& y0, const int& x1, const int& y1, const int& val) {
        int dx[4] = {x0, x1 + 1, x0, x1 + 1};
        int dy[4] = {y0, y1 + 1, y1 + 1, y0};
        int v[4] = {val, val, -val, -val};
        for(int i = 0; i < 4; ++i) {
            core.update(dx[i], dy[i], v[i]);
        }
    }
    int query(const int& x0, const int& y0, const int& x1, const int& y1) {
        int ret = 0;
        int dx[4] = {x0 - 1, x1, x0 - 1, x1};
        int dy[4] = {y0 - 1, y1, y1, y0 - 1};
        int v[4] = {1, 1, -1, -1};
        for(int i = 0; i < 4; ++i) {
            ret += v[i] * core.query(dx[i], dy[i]);
        }
        return ret;
    }
};
BIT_2D<int> bit;

char op[5];
int x[2], y[2], delta;

int main() {
#ifdef ___LOCAL_WONZY___
    freopen("input.txt", "r", stdin);
#endif // ___LOCAL_WONZY___
    IO::begin();
    while(IO::scan_s(op)) {
        if(op[0] == 'X') {
            IO::scan_d(n);
            IO::scan_d(m);
//            a.I(), b.I(), c.I(), d.I();
            continue;
        }
        IO::scan_d(x[0]);
        IO::scan_d(y[0]);
        IO::scan_d(x[1]);
        IO::scan_d(y[1]);
        if(op[0] == 'k') {
            IO::println(bit.query(x[0], y[0], x[1], y[1]));
        } else {
            IO::scan_d(delta);
            bit.update(x[0], y[0], x[1], y[1], delta);
        }
    }
    return 0;
}
版权声明:本文为博主原创文章,未经博主允许不得转载。 举报

相关文章推荐

【BZOJ 3132】上帝造题的七分钟【二维树状数组】

Description“第一分钟,X说,要有矩阵,于是便有了一个里面写满了0的n×m矩阵。 第二分钟,L说,要能修改,于是便有了将左上角为(a,b),右下角为(c,d)的一个矩形区域内的全部数字加上...

TYVJ 1941 BZOJ3038 上帝造题的七分钟2 并查集+树状数组

背景 XLk觉得《上帝造题的七分钟》不太过瘾,于是有了第二部。 描述“第一分钟,X说,要有数列,于是便给定了一个正整数数列。 第二分钟,L说,要能修改,于是便有了对一段数中每个数都开平方(下取整...

我是如何成为一名python大咖的?

人生苦短,都说必须python,那么我分享下我是如何从小白成为Python资深开发者的吧。2014年我大学刚毕业..

BZOJ 3038 上帝造题的七分钟2 树状数组+并查集

题目大意:给定一个序列,提供下列操作: 1.将[l.r]区间内每个数a[i]变为sqrt(a[i]) 2.查询[l,r]区间的和 同BZOJ3211 花神游历各国 链接:htt...

[BZOJ3038]上帝造题的七分钟2 树状数组+并查集

考试的时候用了两个树状数组去优化,暴力修改,树状数组维护修改后区间差值还有最终求和,最后骗了40分。。 这道题有好多种做法,求和好说,最主要的是开方。这道题过的关键就是掌握一点:在数据范围内,最多开方...

【BZOJ3211】【花神游历各国】&【BZOJ3038】上帝造题的七分钟2(树状数组+链表)

DescriptionInputOutput每次x=1时,每行一个整数,表示这次旅行的开心度Sample Input4 1 100 5 5 5 1 1 2 2 1 2 1 1 2 2 2 ...

【树状数组】[CQBZOJ2719]上帝造题的七分钟I

二维树状数组区间修改、查询裸题。#include #define MAXN 2048 int n,m,c[MAXN+10][MAXN+10],ci[MAXN+10][MAXN+10],cj[MAXN+...

【BZOJ3132】上帝造题的七分钟

Description“第一分钟,X说,要有矩阵,于是便有了一个里面写满了0的n×m矩阵。第二分钟,L说,要能修改,于是便有了将左上角为(a,b),右下角为(c,d)的一个矩形区域内的全部数字加上一个...

POJ 1195 Mobile phones(二维树状数组,点修改,区间查询)

典型的二维树状数组 二维树状数组和一维树状数组原理是一样的。 我们先回顾一维树状数组 C【maxn】,

POJ2155【二维树状数组,区间修改,点查询?】【又被输入输出坑】

先上代码。一会解释 #include #include #include using namespace std; //关键还是对树状数组理解比较深,差不多算是知晓了。 //区间修改,点查询 i...

POJ 2155-Matrix(二维树状数组-区间修改 单点查询)

Matrix Time Limit: 3000MS   Memory Limit: 65536K Total Submissions: 27355   Accept...
返回顶部
收藏助手
不良信息举报
您举报文章:深度学习:神经网络中的前向传播和反向传播算法推导
举报原因:
原因补充:

(最多只允许输入30个字)