세그먼트 트리의 기본적인 구조이다.
하나씩 업데이트를 하는과정과
구간의 곱을 구하는 과정을 구현하면 된다.
구하는 과정에서 오버플로우가 발생하는 것을 조심한다.
#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 |