codevs 1080セグメントツリー練習(基本テンプレート単点修正区間加算)
5299 ワード
タイトルはDescriptionの1行のN個の四角格子を説明して、初めに各格子の中ですべて1つの整数があります.質問の形式はある特定のサブ区間[a,b]のすべての要素の和を求めることである.変更されたルールは、ある格子xを指定し、特定の値Aを加算または減算することです.今、質問ごとに正しい答えを求めています.1≦N<100000,,質問と修正の総数m<10000件.
入力記述Input Description入力ファイルの最初の動作は整数Nであり、次にn行n個の整数であり、格子内の元の整数を表す.次の正の整数mに続いてm行があり、m個の問合せを表し、最初の整数は問合せ代号を表し、問合せ代号1は増加を表し、後の2つの数xとAは位置X上の数値増加Aを表し、問合せ代号2は区間和を表し、後の2つの整数はaとbを表し、要求[a,b]間の区間和を表す.
出力記述Output Description共m行、各整数
サンプル入力Sample Input 6
4
5
6
2
1
3
4
1 3 5
2 1 4
1 1 9
2 2 6
サンプル出力Sample Output 22
22
データ範囲および提示Data Size&Hint 1≦N≦100000,m≦10000.
コードは次のとおりです.
入力記述Input Description入力ファイルの最初の動作は整数Nであり、次にn行n個の整数であり、格子内の元の整数を表す.次の正の整数mに続いてm行があり、m個の問合せを表し、最初の整数は問合せ代号を表し、問合せ代号1は増加を表し、後の2つの数xとAは位置X上の数値増加Aを表し、問合せ代号2は区間和を表し、後の2つの整数はaとbを表し、要求[a,b]間の区間和を表す.
出力記述Output Description共m行、各整数
サンプル入力Sample Input 6
4
5
6
2
1
3
4
1 3 5
2 1 4
1 1 9
2 2 6
サンプル出力Sample Output 22
22
データ範囲および提示Data Size&Hint 1≦N≦100000,m≦10000.
コードは次のとおりです.
#include
#include
#include
#define M 100005
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
/*left,right,root,middle*/
int sum[M<<2];
int A[M];
inline void PushPlus(int rt)
{
sum[rt] = sum[rt<<1] + sum[rt<<1|1];
}
void Build(int l, int r, int rt) //
{
if(l == r)
{
//scanf("%d", &sum[rt]);
sum[rt]=A[l];
return ;
}
int m = ( l + r )>>1;
Build(lson);
Build(rson);
PushPlus(rt);
}
void Updata(int p, int add, int l, int r, int rt)// ,p add
{
if( l == r )
{
sum[rt] += add;
//sum[rt]=add;
return ;
}
int m = ( l + r ) >> 1;
if(p <= m)
Updata(p, add, lson);
else
Updata(p, add, rson);
PushPlus(rt);
}
int Query(int L,int R,int l,int r,int rt)
{
if( L <= l && r <= R )
{
return sum[rt];
}
int m = ( l + r ) >> 1;
int ans=0;
if(L<=m )
ans+=Query(L,R,lson);
if(R>m)
ans+=Query(L,R,rson);
return ans;
}
int main()
{
int n;
scanf("%d",&n);
for(int j=1;j<=n;j++)
scanf("%d",&A[j]);
Build(1,n,1);
int m;
scanf("%d",&m);
while(m--)
{
int f,a,b;
scanf("%d %d %d",&f,&a,&b);
if(f==2)
printf("%d
",Query(a,b,1,n,1));
else if(f==1)
Updata(a,b,1,n,1);
}
return 0;
}