[Algorithm/Baekjoon] 태상이의 훈련소 생활 - 19951 (G5/JAVA)

📑문제링크

https://www.acmicpc.net/problem/19951

💻코드

import java.io.*;
import java.util.StringTokenizer;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        StringTokenizer st = new StringTokenizer(br.readLine());
        int N = Integer.parseInt(st.nextToken());
        int M = Integer.parseInt(st.nextToken());

        int[] ground = new int[N];
        int[] directions = new int[N + 1];
        st = new StringTokenizer(br.readLine());
        for (int i = 0; i < N; i++) {
            ground[i] = Integer.parseInt(st.nextToken());
        }

        for (int i = 0; i < M; i++) {
            st = new StringTokenizer(br.readLine());
            int a = Integer.parseInt(st.nextToken()) - 1;
            int b = Integer.parseInt(st.nextToken());
            int k = Integer.parseInt(st.nextToken());
            directions[a] += k;
            directions[b] -= k;
        }
        for (int i = 0; i < N; i++) {
            directions[i + 1] += directions[i];
            System.out.print((ground[i] += directions[i]) + " ");
        }
        br.close();
    }

}

⏳회고

• 요청을 그때그때 처리하면 시간 초과가 발생한다.
• 시작 인덱스와 끝 인덱스에 요청 값을 저장해 두고 한번에 연산해 줘야 한다.