[백준][Java] 19951번 태상이의 훈련소 생활 (누적합)

[백준][Java] 19951번 태상이의 훈련소 생활 (누적합)

import java.io.BufferedReader;

import java.io.BufferedWriter;

import java.io.IOException;

import java.io.InputStreamReader;

import java.io.OutputStreamWriter;

public class Main {

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

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

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

String [] str = br.readLine(). split ( " " );

int N = stoi(str[ 0 ]);

int M = stoi(str[ 1 ]);

int [] res = new int [N + 1 ];

int [] accumK = new int [N + 2 ];

str = br.readLine(). split ( " " );

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

res[i] = stoi(str[i - 1 ]);

}

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

str = br.readLine(). split ( " " );

int a = stoi(str[ 0 ]);

int b = stoi(str[ 1 ]);

int k = stoi(str[ 2 ]);

accumK[a] + = k;

accumK[b + 1 ] - = k;

}

int accum = 0 ;

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

accum + = accumK[i];

res[i] + = accum;

System . out . print (res[i] + " " );

}

// bw.write("");

// bw.flush();

// bw.close();

}

private static int stoi( String input) {

return Integer. parseInt (input);

}

}

from http://aig2029.tistory.com/285 by ccl(A) rewrite - 2021-09-19 04:01:53