树状数组：扩展应用

1.A_Tiny_Problem_with_Inegers

题目描述:
给定长度为 N 的数列$A$,然后输入$M$行操作指令。
第一类指令形如$Clrd$，表示把数列中第$l\sim r$个数都加$d$。
第二类指令形如$Qx$，表示询问数列中第$x$个数的值。
对于每个询问，输出一个整数表示答案。

输入格式:
第一行包含两个整数$N$和$M$。
第二行包含$N$个整数$A[i]$。
接下来$M$行表示$M$条指令，每条指令的格式如题目描述所示。
输出格式:
对于每个询问，输出一个整数表示答案。
每个答案占一行。

CODE:

#include<iostream>
#include<cstdio>
#include<algorithm> 
#include<cmath>
#include<cstring> 
#define maxn 100010
#define ll long long 
using namespace std;
ll a[maxn],t[maxn],ans[maxn];
ll n,m,cnt=0;
inline int wr(){
	int x=0,f=1;char c=getchar();
	while(c>'9'||c<'0'){if(c=='-')f=-1;c=getchar();}
	while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}
	return f*x;
}
void Add(ll x,int y){
	for(;x<=n;x+=(x&-x))
		t[x]+=y;
}
ll ask(ll x){
	int ans=a[x];
	for(;x;x-=(x&-x))
		ans+=t[x];
	return ans;
}
int main(){
	n=wr(),m=wr();char c;
	int l,r,d,x;
	for(int i=1;i<=n;++i)	a[i]=wr();
	for(int i=1;i<=m;++i){
		cin>>c;
		if(c=='C'){
			l=wr(),r=wr(),d=wr(); 
			Add(l,d);
			Add(r+1,-d); 
		}
		else{
			x=wr();
			ans[++cnt]=ask(x);
		}
	}
	for(int i=1;i<=cnt;++i){
		printf("%d\n",ans[i]);
	}
	return 0;	
}

2.A_Simple_Problem_with_Integers

题目描述:
给定一个长度为 N 的数列A，以及M条指令，每条指令可能是以下两种之一：

1. C l r d,表示把 A[l],A[l+1],…,A[r]都加上 d。
2. Q l r，表示询问数列中第l∼r个数的和。

对于每个询问，输出一个整数表示答案。
输入格式
第一行两个整数 N,M。
第二行 N 个整数 A[i]。
接下来 M 行表示 M 条指令，每条指令的格式如题目描述所示。
输出格式
对于每个询问，输出一个整数表示答案。
每个答案占一行。

CODE:

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#define rg register
#define ll long long 
#define lowbit(x) x&-x
using namespace std;
inline ll wri(){
	ll x=0,f=1;char c=getchar();
	while(c>'9'||c<'0') {if(c=='-')f=-1;c=getchar();}
	while(c>='0'&&c<='9') {x=x*10+c-'0';c=getchar();}
	return f*x; 
}
const ll maxn=5e5+10;
ll a[maxn],n,m,ans,c[3][maxn],sum[maxn];
char ops;
void update(ll k,ll x,ll val){
	for(rg ll i=x;i<=n;i+=lowbit(i)) c[k][i]+=val;
}
ll Ask(ll k,ll x){
	ll sum=0;
	for(rg ll i=x;i;i-=lowbit(i)) sum+=c[k][i];
	return sum;
}
int main(){
	n=wri();	m=wri();
	for(rg ll i=1;i<=n;++i){
		a[i]=wri();
		sum[i]=sum[i-1]+a[i];
	}
	for(rg ll i=1;i<=m;++i){
		cin>>ops;
		if(ops=='C'){
			x=wri();	y=wri();	k=wri();
			update(1,x,k); update(1,y+1,(-1)*k);
			update(2,x,x*k);	update(2,y+1,(-1)*(y+1)*k);
		}
		else if(ops=='Q'){
			x=wri(); y=wri();
			ans=sum[y]+(y+1)*Ask(1,y)-Ask(2,y);
			ans-=sum[x-1]+x*Ask(1,x-1)-Ask(2,x-1); 
			printf("%lld\n",ans);
		}
	} 
	return 0;
}

3.Lost_Cows

题目描述：
有 n 头奶牛，已知它们的身高为 1∼n 且各不相同，但不知道每头奶牛的具体身高。现在这 n 头奶牛站成一列，已知第 i 头牛前面有 Ai 头牛比它低，求每头奶牛的身高。
输入描述：
第 1 行：输入整数 n 。
第 2…n 行：每行输入一个整数 Ai，第 i 行表示第 i 头牛前面有 Ai 头牛比它低。（注意：因为第 1 头牛前面没有牛，所以并没有将它列出）
输出描述：
输出包含 n 行，每行输出一个整数表示牛的身高。
第 i 行输出第 i 头牛的身高。
CODE：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
#define MAXN 100100
using namespace std;
int n;
int a[MAXN],t[MAXN],ans[MAXN];
void Add(int x,int y){
	for(;x<=n;x+=(x&-x))
		t[x]+=y; 
}
int Ask(int x){
	int ret=0;
	for(;x;x-=(x&-x)){ret+=t[x];}
	return ret;
}
int main(){
	scanf("%d",&n);Add(1,1); 
	for(int i=2;i<=n;++i){	scanf("%d",&a[i]);	Add(i,1);} 
	for(int i=n;i>=1;--i){
		int l=1,r=n;
		while(l<r){
			int mid=l+r>>1;
			if(Ask(mid)<a[i]+1)
				l=mid+1;
			else 
				r=mid;
		}
		ans[i]=r;
		Add(r,-1);
	}
	for(int i=1;i<=n;++i){
		printf("%d\n",ans[i]);
	}
	return 0; 
}