243. 一个简单的整数问题2(分块,@模板题)

活动 - AcWing

给定一个长度为 N 的数列 A,以及 M 条指令,每条指令可能是以下两种之一:

  1. C l r d,表示把 A[l],A[l+1],…,A[r] 都加上 d。
  2. Q l r,表示询问数列中第 l∼r 个数的和。

对于每个询问,输出一个整数表示答案。

输入格式

第一行两个整数 N,M。

第二行 N 个整数 A[i]。

接下来 M 行表示 M 条指令,每条指令的格式如题目描述所示。

输出格式

对于每个询问,输出一个整数表示答案。

每个答案占一行。

数据范围

1≤N,M≤105
|d|≤10000
|A[i]|≤109

输入样例:
10 5
1 2 3 4 5 6 7 8 9 10
Q 4 4
Q 1 10
Q 2 4
C 3 6 3
Q 2 4
输出样例:
4
55
9
15

解析: 

#include <iostream>
#include <string>
#include <cstring>
#include <cmath>
#include <ctime>
#include <algorithm>
#include <utility>
#include <stack>
#include <queue>
#include <vector>
#include <set>
#include <math.h>
#include <map>
#include <sstream>
#include <deque>
#include <unordered_map>
#include <unordered_set>
#include <bitset>
#include <stdio.h>
#include <tuple>
using namespace std;
/*
ios::sync_with_stdio(false);
cin.tie(nullptr);
cout.tie(nullptr);
*/
typedef long long LL;
// #define int long long
#define ld long double
//#define INT __int128
const LL INF = 0x3f3f3f3f3f3f3f3f;
typedef unsigned long long ULL;
typedef pair<long long, long long> PLL;
typedef pair<int, int> PII;
typedef pair<double, double> PDD;
const int inf = 0x3f3f3f3f;
const LL mod = 998244353;
const ld eps = 1e-12;
const int N = 1e5 + 10, M = 350 + 10;
int n, m;
int A[N];
LL sum[M], add[M];
int len;
int get(int i) {
	return i / len;
}
void modify(int l, int r, int d) {
	if (get(l) == get(r)) {
		for (int i = l; i <= r; i++) {
			A[i] += d;
			sum[get(i)] += d;
		}
	}
	else {
		int i = l, j = r;
		while (get(i) == get(l))A[i] += d, sum[get(i)] += d, i++;
		while (get(j) == get(r))A[j] += d, sum[get(j)] += d, j--;
		for (int k = get(i); k <= get(j); k++) {
			sum[k] += d * len;
			add[k] += d;
		}
	}
}
LL ask(int l, int r) {
	LL ret = 0;
	if (get(l) == get(r)) {
		for (int i = l; i <= r; i++) {
			ret += A[i] + add[get(i)];
		}
	}
	else {
		int i = l, j = r;
		while (get(i) == get(l))ret += A[i] + add[get(i)], i++;
		while (get(j) == get(r))ret += A[j] + add[get(j)], j--;
		for (int k = get(i); k <= get(j); k++) {
			ret += sum[k];
		}
	}
	return ret;
}
signed main() {
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	cout.tie(nullptr);
	cin >> n >> m;
	//scanf("%d%d", &n, &m);
	len = sqrt(n);
	for (int i = 1; i <= n; i++) {
		cin >> A[i];
		sum[get(i)] += A[i];
	}
	char op[2];
	int l, r, d;
	while (m--) {
		cin >> op;
		//scanf("%s", op);
		if (op[0] == 'C') {
			cin >> l >> r >> d;
			//scanf("%d%d%d", &l, &r, &d);
			modify(l, r, d);
		}
		else {
			cin >> l >> r;
			//scanf("%d%d", &l, &r);
			LL ret = ask(l, r);
			//printf("%lld\n", ret);
			cout << ret << endl;
		}
	}
	return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值