求查UB(?)
查看原帖
求查UB(?)
161876
201810tflsdjl楼主2022/7/27 17:50

rt,在线IDE上开O2和不开O2结果不一样,测试数据是第一个数据,这里给出一部分:

10 3
719 583 321 556 741 430 50 28 326 804
A 1 10 321
M 1 10 101
A 1 10 337

代码如下:

#include <cstdio>
#include <iostream>
#include <cmath>
#include <algorithm>
using std::lower_bound, std::sort;

typedef long long lol;
const int Maxn = 1e6+5, Maxnsq = 1e3+5;
lol h[Maxn], copy[Maxn], tag[Maxnsq];
int n, q, tot, start[Maxn], end[Maxn], block;
int query(int, int, lol);
void add(int, int, lol), update(int);

int main(){
//	freopen("data.in", "r", stdin);

	scanf("%d %d", &n, &q);
	block = sqrt(n); tot = n/block;
	for(int i=0; i<tot; i++){
		start[i] = i*block;
		end[i] = (i+1)*block;
		for(int j=start[i]; j<end[i]; j++){
			scanf("%d", &h[j]);
		}
	}
	start[tot] = tot*block; end[tot] = n;
	for(int i=tot*block; i<n; i++){
		scanf("%d", &h[i]);
	}
	for(int i=0; i<n; i++) copy[i] = h[i];
	for(int i=0; i<tot; i++){
		sort(copy+start[i], copy+end[i]);
	}
	int l, r, w, c;
	char ty;
	getchar();
	while(q--){
		scanf("%c", &ty);
		if(ty == 'M'){
			scanf("%d %d %lld\n", &l, &r, &w);
			add(l-1, r-1, w);
		}
		else{
			scanf("%d %d %lld\n", &l, &r, &c);
			printf("%d\n", query(l-1, r-1, c));
		}
	}

	return 0;
}

void add(int l, int r, lol w){
	int lb = l/block, rb = r/block;
	for(int i=l; i==l||i%block!=0; i++){
		h[i] += w;
	}
	update(lb);
	if(lb!=rb){
		for(int i=r; i==r||i%block!=block-1; i--){
			h[i] += w;
		}
	}
	update(rb);
	for(int i=lb+1; i<rb; i++){
		tag[i] += w;
	}
}

int query(int l, int r, lol c){
	int lb = l/block, rb = r/block, ans = 0;
	lol ct;
	for(int i=l; i==l||i%block!=0; i++){
		if(h[i]+tag[lb] >= c) ans++;
	}
	if(lb!=rb){
		for(int i=r; i==r||i%block!=block-1; i--){
			if(h[i]+tag[rb] >= c) ans++;
		}
	}
	for(int i=lb+1; i<rb; i++){
		ct = c-tag[i];
		ans += copy+(i+1)*block-lower_bound(copy+i*block, copy+(i+1)*block, ct);
	}
	return ans;
}

void update(int id){
	for(int i=start[id]; i<end[id]; i++){
		copy[i] = h[i]+tag[id];
		h[i] += tag[id];
	}
	tag[id] = 0;
	sort(copy+start[id], copy+end[id]);
}
2022/7/27 17:50
加载中...