luoguP2781 传教

https://www.luogu.org/problemnew/show/P2781

简化版题意:有 n 个数,初始值为 0,进行 m 次操作,每次操作支持将 [l, r] 加 v 和查询 [l, r] 中所有的数的和

n <= 1e9,m <= 1e3

博主 zz 的打了一个支持分裂节点的 splay,AC 后发现可以 m 方暴力过

方法和方伯伯的OJ这题类似,可以参考它的做法

#include <bits/stdc++.h>
using namespace std;

typedef unsigned long long ull;
typedef long long ll;

template <typename T>
inline void read(T &f) {
	f = 0; T fu = 1; char c = getchar();
	while(c < '0' || c > '9') {if(c == '-') fu = -1; c = getchar();}
	while(c >= '0' && c <= '9') {f = (f << 3) + (f << 1) + (c & 15); c = getchar();}
	f *= fu;
}

struct Node {
	ll val, tag, sum;
	int l, r, size;
	Node *ch[2];
	Node () {
		val = tag = l = r = size = 0;
		ch[0] = ch[1] = NULL;
	}
}*root;

int n, m;

void update(Node *u) {
	u -> size = u -> r - u -> l + 1;
	u -> val = u -> sum;
	if(u -> ch[0]) u -> size += u -> ch[0] -> size, u -> val += u -> ch[0] -> val;
	if(u -> ch[1]) u -> size += u -> ch[1] -> size, u -> val += u -> ch[1] -> val;
}

void pushdown(Node *u) {
	if(u -> tag) {
		if(u -> ch[0]) {
			u -> ch[0] -> tag += u -> tag;
			u -> ch[0] -> sum += (ll)(u -> ch[0] -> r - u -> ch[0] -> l + 1) * u -> tag;
			u -> ch[0] -> val += (ll)u -> ch[0] -> size * u -> tag; 
		}
		if(u -> ch[1]) {
			u -> ch[1] -> tag += u -> tag;
			u -> ch[1] -> sum += (ll)(u -> ch[1] -> r - u -> ch[1] -> l + 1) * u -> tag; 
			u -> ch[1] -> val += (ll)u -> ch[1] -> size * u -> tag;
		}
		u -> tag = 0;
	}
}

void rotate(Node *&u, int d) {
	Node *tmp = u -> ch[d];
	u -> ch[d] = tmp -> ch[d ^ 1];
	tmp -> ch[d ^ 1] = u;
	update(u); update(tmp);
	u = tmp;
}

void splay(Node *&u, int k) {
	if(u == NULL) return;
	pushdown(u);
	int ltree = u -> ch[0] ? u -> ch[0] -> size : 0;
	if(k > ltree && ltree + (u -> r - u -> l + 1) >= k) return;
	int d = k > ltree; 
	splay(u -> ch[d], d ? k - ltree - (u -> r - u -> l + 1) : k);
	rotate(u, d);
}

void split(Node *&u, int x) {
	splay(u, x); ll sum = u -> sum; int l = u -> l, r = u -> r;
	if(u -> l != x) {
		Node *tmp = new Node();
		tmp -> sum = sum / (ll)(r - l + 1) * (x - l);
		tmp -> l = l, tmp -> r = x - 1;
		tmp -> ch[0] = u -> ch[0]; update(tmp);
		u -> ch[0] = tmp; u -> l = x;
	}
	if(u -> r != x) {
		Node *tmp = new Node();
		tmp -> sum = sum / (ll)(r - l + 1) * (ll)(r - x);
		tmp -> l = x + 1, tmp -> r = r;
		tmp -> ch[1] = u -> ch[1]; update(tmp);
		u -> ch[1] = tmp; u -> r = x;
	}
	u -> sum /= (ll)(r - l + 1); update(u);
}

int main() {
	cin >> n >> m;
	root = new Node();
	root -> val = root -> tag = root -> sum = 0;
	root -> l = 1, root -> r = n; root -> size = n;
	root -> ch[0] = root -> ch[1] = NULL;
	for(int i = 1; i <= m; i++) {
//		printf("root -> size = %d\n", root -> size);
		int t; read(t);
		if(t == 1) {
			int a, b; ll c;
			read(a); read(b); read(c);
			if(a == 1 && b == n) {
				root -> val += (ll)n * c;
				root -> tag += c;
				root -> sum += (ll)(root -> r - root -> l + 1) * c;
			} else if(a == 1) {
				split(root, b + 1);
				root -> ch[0] -> val += (ll)root -> ch[0] -> size * c;
				root -> ch[0] -> tag += c;
				root -> ch[0] -> sum += (ll)(root -> ch[0] -> r - root -> ch[0] -> l + 1) * c;
				update(root);
			} else if(b == n) {
				split(root, a - 1);
				root -> ch[1] -> val += (ll)root -> ch[1] -> size * c;
				root -> ch[1] -> tag += c;
				root -> ch[1] -> sum += (ll)(root -> ch[1] -> r - root -> ch[1] -> l + 1) * c;
				update(root);
			} else {
				split(root, b + 1);
				split(root -> ch[0], a - 1);
				root -> ch[0] -> ch[1] -> val += (ll)root -> ch[0] -> ch[1] -> size * c;
				root -> ch[0] -> ch[1] -> tag += c;
				root -> ch[0] -> ch[1] -> sum += (ll)(root -> ch[0] -> ch[1] -> r - root -> ch[0] -> ch[1] -> l + 1) * c;
				update(root -> ch[0]); update(root);
			}
		}
		if(t == 2) {
			int a, b; read(a); read(b);
			if(a == 1 && b == n) {
				printf("%lld\n", root -> val);
			} else if(a == 1) {
				split(root, b + 1);
				printf("%lld\n", root -> ch[0] -> val);
			} else if(b == n) {
				split(root, a - 1);
				printf("%lld\n", root -> ch[1] -> val);
			} else {
				split(root, b + 1);
				split(root -> ch[0], a - 1);
				printf("%lld\n", root -> ch[0] -> ch[1] -> val);
			}
		} 
	}
	return 0;
}
posted @ 2018-09-18 20:42  LJC00118  阅读(168)  评论(0编辑  收藏  举报
/*
*/