結果

問題 No.1000 Point Add and Array Add
ユーザー ks2m
提出日時 2020-02-28 22:21:31
言語 Java
(openjdk 23)
結果
AC  
実行時間 806 ms / 2,000 ms
コード長 1,695 bytes
コンパイル時間 2,109 ms
コンパイル使用メモリ 78,672 KB
実行使用メモリ 77,116 KB
最終ジャッジ日時 2024-10-13 17:53:59
合計ジャッジ時間 10,885 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.List;

public class Main {
	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		String[] sa = br.readLine().split(" ");
		int n = Integer.parseInt(sa[0]);
		int q = Integer.parseInt(sa[1]);

		sa = br.readLine().split(" ");
		long[] a = new long[n];
		for (int i = 0; i < n; i++) {
			a[i] = Integer.parseInt(sa[i]);
		}

		BIT bit = new BIT(n);
		List<Obj> list = new ArrayList<>();
		for (int i = 0; i < q; i++) {
			sa = br.readLine().split(" ");
			if ("A".equals(sa[0])) {
				Obj o = new Obj();
				o.i = Integer.parseInt(sa[1]);
				o.y = Integer.parseInt(sa[2]);
				o.cnt = bit.sum(o.i);
				list.add(o);
			} else {
				bit.add(Integer.parseInt(sa[1]), 1);
				bit.add(Integer.parseInt(sa[2]) + 1, -1);
			}
		}
		br.close();

		long[] b = new long[n];
		for (int i = 0; i < b.length; i++) {
			b[i] = a[i] * bit.sum(i + 1);
		}
		for (Obj o : list) {
			b[o.i - 1] += o.y * (bit.sum(o.i) - o.cnt);
		}

		StringBuilder sb = new StringBuilder();
		for (int i = 0; i < b.length; i++) {
			sb.append(b[i]).append(' ');
		}
		sb.deleteCharAt(sb.length() - 1);
		System.out.println(sb.toString());
	}

	static class Obj {
		int i, cnt;
		long y;
	}

	static class BIT {
		int n;
		long[] arr;

		public BIT(int n) {
			this.n = n;
			arr = new long[n + 1];
		}

		void add(int idx, long val) {
			for (int i = idx; i <= n; i += i & -i) {
				arr[i] += val;
			}
		}

		int sum(int idx) {
			int sum = 0;
			for (int i = idx; i > 0; i -= i & -i) {
				sum += arr[i];
			}
			return sum;
		}
	}
}
0