[BOJ] 백준 11505번 : 구간 곱 구하기 (JAVA)

문제

어떤 N개의 수가 주어져 있다. 그런데 중간에 수의 변경이 빈번히 일어나고 그 중간에 어떤 부분의 곱을 구하려 한다. 만약에 1, 2, 3, 4, 5 라는 수가 있고, 3번째 수를 6으로 바꾸고 2번째부터 5번째까지 곱을 구하라고 한다면 240을 출력하면 되는 것이다. 그리고 그 상태에서 다섯 번째 수를 2로 바꾸고 3번째부터 5번째까지 곱을 구하라고 한다면 48이 될 것이다.

 

 

입력

첫째 줄에 수의 개수 N(1 ≤ N ≤ 1,000,000)과 M(1 ≤ M ≤ 10,000), K(1 ≤ K ≤ 10,000) 가 주어진다. M은 수의 변경이 일어나는 횟수이고, K는 구간의 곱을 구하는 횟수이다. 그리고 둘째 줄부터 N+1번째 줄까지 N개의 수가 주어진다. 그리고 N+2번째 줄부터 N+M+K+1 번째 줄까지 세 개의 정수 a,b,c가 주어지는데, a가 1인 경우 b번째 수를 c로 바꾸고 a가 2인 경우에는 b부터 c까지의 곱을 구하여 출력하면 된다.

 

입력으로 주어지는 모든 수는 0보다 크거나 같고, 1,000,000보다 작거나 같은 정수이다.

 

 

출력

첫째 줄부터 K줄에 걸쳐 구한 구간의 곱을 1,000,000,007로 나눈 나머지를 출력한다.

 

 

풀이

세그먼트 트리 입문용 문제였습니다. 특히, 이전 게시글인 '구간 합 구하기'와 상당히 유사했습니다.

 

 

[BOJ] 백준 2042번 : 구간 합 구하기 (JAVA)

 

 

위 게시글에서 구간 합을 구할 때, update 메소드에서 리프 노드부터 시작하지 않고, 위에서부터 아래로 구간에 해당되는 원소가 있으면, 그 요소에 dif만큼 더해주었습니다.

 

하지만, 곱셈에서는 0을 곱하게 될 경우, 한 번 0으로 초기화되면 다시 0이 아닌 수로 복구가 안 됩니다.

따라서, 구간 곱을 구할 때는 리프 노드부터 위로, 트리의 특정 가지 전체에 대해서 곱셈을 수행해 주어야합니다.

 

이 부분만 주의하시면 쉽게 해결하실 수 있습니다.

아래 소스코드를 참고하시길 바랍니다.

 

 

소스코드

 

지적 혹은 조언 환영합니다! 언제든지 댓글로 남겨주세요.