#993370 (C++17) No.1099 Range Square Sum

提出ソース

結果

問題	No.1099 Range Square Sum
ユーザー	cn_449
提出日時	2024-07-09 17:14:47
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	132 ms / 2,000 ms
コード長	6,273 bytes
コンパイル時間	5,953 ms
コンパイル使用メモリ	226,240 KB
実行使用メモリ	23,960 KB
最終ジャッジ日時	2024-07-09 17:14:58
合計ジャッジ時間	9,911 ms
ジャッジサーバーID （参考情報）	judge2 / judge1

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,812 KB
testcase_01	AC	2 ms 6,816 KB
testcase_02	AC	2 ms 6,944 KB
testcase_03	AC	2 ms 6,940 KB
testcase_04	AC	2 ms 6,940 KB
testcase_05	AC	2 ms 6,944 KB
testcase_06	AC	2 ms 6,940 KB
testcase_07	AC	2 ms 6,940 KB
testcase_08	AC	2 ms 6,944 KB
testcase_09	AC	2 ms 6,944 KB
testcase_10	AC	2 ms 6,944 KB
testcase_11	AC	3 ms 6,940 KB
testcase_12	AC	3 ms 6,940 KB
testcase_13	AC	3 ms 6,940 KB
testcase_14	AC	3 ms 6,944 KB
testcase_15	AC	3 ms 6,940 KB
testcase_16	AC	2 ms 6,940 KB
testcase_17	AC	3 ms 6,940 KB
testcase_18	AC	3 ms 6,944 KB
testcase_19	AC	3 ms 6,940 KB
testcase_20	AC	2 ms 6,944 KB
testcase_21	AC	131 ms 23,916 KB
testcase_22	AC	132 ms 23,784 KB
testcase_23	AC	129 ms 23,792 KB
testcase_24	AC	129 ms 23,800 KB
testcase_25	AC	128 ms 23,920 KB
testcase_26	AC	106 ms 23,840 KB
testcase_27	AC	104 ms 23,856 KB
testcase_28	AC	103 ms 23,920 KB
testcase_29	AC	104 ms 23,960 KB
testcase_30	AC	107 ms 23,876 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <iostream>
#include <vector>
#include <algorithm>
#include <cmath>
#include <string>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <array>
#include <iomanip>
#include <utility>
#include <tuple>
#include <functional>
#include <bitset>
#include <cassert>
#include <complex>
#include <stdio.h>
#include <time.h>
#include <numeric>
#include <random>
#include <unordered_set>
#include <unordered_map>
#define all(a) (a).begin(), (a).end()
#define rep(i, n) for (ll i = 0; i < (n); i++)
#define req(i, a, b) for (ll i = (a); i < (b); i++)
#define pb push_back
#define debug(x) cerr << __LINE__ << ' ' << #x << ':' << (x) << '\n'
#define debug2(x, y) cerr << __LINE__ << ' ' << #x << ':' << (x) << ',' << #y << ':' << (y) << '\n'
#define debug3(x, y, z) cerr << __LINE__ << ' ' << #x << ':' << (x) << ',' << #y << ':' << (y) << ',' << #z << ':' << (z) << '\n'
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
using namespace std;
typedef long long ll;
typedef unsigned int uint;
typedef unsigned long long ull;
typedef long double ld;
template<class T> using P = pair<T, T>;
template<class T> using pri_s = priority_queue<T, vector<T>, greater<T>>;
template<class T> using pri_b = priority_queue<T>;
constexpr int inf = 1000000010;
constexpr int inf2 = 2000000010;
constexpr ll INF = 1000000000000000010;
constexpr ll INF4 = 4000000000000000010;
constexpr int mod1e9 = 1000000007;
constexpr int mod998 = 998244353;
constexpr ld eps = 1e-12;
constexpr ld pi = 3.141592653589793238;
constexpr ll ten(int n) { return n ? 10 * ten(n - 1) : 1; };
int dx[] = { 1,0,-1,0,1,1,-1,-1,0 }; int dy[] = { 0,1,0,-1,1,-1,1,-1,0 };
ll mul(ll a, ll b) { return (b != 0 && a > INF / b ? INF : a * b); }
void fail() { cout << "-1\n"; exit(0); } void no() { cout << "No\n"; exit(0); }
template<class T> void er(T a) { cout << a << '\n'; exit(0); }
template<class T, class U> inline bool chmax(T& a, const U& b) { if (a < b) { a = b; return true; } return false; }
template<class T, class U> inline bool chmin(T& a, const U& b) { if (a > b) { a = b; return true; } return false; }
template<class T> istream& operator >>(istream& s, vector<T>& v) { for (auto& e : v) s >> e; return s; }
template<class T> ostream& operator <<(ostream& s, const vector<T>& v) { for (auto& e : v) s << e << ' '; return s; }
template<class T, class U> ostream& operator << (ostream& s, const pair<T, U>& p) { s << p.first << ' ' << p.second; return s; }

struct fastio {
	fastio() {
		cin.tie(0); cout.tie(0);
		ios::sync_with_stdio(false);
		cout << fixed << setprecision(20);
		cerr << fixed << setprecision(20);
	}
}fastio_;

