BZOJ 2683: 简单题(CDQ分治 + 树状数组)

题意

你有一个\(N*N\)的棋盘,每一个格子内有一个整数,初始时的时候所有为\(0\),如今须要维护两种操做:c++

命令 参数限制 内容
\(1\ x\ y\ A\) \(1\le x,y \le N\),A是正整数 将格子\(x,y\)里的数字加上\(A\)
\(2\ x1\ y1\ x2\ y2\) \(1\le x1\le x2\le N,1\le y1\le y2\le N\) 输出\(x1\ y1\ x2\ y2\)这个矩形内的数字和
\(3\) 终止程序

\(1<=N<=500000\),操做数不超过\(200000\)个,内存限制\(20M\)git

题解

这个题是 cdq分治 的裸题吧。算法

一维:时间(按输入顺序就好了)数组

二维:\(x\)坐标(cdq分治)数据结构

三维:\(y\)坐标(树状数组)spa

这个题比较裸,可是cdq分治细节仍是有一点的(调的错误我能够列一版了。。)code

算法讲解

但我想简单讲一下cdq分治(由于网上不少都很坑没讲清楚)排序

cdq是专门解决多维偏序的问题,好比像这一道题统计二维矩形的权值,或者直接求高维偏序的个数。递归

若是不用cdq分治,就只能树套树或者KD-tree这种巨型工业数据结构。并且树套树经常空间和常数都很恐怖,而且很难写……内存

cdq分治是个比较好写的东西,但其中的思想十分的巧妙和神奇。

你应该学过归并排序求逆序对吧,那是最裸的cdq了。他就是利用了左边的答案来更新右边的答案,cdq就是在这个方面不一样于普通的分治。

它每次算答案,只能在右边区间算也就是\([mid+1,r]\)。这是为何能够这样呢,由于你初始给它的序列,按这样算的话,绝对只会算它原序列左边的贡献,不会算到右边去。(想想,为何) 这个只须要本身模拟下分治的区间划分和左右区间考虑就好了。

这就能够会强制使你一开始的那一维有序,对答案计算是正确的。(但切记最后给你的序列不必定是按你给它的顺序了!!!)

而后它中间会有一个排序比较的过程,这就可使第二个维度变得有序了。(最后的序列必定是第二维度有序的) 而后根据前两个维度算答案就好了,后面的维度全都是附加在这两个维度上面的。

总的步骤:

  1. 分开(递归计算左区间和右区间)
  2. 计算(用左区间来统计,右区间来加上贡献)
  3. 合并(将当前序列变得有序)

又回到题解

这道题,就是对于全部操做进行cdq分治(通常都是对于操做进行分治)。

第三维用树状数组统计\(y\)的前缀和就好了,由于\(x\)已经排好序了,因此能够直接算了。

左区间只执行Add操做,右区间只执行Sum操做。

对于一个询问操做,要将它拆成4个询问操做(就是相似询问二维前缀和),加加减减就好了。

注意几个细节(我调了好久的点)

  1. 树状数组清空的时候,下标不是val而是y
  2. 拆矩阵的时候,必定要不要写错下标;

而后多拍几组,写个暴力很容易查出来的。

代码

#include <bits/stdc++.h>
#define For(i, l, r) for(register int i = (l), _end_ = (int)(r); i <= _end_; ++i)
#define Fordown(i, r, l) for(register int i = (r), _end_ = (int)(l); i >= _end_; --i)
#define Set(a, v) memset(a, v, sizeof(a))
using namespace std;

inline bool chkmin(int &a, int b) {return b < a ? a = b, 1 : 0;}
inline bool chkmax(int &a, int b) {return b > a ? a = b, 1 : 0;}

inline int read() {
    int x = 0, fh = 1; char ch = getchar();
    for (; !isdigit(ch); ch = getchar() ) if (ch == '-') fh = -1;
    for (; isdigit(ch); ch = getchar() ) x = (x<<1) + (x<<3) + (ch ^ '0');
    return x * fh;
}

void File() {
#ifdef zjp_shadow
    freopen ("P2683.in", "r", stdin);
    freopen ("P2683.out", "w", stdout);
#endif
}

int n;

const int N = 800100;
struct Opt {
    int x, y, type, id, val;
    inline bool operator < (const Opt &rhs) const {
        return (x ^ rhs.x) ? x < rhs.x : type < rhs.type;
    }
};
Opt lt[N], tmp[N];

#define lowbit(x) (x & -(x))
struct Fenwick_Tree {
    int c[500100];
    inline void Add(int pos, int val) { for (; pos <= n; pos += lowbit(pos) ) c[pos] += val; }
    inline int Sum(int pos) { int res = 0; for (; pos; pos -= lowbit(pos) ) res += c[pos]; return res; }
    inline void Clear(int pos) { for (; pos <= n; pos += lowbit(pos) ) if (c[pos]) c[pos] = 0; else break; }
};
Fenwick_Tree T;

int ans[N];

void Cdq(int l, int r) {
    if (l == r) return ;
    int mid = (l + r) >> 1;
    Cdq(l, mid); Cdq(mid + 1, r);
    int lp = l, rp = mid + 1, o = l;
    while (lp <= mid && rp <= r) {
        if (lt[lp] < lt[rp]) {
            if (lt[lp].type == 1) T.Add(lt[lp].y, lt[lp].val);
            tmp[o ++] = lt[lp ++];
        } else {
            if (lt[rp].type == 2) ans[lt[rp].id] += lt[rp].val * T.Sum(lt[rp].y);
            tmp[o ++] = lt[rp ++];
        }
    }

    while (lp <= mid) tmp[o ++] = lt[lp ++];
    while (rp <= r) {
        if (lt[rp].type == 2) ans[lt[rp].id] += lt[rp].val * T.Sum(lt[rp].y);
        tmp[o ++] = lt[rp ++];
    }

    For (i, l, mid) if (lt[i].type == 1) T.Clear(lt[i].y);
    For (i, l, r) lt[i] = tmp[i];
}

int qcnt, acnt;
inline void Addq(int x, int y, int type, int id, int val) {
    lt[++qcnt] = (Opt){x, y, type, id, val};
}

int main () {
    File() ;
    n = read();
    for (;;) {
        int opt = read();
        if (opt == 3) break ;
        int xa, ya, xb, yb, val;
        if (opt == 1) {
            xa = read(); ya = read(); val = read();
            Addq(xa, ya, 1, 0, val);
        } else {
            xa = read(); ya = read();
            xb = read(); yb = read();
            Addq(xa - 1, ya - 1, 2, (++ acnt), 1);
            Addq(xa - 1, yb, 2, acnt, -1);
            Addq(xb, ya - 1, 2, acnt, -1);
            Addq(xb, yb, 2, acnt, 1);
        }
    }
    Cdq(1, qcnt);
    For (i, 1, acnt) printf ("%d\n", ans[i]);
    return 0;
}

相关文章
相关标签/搜索