AcWing 242. 간단한 정수 문제 차이 + 트리 배열

AcWing 242. 간단한 정수 문제

길이 N의 숫자 시퀀스 A가 주어지면 M 행의 조작 명령을 입력하십시오.

첫 번째 유형의 명령어는 "C lrd"형식으로, 시퀀스의 모든 l ~ r 번호에 d를 더하는 것을 의미합니다.

두 번째 유형의 명령어는 "QX"형식으로 쿼리 시퀀스에서 x 번째 숫자의 값을 의미합니다.

각 쿼리에 대해 답을 나타내는 정수를 출력합니다.

입력 형식 첫
번째 줄에는 두 개의 정수 N과 M이 포함됩니다.

두 번째 줄에는 N 개의 정수 A [i]가 포함됩니다.

다음 M 행은 M 명령어를 나타내며 각 명령어의 형식은 제목 설명에 표시됩니다.

출력 형식
각 쿼리에 대해 답을 나타내는 정수가 출력됩니다.

각 답변은 한 줄을 차지합니다.

데이터 범위
1≤N, M≤105,
| d | ≤10000,
| A [i] | ≤1000000000
입력 예 :

10 5
1 2 3 4 5 6 7 8 9 10
Q 4
Q 1
Q 2
C 1 6 3
Q 2

샘플 출력 :

4
1
2
5

이 질문은 제목에서 언급했듯이 간단한 정수 문제입니다. 숫자가 무엇인지 물어 보겠습니다.이 문제를 해결하기 위해 차이와 트리 배열을 사용합니다. 추가 및 쿼리 작업이 빠르기 때문에 트리 배열을 사용하는 이유.

코드 쇼 :

#include<iostream>

using namespace std;

typedef long long LL;

const int N=1e5+10;

LL trie[N],a[N];
int n,m;

int lowbit(int x)
{
    
    
    return x&(-x);
}

void add(int k,int v)
{
    
    
    for(int i=k;i<=n;i+=lowbit(i)) trie[i]+=v;
}

LL query(int k)
{
    
    
    LL res=0;
    for(int i=k;i;i-=lowbit(i))
    res+=trie[i];
    return res;
}

int main(void)
{
    
    
    cin>>n>>m;
    for(int i=1;i<=n;i++) 
    {
    
    
        cin>>a[i];
        add(i,a[i]-a[i-1]);
    }
    while(m--)
    {
    
    
        char op;
        cin>>op;
        if(op=='Q')
        {
    
    
            int t;
            cin>>t;
            cout<<query(t)<<endl;
        }
        else
        {
    
    
            int l,r,v;
            cin>>l>>r>>v;
            add(l,v);add(r+1,-v);
        }
    }
}