結果

問題 No.3078 Difference Sum Query
ユーザー MMRZ
提出日時 2025-05-04 08:51:11
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 612 ms / 2,000 ms
コード長 4,320 bytes
コンパイル時間 3,934 ms
コンパイル使用メモリ 286,828 KB
実行使用メモリ 49,476 KB
最終ジャッジ日時 2025-05-04 08:51:32
合計ジャッジ時間 19,697 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

# include <bits/stdc++.h>
using namespace std;
using ll = long long;
using ull = unsigned long long;
const double pi = acos(-1);
template<class T>constexpr T inf() { return ::std::numeric_limits<T>::max(); }
template<class T>constexpr T hinf() { return inf<T>() / 2; }
template <typename T_char>T_char TL(T_char cX) { return tolower(cX); }
template <typename T_char>T_char TU(T_char cX) { return toupper(cX); }
template<class T> bool chmin(T& a,T b) { if(a > b){a = b; return true;} return false; }
template<class T> bool chmax(T& a,T b) { if(a < b){a = b; return true;} return false; }
int popcnt(unsigned long long n) { int cnt = 0; for (int i = 0; i < 64; i++)if ((n >> i) & 1)cnt++; return cnt; }
int d_sum(ll n) { int ret = 0; while (n > 0) { ret += n % 10; n /= 10; }return ret; }
int d_cnt(ll n) { int ret = 0; while (n > 0) { ret++; n /= 10; }return ret; }
ll gcd(ll a, ll b) { if (b == 0)return a; return gcd(b, a%b); };
ll lcm(ll a, ll b) { ll g = gcd(a, b); return a / g*b; };
ll MOD(ll x, ll m){return (x%m+m)%m; }
ll FLOOR(ll x, ll m) {ll r = (x%m+m)%m; return (x-r)/m; }
template<class T> using dijk = priority_queue<T, vector<T>, greater<T>>;
# define all(qpqpq)           (qpqpq).begin(),(qpqpq).end()
# define UNIQUE(wpwpw)        (wpwpw).erase(unique(all((wpwpw))),(wpwpw).end())
# define LOWER(epepe)         transform(all((epepe)),(epepe).begin(),TL<char>)
# define UPPER(rprpr)         transform(all((rprpr)),(rprpr).begin(),TU<char>)
# define rep(i,upupu)         for(ll i = 0, i##_len = (upupu);(i) < (i##_len);(i)++)
# define reps(i,opopo)        for(ll i = 1, i##_len = (opopo);(i) <= (i##_len);(i)++)
# define len(x)                ((ll)(x).size())
# define bit(n)               (1LL << (n))
# define pb push_back
# define eb emplace_back
# define exists(c, e)         ((c).find(e) != (c).end())

struct INIT{
	INIT(){
		std::ios::sync_with_stdio(false);
		std::cin.tie(0);
		cout << fixed << setprecision(20);
	}
}INIT;

namespace mmrz {
	void solve();
}

int main(){
	mmrz::solve();
}
#define debug(...) (static_cast<void>(0))

using namespace mmrz;


template<typename T>
struct merge_sort_tree {
	int n;
	vector<vector<T>> x, cs;
	merge_sort_tree(const vector<T> &v) : n(ssize(v)) {
		x.resize(n*2);
		cs.resize(n*2);
		for(int i = 0;i < n;i++)x[n+i] = {v[i]}, cs[n+i] = {0, v[i]};
		for(int i = n-1;i > 0;i--){
			merge(x[i*2].begin(), x[i*2].end(), x[i*2 + 1].begin(), x[i*2 + 1].end(), back_inserter(x[i]));
			cs[i].resize(ssize(x[i]) + 1);
			for(int j = 0;j < ssize(x[i]);j++){
				cs[i][j+1] += cs[i][j] + x[i][j];
			}
		}
	}
	int count_less(int l, int r, T q) {
		l += n, r += n;
		int ret = 0;
		while(l < r) {
			if(l & 1){
				ret += lower_bound(x[l].begin(), x[l].end(), q) - x[l].begin();
				l++;
			}
			if(r & 1){
				r--;
				ret += lower_bound(x[r].begin(), x[r].end(), q) - x[r].begin();
			}
			l >>= 1, r >>= 1;
		}
		return ret;
	}

	int count_leq(int l, int r, T q) {
		l += n, r += n;
		int ret = 0;
		while(l < r) {
			if(l & 1){
				ret += x[l].end() - lower_bound(x[l].begin(), x[l].end(), q);
				l++;
			}
			if(r & 1){
				r--;
				ret += x[r].end() - lower_bound(x[r].begin(), x[r].end(), q);
			}
			l >>= 1, r >>= 1;
		}
		return ret;
	}

	T sum_less(int l, int r, T q) {
		l += n, r += n;
		T ret = 0;
		while(l < r) {
			if(l & 1){
				ret += cs[l][lower_bound(x[l].begin(), x[l].end(), q) - x[l].begin()];
				l++;
			}
			if(r & 1){
				r--;
				ret += cs[r][lower_bound(x[r].begin(), x[r].end(), q) - x[r].begin()];
			}
			l >>= 1, r >>= 1;
		}
		return ret;
	}

	T sum_leq(int l, int r, T q) {
		l += n, r += n;
		T ret = 0;
		while(l < r) {
			if(l & 1){
				ret += cs[l].back() - cs[l][lower_bound(x[l].begin(), x[l].end(), q) - x[l].begin()];
				l++;
			}
			if(r & 1){
				r--;
				ret += cs[r].back() - cs[r][lower_bound(x[r].begin(), x[r].end(), q) - x[r].begin()];
			}
			l >>= 1, r >>= 1;
		}
		return ret;
	}
};

void SOLVE(){
	int n, q;
	cin >> n >> q;
	vector<ll> a(n);
	for(auto &x : a)cin >> x;
	merge_sort_tree<ll> mst(a);
	while(q--){
		int l, r;
		ll x;
		cin >> l >> r >> x;
		l--;
		ll cl = mst.count_less(l, r, x);
		ll sl = mst.sum_less(l, r, x);
		ll cr = mst.count_leq(l, r, x);
		ll sr = mst.sum_leq(l, r, x);
		cout << (sr-cr*x) + (cl*x-sl) << '\n';
	}
}

void mmrz::solve(){
	int t = 1;
	//cin >> t;
	while(t--)SOLVE();
}
0