티스토리 뷰
N개의 값이 1번 index부터 차례대로 주어진다. 그 후에 명령을 입력받는데 "1 a b"의 형태라면 a를 b라는 수로 바꿔야하고, "2 a b"의 형태라면 a번 index부터 b번 index까지의 곱을 출력해야 한다.
값을 update하는 작업이 있으므로 단순히 구간의 곱을 저장하는 것으로는 시간내에 해결 할 수 없다.
따라서 값을 update하고 구하는 과정을 O(logN)에 수행할 수 있는 세그먼트 트리 자료구조를 이용했다.
일반적으론 구간의 합을 세그먼트 트리 구조로 저장하지만, 이번 문제는 구간의 곱을 저장해주면 된다. 따라서 값에 영향을 주지 않도록 비어있는 노드는 1로 초기화 해주어야하며, 값을 update할때는 botoom-up방식으로 구현 하는 것이 깔끔하다.
/*
21.01.16
BOJ : 11505 구간 곱 구하기 (https://www.acmicpc.net/problem/11505)
자료구조/세그먼트 트리
*/
#include <bits/stdc++.h>
using namespace std;
#define ll long long int
#define MAX 1000000007
ll* tree;
ll* nums;
int S;
int N, M,K;
void init() {
for (int i = S; i < 2 * S; i++) {
if (i - S + 1 <= N) {
tree[i] = nums[i-S+1];
}
else {
tree[i] = 1;
}
}
for (int i = S - 1; i >= 1; i--) {
tree[i] = (tree[i * 2] * tree[i * 2 + 1]) % MAX;
}
}
ll query(int node, int l, int r, int ql, int qr) {
if (l > qr || r < ql) return 1;
else if (l >= ql && r <= qr) return tree[node];
else {
int mid = (l + r) / 2;
return (query(node * 2, l, mid, ql, qr) * query(node * 2 + 1, mid + 1, r, ql, qr)) % MAX;
}
}
void update(int node,int newValue) {
nums[node] = newValue;
node = S + node - 1;
tree[node] = newValue;
while (node >= 1) {
node /= 2;
tree[node] = (tree[node * 2] * tree[node * 2 + 1]) % MAX;
}
}
int main() {
cin.tie(0);
ios::sync_with_stdio(0);
cin >> N >> M >> K;
nums = new ll[N + 1];
for (int i = 1; i <= N; i++)
{
cin >> nums[i];
}
S = ceil(log2(N));
S = (1 << S);
tree = new ll[2 * S] ;
init();
for (int i = 0; i < M + K; i++) {
int a, b, c;
cin >> a >> b >> c;
if (a == 1) {
update(b, c);
}
else if (a == 2) {
cout << query(1, 1, S, b, c)<<'\n';
}
}
return 0;
}
'알고리즘 풀이 > 트리' 카테고리의 다른 글
BOJ : 13537 수열과 쿼리 1 (0) | 2021.01.25 |
---|---|
BOJ : 5670 휴대폰 자판 (0) | 2021.01.16 |
BOJ : 9202 Boggle (0) | 2021.01.14 |
BOJ : 2243 사탕상자 (0) | 2021.01.14 |
댓글
공지사항
최근에 올라온 글
최근에 달린 댓글
- Total
- Today
- Yesterday
링크
TAG
- ReactNative
- 알고리즘
- 그래프
- BFS
- boj
- 백준
- 컴퓨터 통신
- 중앙대학교
- 벨만포드
- nestjs
- nest.js
- 스레드
- java
- dfs
- typeORM
- 세그먼트 트리
- 백트래킹
- nodeJS
- 동적계획법
- 그리디
- 자바스크립트
- 구현
- 자바
- Computer Architecture
- node.js
- 예외처리
- 시뮬레이션
- 투포인터
- 컴퓨터 구조
- 재귀
일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
1 | ||||||
2 | 3 | 4 | 5 | 6 | 7 | 8 |
9 | 10 | 11 | 12 | 13 | 14 | 15 |
16 | 17 | 18 | 19 | 20 | 21 | 22 |
23 | 24 | 25 | 26 | 27 | 28 |
글 보관함