AcWing242。単純な整数の問題(ツリー配列+差)

AcWing242。単純な整数問題()

長さNのシーケンスAが与えられた場合、M行の操作命令を入力します。

最初のタイプの命令は「Clrd」のようなものです。これは、シーケンス内のすべてのl〜r番号にdを追加することを意味します。

2番目のタイプの命令は、「QX」のようなものです。これは、クエリシーケンスのx番目の数値の値を意味します。

クエリごとに、答えを示す整数を出力します。

入力フォーマット

最初の行には、2つの整数NとMが含まれています。

2行目には、N個の整数A [i]が含まれています。

次のM行はM命令を表し、各命令の形式はタイトルの説明に示されています。

出力フォーマット

クエリごとに、答えを示す整数を出力します。

それぞれの答えは一列に並んでいます。

データ範囲

1≤N、M≤1051≤N、M≤10_{5} 1N M1 05
∣d∣≤10000 | d |≤10000d 1 0 0 0 0
∣ A [i] ∣≤1000000000 | A [i] |≤1000000000| A [ I ] |1 0 0 0 0 0 0 0 0 0

コード

#include<iostream>
#include<cstdio>
#include<queue>
#include<string>
#include<cstring>
#include<map>
#include<vector>
#include<set>
#include<stack>
#include<algorithm>
#include<vector>
#include<utility>
#include<deque>
#include<unordered_map>
#define INF 0x3f3f3f3f
#define mod 1000000007
#define endl '\n'
#define eps 1e-6
inline int gcd(int a, int b) {
    
     return b ? gcd(b, a % b) : a; }
inline int lowbit(int x) {
    
     return x & -x; }


using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int, int> PII;
const int N = 100010;

int n, m;
int a[N], tr[N];

void add(int x, int c) {
    
    
	for (int i = x;i <= n;i += lowbit(i))tr[i] += c;
}

LL sum(int x) {
    
    
	LL res = 0;
	for (int i = x;i;i -= lowbit(i))res += tr[i];
	return res;
}
int main() {
    
    
	scanf("%d%d", &n, &m);

	for (int i = 1;i <= n;++i)scanf("%d", &a[i]);
	for (int i = 1;i <= n;++i)add(i, a[i] - a[i - 1]);
	while (m--) {
    
    
		char op[2];
		int l, r, d;
		scanf("%s%d", op, &l);

		if (op[0] == 'C') {
    
    
			scanf("%d%d", &r, &d);
			add(l, d), add(r + 1, -d);
		}
		else {
    
    
			printf("%lld\n", sum(l));
		}
	}

	return 0;
}

おすすめ

転載: blog.csdn.net/zzq0523/article/details/113100779