LOJ6278 数列分块入门2

LOJ6278 数列分块入门 2

标签

• 分块入门

前言

• 我的csdn和博客园是同步的，欢迎来访danzh-博客园~

简明题意

• 给一个序列，需要支持两种操作：
  1. 区间加和
  2. 查询区间中小于c的数的个数

思路

• 分块可以很好的处理这一类题~
• 首先分块，然后对每一块内排序。可以用一个vector保存每一块的数。
• 对于查询操作：整块的二分一下，不整块的暴力统计一下。
• 对于修改操作：整块的可以打标记，不整块的可以直接暴力对原数组修改，然后再把这一块的值重新导入

注意事项

• 注意最后一块的元素的个数!=len。当处理最后一块时，要特判一下。

总结

• 无

AC代码

#include<cstdio>
#include<algorithm>
#include<cmath>
#include<vector>
using namespace std;

const int maxn = 5e4 + 10;

int n, a[maxn];
int pos[maxn], len, tag[maxn];
vector<int> b[maxn];

void reset(int id)
{
    b[id].clear();
    for (int i = (id - 1) * len + 1; i <= min(id * len, n); i++)
        b[id].push_back(a[i]);
    sort(b[id].begin(), b[id].end());
}

void change(int l, int r, int c)
{
    for (int i = l; i <= min(pos[l] * len, r); i++)
        a[i] += c;
    reset(pos[l]);

    if (pos[l] != pos[r])
    {
        for (int i = r; i >= (pos[r] - 1) * len + 1; i--)
            a[i] += c;
        reset(pos[r]);
    }

    for (int i = pos[l] + 1; i <= pos[r] - 1; i++)
        tag[i] += c;
}

int cal(int l, int r, int c)
{
    int cnt = 0;
    for (int i = l; i <= min(pos[l] * len, r); i++)
        cnt += (a[i] + tag[pos[i]]) < c;

    if (pos[l] != pos[r])
        for (int i = r; i >= max((pos[r] - 1) * len + 1, l); i--)
            cnt += (a[i] + tag[pos[i]]) < c;

    for (int i = pos[l] + 1; i <= pos[r] - 1; i++)
        cnt += lower_bound(b[i].begin(), b[i].end(), c - tag[i]) - b[i].begin();

    return cnt;
}

void solve()
{
    scanf("%d", &n);
    len = sqrt(n);
    for (int i = 1; i <= n; i++)
        scanf("%d", &a[i]), b[pos[i] = (i - 1) / len + 1].push_back(a[i]);

    //预处理
    for (int i = 1; i <= pos[n]; i++)
        sort(b[i].begin(), b[i].end());

    for (int i = 1; i <= n; i++)
    {
        int opt, l, r, c;
        scanf("%d%d%d%d", &opt, &l, &r, &c);
        if (opt == 0)
            change(l, r, c);
        else
            printf("%d\n", cal(l, r, c * c));
    }
}

int main()
{
    //freopen("Testin.txt", "r", stdin);
    //freopen("Testout.txt", "w", stdout);
    solve();
    return 0;
}