oj.uz

Submission #765933

# Submission time Handle Problem Language Result Execution time Memory
765933 2023-06-25T07:50:46 Z vjudge1 Addk (eJOI21_addk) C++14
0 / 100
 20 ms 2664 KB 
//srand(time(0)) - always changing
//order_of_key(k): Number of items strictly smaller than k .
//find_by_order(k): K-th element in a set (counting from zero).
//#pragma GCC optimize("O3,unroll-loops")
//#pragma GCC target("avx2,bmi,bmi2,lzcnt,popcnt")
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>

#define ordered_set tree<ll,null_type,less<ll>,rb_tree_tag,tree_order_statistics_node_update>
#define fopen(s) freopen(s".in","r",stdin);freopen(s".out","w",stdout)
#define all(x) (x).begin(), (x).end()
#define len(x) (int)x.size()
#define pb push_back
#define pf push_front
#define popb pop_back
#define popf pop_front
#define elif else if
#define F first
#define S second
#define int long long

typedef unsigned long long ull;
typedef long long ll;

using namespace std;
using namespace __gnu_pbds;

const int MOD = 1e9 + 7;
const int N = 2e5 + 7;
const int P = 911;
const ll INF = 1e18;

int gcd(int a, int b) {
	while (b) {
		a %= b;
		swap (a, b);
	}
	return a;
}

ll __sqrt(ll x) {
	ll result = 0;
	for (ll k = 1ll << 30; k != 0; k >>= 1) {
		if ((result + k) * (result + k) <= x) {
			result += k;
		}
	}
	return result;
}

int a[N], ps[N], ss[N], sum[N];

void solve() {
	int n, k;
	cin >> n >> k;
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
		ps[i] = ps[i - 1] + a[i] * i;
		sum[i] = sum[i - 1] + a[i];
	}
	for (int i = n; i > 0; i--) {
		ss[i] = ss[i + 1] + a[i] * (n - i + 1);
	}
	int q;
	cin >> q;
	while (q--) {
		int ty, l, r, m;
		cin >> ty >> l >> r >> m;
		if (ty == 1) continue;
		int ln = r - l + 1;
		if (ln < m * 2 - 1) {
			int act = ln - m + 1;
			int cur = l + (ln >> 1) - 1;
			int ans1 = (ps[cur] - ps[l - 1]) - (sum[cur] - sum[l - 1]) * (l - 1);
//			cout << "ans1 =  " << ps[cur] << ' ' << ps[l - 1] << ' ' << sum[cur] << ' ' << sum[l - 1] << ' ' << l - 1 << '\n';
			cur++;
			if (ln & 1) cur++;
			int ans2 = (ss[cur] - ss[r + 1]) - (sum[r] - sum[cur - 1]) * (n - r);
			cout << ans1 + ans2 + a[cur - 1] * act * (ln & 1) << '\n'; 
		} else {
			int ans1 = (ps[l + m - 2] - ps[l - 1]) - (sum[l + m - 2] - sum[l - 1]) * (l - 1);
			int ans2 = (sum[r - m + 1] - sum[l + m - 2]) * m;
			int ans3 = (ss[r - m + 2] - ss[r + 1]) - (sum[r] - sum[r - m + 1]) * (n - r);
			cout << ans1 + ans2 + ans3 << '\n';
//			cout << ans1 << ' ' << ans2 << ' ' << ans3 << '\n';
//			cout << "ans3 =  " << ss[r - m + 2] << ' ' << ss[r + 1] << ' ' << sum[r] << ' ' << sum[r - m + 1] << ' ' << n - r << '\n';
		}
	}
}

const bool Cases = 0;

signed main() {
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	int TT = 1;
	if (Cases) cin >> TT;
	while (TT--) {
		solve();
	}
}

Subtask #1
0 / 36.0

# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 340 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #2
0 / 56.0

# Verdict Execution time Memory Grader output
1 Incorrect 9 ms 1372 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #3
0 / 8.0

# Verdict Execution time Memory Grader output
1 Incorrect 20 ms 2664 KB Output isn't correct
2 Halted 0 ms 0 KB -