結果

問題 No.1099 Range Square Sum
コンテスト
ユーザー k82b
提出日時 2024-03-27 19:48:27
言語 D
(dmd 2.109.1)
結果
WA  
実行時間 -
コード長 2,428 bytes
コンパイル時間 2,546 ms
コンパイル使用メモリ 210,048 KB
実行使用メモリ 14,980 KB
最終ジャッジ日時 2024-09-30 14:34:50
合計ジャッジ時間 6,063 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 25 WA * 5
権限があれば一括ダウンロードができます

ソースコード

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;
	int[] dat0;
	long[] dat1;
	int[] laz;
	this(int[] A) {
		int n = cast(int)(A.length);
		for (N = 1; N < n; N <<= 1) {}
		dat0 = new int[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] += 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