본문 바로가기
IT/BOJ

백준(BOJ) 11505 구간 곱 구하기 *

by 빨강자몽 2018. 6. 1.

세그먼트 트리의 기본적인 구조이다.
하나씩 업데이트를 하는과정과
구간의 곱을 구하는 과정을 구현하면 된다.

구하는 과정에서 오버플로우가 발생하는 것을 조심한다.


#include <iostream>
#include <algorithm>
#include <vector>
#include <string.h>
#include <math.h>
#include <queue>
#include <set>
#include <utility>
#include <functional>
#define MAX 1000005
#define MOD 1000000007
#pragma warning(disable:4996)
using namespace std;

int n,m,a,b,c,d,arr[MAX*4];
long long ans;

long long update(int cur,int dest,int val,int left,int right)
{
    if(!(left<=dest&&dest<=right))
        return arr[cur];
    if(left==right)
        return arr[cur]=val;
    int mid=(left+right)/2;
    return arr[cur]=(update(cur*2,dest,val,left,mid)*update(cur*2+1,dest,val,mid+1,right))%MOD;
}
// 하나씩 업데이트를 한다.(두 arr[?]의 곱이 long long 형이 될 수 있다는것을 주의한다!)

void query(int cur,int dleft,int dright,int left,int right)
{
    if(dright<left||right<dleft)
        return ;
    if(dleft<=left&&right<=dright)
    {
        ans=(ans*arr[cur])%MOD;
        return;
    }
    int mid=(left+right)/2;
    query(cur*2,dleft,dright,left,mid);
    query(cur*2+1,dleft,dright,mid+1,right);
}
// 곱을 구한다.

int main()
{
    fill_n(&arr[0],MAX*4,1);
    scanf("%d%d%d",&n,&m,&a);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&b);
        update(1,i,b%MOD,1,n);
    }
    for(int i=0;i<m+a;i++)
    {
        ans=1;
        scanf("%d%d%d",&b,&c,&d);
        if(b==1)
            update(1,c,d,1,n);
        else
        {
            query(1,c,d,1,n);
            printf("%d\n",ans);
        }
    }
}


'IT > BOJ' 카테고리의 다른 글

백준(BOJ) 6086 최대 유량 ***  (0) 2018.06.01
백준(BOJ) 5676 음주코딩 **  (0) 2018.06.01
백준(BOJ) 2336 굉장한 학생 ***  (0) 2018.06.01
백준(BOJ) 1920 수찾기 *  (0) 2018.06.01
백준(BOJ) 10815 숫자 카드 *  (0) 2018.06.01