Cod sursa(job #371720)

Utilizator Bogdan_tmmTirca Bogdan Bogdan_tmm Data 6 decembrie 2009 13:26:04
Problema Arbori indexati binar Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.95 kb
#include<stdio.h>
#include<vector>
using namespace std;
int n,m;
vector <int> aib;
void Update(int ind,int val)
{
	for(;ind<=n;ind+=(ind&(ind-1)^ind))
		aib[ind]+=val;
}
int GetSum(int ind)
{
	int sum=0;
	for(;0<ind;ind-=(ind&(ind-1)^ind))
		sum+=aib[ind];
	return sum;
}
int PozMin(int sum)
{
	int i,step;
	for(step=1;step<n;step<<=1);
	for(i=0;step;step>>=1)
		if(i+step<=n&&aib[i+step]<=sum)
		{
			i+=step;
			sum-=aib[i];
			if(!sum)
				return i;
		}
	return -1;
}
int main()
{
	freopen("aib.in","r",stdin);
	freopen("aib.out","w",stdout);
	scanf("%d%d",&n,&m);
	aib.resize(n+1,0);
	for(int i=1;i<=n;i++)
	{
		int x;
		scanf("%d",&x);
		Update(i,x);
	}
	for(;m;m--)
	{
		int x,y,z;
		scanf("%d",&x);
		if(x<2)
		{
			scanf("%d%d",&y,&z);
			if(x==0)
			{
				Update(y,z);
			}
			else
				printf("%d\n",GetSum(z)-GetSum(y-1));
			continue;
		}
		scanf("%d",&y);
		printf("%d\n",PozMin(y));
	}
	return 0;
}