namespace rdv {
	random_device seed_gen;
	mt19937_64 engine(seed_gen());
	ll rnum(ll r) { return engine() % r; } // [0, r)
	ll rnum(ll l, ll r) { return rnum(r - l) + l; } // [l, r)
	// ll rng(ll r) { return rnum(r) + 1; } // [1, r]
	ll rng(ll l, ll r) { return rnum(l, r + 1); } // [l, r]
	double rng01() { return engine() * pow(2, -64); }
	template<class T> void shuf(vector<T>& v) { shuffle(all(v), engine); }
	void shuf(string& s) { shuffle(all(s), engine); }
	vector<vector<ll>> avec(ll n, ll l, ll r) {
		if (n == 0) return { {} };
		vector<vector<ll>> pre = avec(n - 1, l, r);
		vector<vector<ll>> res;
		for (vector<ll> v : pre) {
			for (int i = l; i < r; i++) {
				vector<ll> w = v;
				w.pb(i);
				res.pb(w);
			}
		}
		return res;
	}
	vector<vector<ll>> avec(ll n, ll r) { return avec(n, 0, r); }
	vector<vector<ll>> bvec(ll n, ll l, ll r) {
		if (n == 0) return { {} };
		vector<vector<ll>> pre = bvec(n - 1, l, r);
		vector<vector<ll>> res;
		for (vector<ll> v : pre) {
			int b = l;
			if (!v.empty()) b = v.back();
			for (int i = b; i < r; i++) {
				vector<ll> w = v;
				w.pb(i);
				res.pb(w);
			}
		}
		return res;
	}
	vector<vector<ll>> bvec(ll n, ll r) { return bvec(n, 0, r); }
	vector<vector<ll>> cvec(ll n, ll sum) {
		if (n == 0) return { {} };
		vector<vector<ll>> pre = cvec(n - 1, sum);
		vector<vector<ll>> res;
		for (vector<ll> v : pre) {
			ll lim = sum;
			for (ll w : v) lim -= w;
			for (int i = 0; i <= lim; i++) {
				vector<ll> w = v;
				w.pb(i);
				res.pb(w);
			}
		}
		return res;
	}
}

using namespace rdv;

template<class T> vector<int> compress(vector<T> v) {
	int n = v.size();
	vector<T> tmp = v;
	sort(tmp.begin(), tmp.end());
	tmp.erase(unique(tmp.begin(), tmp.end()), tmp.end());
	vector<int> res(n);
	for (int i = 0; i < n; i++) res[i] = lower_bound(tmp.begin(), tmp.end(), v[i]) - tmp.begin();
	return res;
}

#include <atcoder/all>
using namespace atcoder;

constexpr ll mod = mod1e9;
using mint = static_modint<mod>;

istream& operator >>(istream& s, mint& m) { ll y; s >> y; m = y; return s; }
ostream& operator <<(ostream& s, mint& m) { return s << m.val(); }
ostream& operator <<(ostream& s, const vector<mint>& v) { for (auto& e : v) s << e.val() << ' '; return s; }

vector<mint> fac, inv, facinv;

void modcalc(int n) {
	fac.resize(n); inv.resize(n); facinv.resize(n);
	fac[0] = 1; fac[1] = 1; inv[1] = 1;
	facinv[0] = 1; facinv[1] = 1;
	for (ll i = 2; i < n; i++) {
		fac[i] = fac[i - 1] * i;
		inv[i] = -inv[mod % i] * (mod / i);
		facinv[i] = facinv[i - 1] * inv[i];
	}
}

mint comb(ll n, ll k) {
	if (n < 0 || k < 0 || n < k) return 0;
	return fac[n] * facinv[k] * facinv[n - k];
}

mint perm(ll n, ll k) {
	if (n < 0 || k < 0 || n < k) return 0;
	return fac[n] * facinv[n - k];
}

mint hom(ll n, ll k) {
	if (n < 0 || k < 0 || n == 0 && k > 0) return 0;
	if (n == 0 && k == 0) return 1;
	return fac[n + k - 1] * facinv[k] * facinv[n - 1];
}

struct S {
	ll v0, v1, v2;
};
S op(S a, S b) { return { a.v0 + b.v0, a.v1 + b.v1, a.v2 + b.v2 }; }
S e() { return { 0, 0, 0 }; }
using F = ll;
S mapping(F a, S b) { return { b.v0, b.v1 + b.v0 * a, b.v2 + 2 * b.v1 * a + b.v0 * a * a }; }
F composition(F a, F b) { return a + b; }
F id() { return 0; }

int main() {
	ll n;
	cin >> n;
	vector<ll> a(n);
	cin >> a;
	vector<S> init(n);
	rep(i, n) init[i] = { 1, a[i], a[i] * a[i] };
	lazy_segtree<S, op, e, F, mapping, composition, id> seg(init);
	int q;
	cin >> q;
	while (q--) {
		int t;
		cin >> t;
		if (t == 1) {
			ll l, r, x;
			cin >> l >> r >> x;
			l--;
			seg.apply(l, r, x);
		}
		else {
			ll l, r;
			cin >> l >> r;
			l--;
			cout << seg.prod(l, r).v2 << '\n';
		}
	}
}

yukicoder

結果

テストケース

ソースコード