Cod sursa(job #3286270)

Utilizator SleepyOverlordPatcas Csaba SleepyOverlord Data 13 martie 2025 21:47:33
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.98 kb
//Code by Patcas Csaba aka SleepyOverlord
#include <vector>
#include <array>
#include <string> 
#include <set> 
#include <map> 
#include <unordered_set>
#include <unordered_map>
#include <queue> 
#include <bitset> 
#include <stack>
#include <list>

#include <numeric> 
#include <algorithm> 
#include <random>
#include <chrono>

#include <cstdio>
#include <fstream>
#include <iostream> 
#include <sstream> 
#include <iomanip>
#include <climits>

#include <cctype>
#include <cmath> 
#include <ctime>
#include <cassert>

using namespace std;

#define ULL unsigned long long
#define LL long long
#define PII pair <int, int>
#define PLL pair <LL, LL>
#define VB vector <bool>
#define VI vector <int>
#define VLL vector <LL>
#define VD vector <double>
#define VS vector <string>
#define VPII vector <pair <int, int> >
#define VVI vector < VI >
#define VVLL vector < VLL >
#define VVB vector < VB >
#define SI set < int >
#define USI unordered_set <int>
#define MII map <int, int>
#define UMII unordered_map <int, int>

#define FORN(i, n) for(int i = 0; i < (n); ++i)
#define FOR(i, a, b) for(int i = (a); i <= (b); ++i)
#define FORD(i, a, b) for(int i = (a); i >= (b); --i)
#define MX(x, y) x = max(x, y)
#define MN(x, y) x = min(x, y)

#define SZ size()
#define BG begin() 
#define EN end() 
#define CL clear()
#define X first
#define Y second
#define RS resize
#define PB push_back
#define MP make_pair
#define ALL(x) x.begin(), x.end()
#define INS insert
#define ER erase
#define CNT count

template <class T> ostream& operator <<(ostream & os, const vector<T> &vec)
{
	for (int i = 0; i < vec.size() - 1; ++i) os << vec[i] << ' ';
	return os << vec[vec.size() - 1];
}

template <class T1, class T2> ostream& operator <<(ostream & os, const pair<T1, T2> &p)
{
	return os << p.X << " " << p.Y;
}

template <typename T>
void pr(T var1)
{
	cout << var1 << '\n';
}
template <typename T, typename... Types>
void pr(T var1, Types... var2)
{
	cout << var1;
	pr(var2...);
}

int n, q;
VI tree;

void update(int ind, int delta)
{
	while (ind <= n)
	{
		tree[ind] += delta;
		ind += ind & -ind;
	}
}

int query(int ind)
{
	int ans = 0;
	while (ind)
	{
		ans += tree[ind];
		ind -= ind & -ind;
	}
	return ans;
}

int main()
{
	freopen("aib.in", "r", stdin);
	freopen("aib.out", "w", stdout);

	ios_base::sync_with_stdio(false), cin.tie(0), cout.tie(0);

	cin >> n >> q;
	tree.RS(n + 1);
	FOR(i, 1, n) 
	{
		int x;
		cin >> x;
		update(i, x);
	}
	FOR(qq, 1, q)
	{
		int op;
		cin >> op;
		if (op == 0)
		{
			int ind, delta;
			cin >> ind >> delta;
			update(ind, delta);
		}
		else
			if (op == 1)
			{
				int l, r;
				cin >> l >> r;
				pr(query(r) - query(l - 1));
			}
			else
			{
				int s;
				cin >> s;
				int l = 0, r = n + 1;
				while (r - l > 1)
				{
					int m = (l + r) / 2;
					if (query(m) < s) l = m;
					else r = m;
				}
				pr((query(r) == s) ? r : -1);
			}
	}
	return 0;
}