Array Transformer-分块思想

OIwiki上讲的很好

题目

题目链接
给三个数字n, m, u，接下来给长度为n的数组，给m次操作，每次操作输入四个数l, r, v, p，每次操作在[l, r]这个区间里计算严格小于v的的个数，个数记为k，然后将a[p]的值修改为u * k / (r - l + 1)。操作完成以后要求输出a数组。

解决

如果暴力做的话，会很不优雅的t掉…m范围是1e4, 1 <= l, r <= n(3e5)，复杂度就3e9…
所以需要优化一下…（一开始想了下线段树，也还是8星，哈哈我是傻逼 ）
我们这时候就需要用到分块的思想了。

将数组分为sqrt(n)块，每一个块升序排个序，这样，当[l, r]这个区间左右两边处于不完整的块的时候，暴力找，如果中间完整的话，一个块一个块直接二分查找第一个大于等于v的，然后这个二分得到的下标左边就是小于v的个数。

#include <cstdio>
#include <cstring>
#include <string>
#include <cstdlib>
#include <algorithm>
#include <iostream>
#include <queue>
#include <set>
#include <cmath>
#include <map>
#include <bitset>
#include <vector>
#define pb push_back
using namespace std;
typedef long long ll;
const int inf = 0x3f3f3f3f;
const ll INF = 0x3f3f3f3f3f3f3f3f;
#define  pq priority_queue
#define eps 1e-10
#define PI acos(-1.0)

int n, m, p, tot, len, l, r;
ll u, v;
const int maxn = 3e6 + 5;
ll a[maxn];
int ind[maxn];
vector<ll>d[maxn];


void solve(){
	int L = ind[l], R = ind[r];
	int cnt = 0;
	if(L == R){
		//直接暴力跑l,r之间的
		for(int i = l; i <= r; i++){
			if(a[i] < v)
				cnt++;
		}
	}
	else{
		//先暴力跑左边不完整的块和右边不完整的块
		for(int i = l; L == ind[i]; i++) 
			if(a[i] < v) 
				cnt++;
		for(int i = L + 1; i < R; i++){
			int res = lower_bound(d[i].begin(), d[i].end(), v) - d[i].begin();
			cnt += res;
		}
		for(int i = r; R == ind[i]; i--)
			if(a[i] < v)
				cnt++;
	}
	a[p] = u * cnt / (r - l + 1);
	//错的：不能这样写，因为d数组是sort后的数组，不是与a数组一一对应。d[L][p - 1] = u * cnt / (r - l + 1);
	for(int i = 0; i < len; i++){
		//第几个块的第几个位置
		d[ind[p]][i] = a[(ind[p] - 1) * len + i + 1];;
		if((ind[p] - 1) * len + i + 1 == n)
			break;
	}
	sort(d[ind[p]].begin(), d[ind[p]].end());
}

int main(){
	scanf("%d %d %lld", &n, &m, &u);
	len = sqrt(n);
	tot = 0;
	for(int i = 1; i <= n; i++){
		scanf("%lld", &a[i]);
		ind[i] = (i - 1) / len + 1;
		d[ind[i]].pb(a[i]);
	}
	for(int i = 1; i <= (n - 1) / len + 1; i++)
		sort(d[i].begin(), d[i].end());
	while(m--){
		scanf("%d %d %lld %d", &l, &r, &v, &p);
		solve();
	}
	for(int i = 1; i <= n; i++)
		printf("%lld\n", a[i]);

	return 0;
}