萌新求助,线段树1那题,本地AC提交RE
查看原帖
萌新求助,线段树1那题,本地AC提交RE
482610
Mortidesperatslav楼主2024/3/23 13:53

代码:

#include<bits/stdc++.h>
#define int long long
using namespace std;
#define mxn 200005
struct node{
	int l, r, vl, lzy;
}tr[mxn << 2 | 3];
int a[mxn], n, q;
int ls(int u){
	return u << 1;
}
int rs(int u){
	return u << 1 | 1;
}
int glz(int u, int k){
	tr[u].lzy += k;
}
int gv(int u, int k){
	tr[u].vl += k;
}
int md(int l, int r){
	return (l + r) >> 1;
}
void pup(int u){
	tr[u].vl = tr[ls(u)].vl + tr[rs(u)].vl;
}
void build(int u, int l, int r){
	tr[u].l = l;
	tr[u].r = r;
	if (l == r){
		tr[u].vl = a[l];
		return;
	}
	int mid = md(l, r);
	build(ls(u), l, mid);
	build(rs(u), mid + 1, r);
	pup(u);
}
bool inr(int l, int r, int ql, int qr){
	return (ql <= l && qr >= r);
}
bool otr(int l, int r, int ql, int qr){
	return (ql > r || qr < l);
}
void pdn(int u, int l, int r){
	int mid = md(l, r);
	glz(ls(u), tr[u].lzy);
	gv(ls(u), tr[u].lzy * (mid - l + 1));
	glz(rs(u), tr[u].lzy);
	gv(rs(u), tr[u].lzy * (r - mid));
	tr[u].lzy = 0;
}
void upd(int u, int ql, int qr, int x){
	if (inr(tr[u].l, tr[u].r, ql, qr)){
		glz(u, x);
		gv(u, x * (tr[u].r - tr[u].l + 1));
		return;
	}
	if (otr(tr[u].l, tr[u].r, ql, qr))
		return;
	pdn(u, ql, qr);
	int mid = md(tr[u].l, tr[u].r);
	if (ql <= mid)
		upd(ls(u), ql, qr, x);
	if (qr > mid)
		upd(rs(u), ql, qr, x);
}
int qry(int u, int ql, int qr){
	int res = 0;
	if (inr(tr[u].l, tr[u].r, ql, qr))
		return tr[u].vl;
	if (otr(tr[u].l, tr[u].r, ql, qr))
		return 0;
	pdn(u, ql, qr);
	int mid = md(tr[u].l, tr[u].r);
	if (ql <= mid)
		res += qry(ls(u), ql, qr);
	if (qr > mid)
		res += qry(rs(u), ql, qr);
	return res;
}
signed main(){
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	cin >> n >> q;
	for (int i = 1; i <= n; i++)
		cin >> a[i];
	build(1, 1, n);
	while (q--){
		int op;
		cin >> op;
		if (op == 1){
			int l, r, x;
			cin >> l >> r >> x;
			upd(1, l, r, x); 
		}else{
			int l, r;
			cin >> l >> r;
			cout << qry(1, l, r) << "\n"; 
		}
	}
}

我下载了数据点1,本地运行和答案结果完全相同,但是洛谷RE。

2024/3/23 13:53
加载中...