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;
}