結果

問題 No.1099 Range Square Sum
ユーザー k82bk82b
提出日時 2024-03-27 19:49:28
言語 D
(dmd 2.106.1)
結果
AC  
実行時間 288 ms / 2,000 ms
コード長 2,427 bytes
コンパイル時間 2,770 ms
コンパイル使用メモリ 209,956 KB
実行使用メモリ 19,984 KB
最終ジャッジ日時 2024-09-30 14:34:58
合計ジャッジ時間 6,957 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,820 KB
testcase_01 AC 1 ms
6,816 KB
testcase_02 AC 1 ms
6,816 KB
testcase_03 AC 1 ms
6,820 KB
testcase_04 AC 1 ms
6,820 KB
testcase_05 AC 1 ms
6,820 KB
testcase_06 AC 1 ms
6,820 KB
testcase_07 AC 1 ms
6,816 KB
testcase_08 AC 1 ms
6,816 KB
testcase_09 AC 1 ms
6,816 KB
testcase_10 AC 1 ms
6,816 KB
testcase_11 AC 3 ms
6,816 KB
testcase_12 AC 3 ms
6,816 KB
testcase_13 AC 3 ms
6,820 KB
testcase_14 AC 3 ms
6,820 KB
testcase_15 AC 2 ms
6,816 KB
testcase_16 AC 3 ms
6,816 KB
testcase_17 AC 3 ms
6,816 KB
testcase_18 AC 3 ms
6,816 KB
testcase_19 AC 2 ms
6,820 KB
testcase_20 AC 3 ms
6,816 KB
testcase_21 AC 288 ms
19,268 KB
testcase_22 AC 256 ms
19,816 KB
testcase_23 AC 264 ms
19,004 KB
testcase_24 AC 254 ms
19,984 KB
testcase_25 AC 254 ms
18,560 KB
testcase_26 AC 238 ms
15,852 KB
testcase_27 AC 245 ms
14,216 KB
testcase_28 AC 266 ms
14,612 KB
testcase_29 AC 244 ms
14,384 KB
testcase_30 AC 242 ms
15,240 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std, core.bitop;
bool chmin(T)(ref T SE, T MI) { if (SE > MI) { SE = MI; return true; } else { return false; } }
bool chmax(T)(ref T SE, T MI) { if (SE < MI) { SE = MI; return true; } else { return false; } }
int lowerBound(T)(T[] SE, T MI) { int lo = -1, hi = cast(int)(SE.length); while (hi - lo > 1) { int mid = lo + hi >> 1; (SE[mid] < MI ? lo : hi) = mid; } return hi; }
int upperBound(T)(T[] SE, T MI) { int lo = -1, hi = cast(int)(SE.length); while (hi - lo > 1) { int mid = lo + hi >> 1; (SE[mid] > MI ? hi : lo) = mid; } return hi; }
struct SegmentTree {
	int N;
	long[] dat0, dat1;
	int[] laz;
	this(int[] A) {
		int n = cast(int)(A.length);
		for (N = 1; N < n; N <<= 1) {}
		dat0 = new long[N << 1];
		dat1 = new long[N << 1];
		laz = new int[N << 1];
		foreach (i; 0 .. n) {
			dat0[i + N] = A[i];
			dat1[i + N] = 1L * A[i] * A[i];
		}
		foreach_reverse (i; 0 .. N) {
			dat0[i] = dat0[i << 1] + dat0[i << 1 | 1];
			dat1[i] = dat1[i << 1] + dat1[i << 1 | 1];
		}
	}
	void propagate(int i, int w) {
		if (i < N) {
			laz[i << 1] += laz[i];
			laz[i << 1 | 1] += laz[i];
		}
		dat1[i] += 2L * laz[i] * dat0[i] + 1L * laz[i] * laz[i] * w;
		dat0[i] += 1L * laz[i] * w;
		laz[i] = 0;
	}
	void rangeAdd(int L, int R, int x, int i, int l, int r) {
		propagate(i, r - l);
		if (r <= L || R <= l) {
			return;
		}
		if (L <= l && r <= R) {
			laz[i] += x;
			propagate(i, r - l);
		} else {
			int m = l + r >> 1;
			rangeAdd(L, R, x, i << 1, l, m);
			rangeAdd(L, R, x, i << 1 | 1, m, r);
			dat0[i] = dat0[i << 1] + dat0[i << 1 | 1];
			dat1[i] = dat1[i << 1] + dat1[i << 1 | 1];
		}
	}
	void rangeAdd(int L, int R, int x) {
		rangeAdd(L, R, x, 1, 0, N);
	}
	long rangeSquareSum(int L, int R, int i, int l, int r) {
		propagate(i, r - l);
		if (r <= L || R <= l) {
			return 0;
		}
		if (L <= l && r <= R) {
			return dat1[i];
		}
		int m = l + r >> 1;
		return rangeSquareSum(L, R, i << 1, l, m) + rangeSquareSum(L, R, i << 1 | 1, m, r);
	}
	long rangeSquareSum(int L, int R) {
		return rangeSquareSum(L, R, 1, 0, N);
	}
};
void main() {
	int N;
	readf("%s\n", N);
	auto A = readln.chomp.split.to!(int[]);
	int Q;
	readf("%s\n", Q);
	auto ST = new SegmentTree(A);
	while (Q--) {
		int op;
		readf("%s ", op);
		if (op == 1) {
			int l, r, x;
			readf("%s %s %s\n", l, r, x);
			ST.rangeAdd(--l, r, x);
		}
		if (op == 2) {
			int l, r;
			readf("%s %s\n", l, r);
			ST.rangeSquareSum(--l, r).writeln;
		}
	}
}
0