区间修改&单点查询

题目描述

如题，已知一个数列，你需要进行下面两种操作：

1.将某区间每一个数数加上x

2.求出某一个数的值

输入格式

第一行包含两个整数N、M，分别表示该数列数字的个数和操作的总个数。

第二行包含N个用空格分隔的整数，其中第i个数字表示数列第i项的初始值。

接下来M行每行包含2或4个整数，表示一个操作，具体如下：

操作1：格式：1 x y k 含义：将区间[x,y]内每个数加上k

操作2：格式：2 x 含义：输出第x个数的值

输出格式

输出包含若干行整数，即为所有操作2的结果。

输入输出样例

输入 #1

输出 #1

6
10

说明/提示

时空限制：1000ms,128M

数据规模：

对于30%的数据：N<=8，M<=10

对于70%的数据：N<=10000，M<=10000

对于100%的数据：N<=500000，M<=500000

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#pragma warning(disable:4996)
struct Node
{
	int l, r;
	int lazy = 0, sum = 0;
}tree[4000000];
int input[500005];
void pushup(int k)
{
	if (tree[k].l == tree[k].r)
	{
		tree[k].lazy = 0;
		return;
	}
	tree[2 * k].sum += (tree[2 * k].r - tree[2 * k].l + 1)*tree[k].lazy;
	tree[2 * k + 1].sum += (tree[2 * k + 1].r - tree[2 * k + 1].l + 1)*tree[k].lazy;
	tree[2 * k].lazy += tree[k].lazy;
	tree[2 * k + 1].lazy += tree[k].lazy;
	tree[k].lazy = 0;
}
void build(int k, int l, int r)
{
	tree[k].l = l, tree[k].r = r;
	if (l == r)
	{
		tree[k].sum = input[l]; return;
	}
	int mid = (l + r) >> 1;
	build(2 * k, l, mid);
	build(2 * k + 1, mid + 1, r);
	tree[k].sum = tree[2 * k].sum + tree[2 * k + 1].sum;
}
void add1(int k, int dis, int x)//单点修改
{
	if (tree[k].l == tree[k].r)
	{
		//cout << "!!!!!   " << tree[k].l<<" "<< tree[k].sum<<"  << endl;
		tree[k].sum += x;
		return;
	}
	int mid = (tree[k].l + tree[k].r) >> 1;
	if (mid >= dis)add1(2 * k, dis, x);
	else add1(2 * k + 1, dis, x);
	tree[k].sum = tree[2 * k].sum + tree[2 * k + 1].sum;
}
void add2(int k, int l, int r, int x)
{
	if (tree[k].l == l && tree[k].r == r)
	{
		tree[k].sum += (r - l + 1)*x;
		tree[k].lazy += x;
		return;
	}
    pushup(k);
	int mid = (tree[k].l + tree[k].r) >> 1;
	if (mid >= r)add2(2 * k, l, r, x);
	else if (mid < l)add2(2 * k + 1, l, r, x);
	else add2(2 * k, l, mid, x), add2(2 * k + 1, mid + 1, r, x);
	tree[k].sum = tree[2 * k].sum + tree[2 * k + 1].sum;
}
int query(int k, int l, int r)
{
	if (tree[k].lazy)pushup(k);
	if (tree[k].l == l && tree[k].r == r)
	{
		return tree[k].sum;
	}
	int mid = (tree[k].l + tree[k].r) >> 1;
	if (mid >= r)return query(2 * k, l, r);
	else if (mid < l)return query(2 * k + 1, l, r);
	else return query(2 * k, l, mid) + query(2 * k + 1, mid + 1, r);
}
int main()
{
	int n, m;
	cin >> n >> m;
	for (int i = 1; i <= n; i++)scanf("%d", &input[i]);
	build(1, 1, n);
	//for (int i = 1; i <= 7; i++)cout << tree[i].sum << " ";
	while (m--)
	{
		int a;
		scanf("%d", &a);
		if (a == 1)
		{
			int x, y, k;
			scanf("%d%d%d", &x, &y, &k);
			add2(1, x, y, k);
		}
		else
		{
			int x;
			scanf("%d", &x);
			cout << query(1, x, x) << endl;
		}
	}



}