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