結果
問題 | No.1099 Range Square Sum |
ユーザー |
![]() |
提出日時 | 2020-06-26 22:59:19 |
言語 | C++11 (gcc 13.3.0) |
結果 |
AC
|
実行時間 | 918 ms / 2,000 ms |
コード長 | 3,503 bytes |
コンパイル時間 | 1,816 ms |
コンパイル使用メモリ | 163,068 KB |
実行使用メモリ | 5,376 KB |
最終ジャッジ日時 | 2024-07-04 23:15:59 |
合計ジャッジ時間 | 9,685 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 30 |
ソースコード
#include "bits/stdc++.h"using namespace std;#define int long long#define FOR(i, a, b) for(int i=(a);i<(b);i++)#define RFOR(i, a, b) for(int i=(b-1);i>=(a);i--)#define REP(i, n) for(int i=0; i<(n); i++)#define RREP(i, n) for(int i=(n-1); i>=0; i--)#define REP1(i, n) for(int i=1; i<=(n); i++)#define RREP1(i, n) for(int i=(n); i>=1; i--)#define ALL(a) (a).begin(),(a).end()#define UNIQUE_SORT(l) sort(ALL(l)); l.erase(unique(ALL(l)), l.end());#define CONTAIN(a, b) find(ALL(a), (b)) != (a).end()#define out(...) printf(__VA_ARGS__)#if DEBUG#define debug(...) printf(__VA_ARGS__)#else#define debug(...) /* ... */#endiftemplate<class T>bool chmax(T &a,const T &b){if(a<b){a=b;return true;}return false;}template<class T>bool chmin(T &a,const T &b){if(b<a){a=b;return true;}return false;}void solve();signed main(){#if DEBUGstd::ifstream in("input.txt");std::cin.rdbuf(in.rdbuf());#endifcin.tie(0);ios::sync_with_stdio(false);solve();return 0;}/*================================*/#if DEBUG#define SIZE 100#else#define SIZE 123450#endifint N,M,Q;/********************************* 平方分割*******************************/class SqrtDecomposition {public:vector<int> A; // 全体に足される分vector<int> S; // Tの合計vector<int> S2; // T^2の合計vector<int> T; // 多い方int K;SqrtDecomposition(int size) {K = sqrt(size);T = vector<int>(size, 0);S = vector<int>(size/K+1, 0);S2 = vector<int>(size/K+1, 0);A = vector<int>(size/K+1, 0);}void udpate() {REP(i,T.size()) {int lk = i/K;S2[lk] += T[i]*T[i];S[lk] += T[i];}}void add(int l, int r, int w) {while(l < r) {int lk = l / K;int rk = r / K;if (lk == rk) {FOR(j, l, r) {S2[lk] -= T[j]*T[j];S[lk] += w;T[j] += w;S2[lk] += T[j]*T[j];}}else if (l % K == 0) {A[lk] += w;}else {FOR(j, l, (lk+1)*K) {S2[lk] -= T[j]*T[j];S[lk] += w;T[j] += w;S2[lk] += T[j]*T[j];}}l = (lk+1)*K;}}int query(int l, int r) {int ret = 0;while(l < r) {int lk = l / K;int rk = r / K;if (lk == rk) {FOR(j, l, r) {int c = A[lk] + T[j];ret += c*c;}}else if (l % K == 0) {int add = K*A[lk]*A[lk] + 2*A[lk]*S[lk]+S2[lk];ret += add;}else {FOR(j, l, (lk+1)*K) {int c = A[lk] + T[j];ret += c*c;}}l = (lk+1)*K;}return ret;}};void solve() {cin>>N;int a;auto sd = SqrtDecomposition(N);REP(i,N) {cin>>a;sd.T[i]=a;}sd.udpate();cin>>Q;int c,l,r,x;while(Q--) {cin>>c>>l>>r;if (c==1) {cin>>x;sd.add(--l, r, x);} else {out("%lld\n", sd.query(--l, r));}}}