글 읽기 - "틀렸습니다" 어디가 틀린걸가요?

happyday 3년 전 0

로컬에서 샘플 입력값으로 테스트 해보면, 결과는 정상입니다.

혹시 어떤 부분이 문제일까요?

import java.io.*;

import java.util.*;

public class Main {

static int N, M, K, ans, base;

static int _MOD_ = 1000000007;

static long[] tree;

static int[] input;

public static void main(String[] args) throws IOException {

// TODO Auto-generated method stub

BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

StringTokenizer st = new StringTokenizer(br.readLine());

N = Integer.parseInt(st.nextToken()); // 수의 개수

M = Integer.parseInt(st.nextToken()); // 변경 횟수

K = Integer.parseInt(st.nextToken()); // 구간 곱 구하는 횟수

input = new int[N+1];

for(int i=1; i<=N; i++ ) {

st = new StringTokenizer(br.readLine());

input[i] = Integer.parseInt(st.nextToken());

}

base = 1;

do {

base *= 2;

} while( base < N );

tree = new long[base*2];

initTree();

int a, b, c;

for(int i=1; i<=M+K; i++) {

st = new StringTokenizer(br.readLine());

a = Integer.parseInt(st.nextToken());

b = Integer.parseInt(st.nextToken());

c = Integer.parseInt(st.nextToken());

if ( a == 1 ) {

update(b, c);

} else if ( a == 2 ) {

bw.write(String.valueOf(query( b, c) % _MOD_));

}

// bw.write("\n");

bw.flush();

}

bw.close();

br.close();

}

static void initTree() {

int before = base + N ;

int initIdx = 1;

int initValue;

for(int i=base; i< before; i++) {

initValue = input[initIdx];

tree[i] = initValue;

initIdx++;

}

int left, right;

for(int i=base-1; i >= 1; i--) {

left = i*2;

right = i*2+1;

tree[i] = tree[left] * tree[right];

}

}

static void update( int pos, int value ) {

int index = pos + base -1;

int left, right;

tree[index] = value;

while ( index > 1 ) {

index = index/2;

left = index * 2 ;

right = index * 2 + 1;

tree[index] = tree[left] * tree[right];

}

}

static long query ( int start, int end ) {

long ret = 1;

int fromIdx = base + start -1;

int toIdx = base + end -1;

while( fromIdx <= toIdx) {

if ( (fromIdx & 1) == 1 ) { // 혹수

ret *= tree[fromIdx++];

}

if ( ( toIdx & 1 ) == 0 ) { // 짝수

ret *= tree[toIdx--];

}

fromIdx = fromIdx/2; // parent

toIdx = toIdx/2; // parent

}

return ret;

}

}

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

"틀렸습니다" 어디가 틀린걸가요?

happyday 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말