CCF 201709-5 除法 Java

问题描述

　　小葱喜欢除法，所以他给了你N个数a1, a2, ⋯, aN，并且希望你执行M次操作，每次操作可能有以下两种：

　　给你三个数l, r, v，你需要将al, al+1, ⋯, ar之间所有v的倍数除以v。

　　给你两个数l, r，你需要回答al + al+1 + ⋯ + ar的值是多少。

输入格式

　　第一行两个整数N, M，代表数的个数和操作的次数。

　　接下来一行N个整数，代表N个数一开始的值。

　　接下来M行，每行代表依次操作。每行开始有一个整数opt。如果opt=1，那么接下来有三个数l, r, v，代表这次操作需要将第l个数到第r个数中v的倍数除以v；如果opt = 2，那么接下来有两个数l, r，代表你需要回答第l个数到第r个数的和。

输出格式

　　对于每一次的第二种操作，输出一行代表这次操作所询问的值。

样例输入

5 3

1 2 3 4 5

2 1 5

1 1 3 2

2 1 5

样例输出

15

14

评测用例规模与约定

　　对于30%的评测用例，1 ≤ N, M ≤ 1000；

　　对于另外20%的评测用例，第一种操作中一定有l = r；

　　对于另外20%的评测用例，第一种操作中一定有l = 1 , r = N；

　　对于100%的评测用例，1 ≤ N, M ≤ 105，0 ≤ a1, a2, ⋯, aN ≤ 106, 1 ≤ v ≤ 106, 1 ≤ l ≤ r ≤ N。

import java.util.Scanner;

public class Main {

private static int[] arr = new int[100010];
private static long[] tarr = new long[100010];
private static int N;

private static int lowbit(int i) {
return i & (-i);
}

private static void toValue(int i, int num) {
while (i <= N) {
tarr[i] += num;
i += lowbit(i);
}
}

private static long sum(int i) {
long total = 0;
while(i != 0) {
total += tarr[i];
i -= lowbit(i);
}
return total;
}

private static void fun1(int l, int r, int v) {
if (v == 1) return;
for (int i = l; i <= r; i++) {
if (arr[i] >= v && arr[i] % v == 0) {
toValue(i, arr[i] / v - arr[i]);
arr[i] /= v;
}
}
}

private static void fun2(int l, int r) {
System.out.println(sum(r) - sum(l-1));
}

public static void main(String[] args) {
Scanner In = new Scanner(System.in);
N = In.nextInt();
int M = In.nextInt();

arr = new int[N + 1];
tarr = new long[N + 1];
for (int i = 1; i <= N; i++) {
arr[i] = In.nextInt();
toValue(i, arr[i]);
}

while (M-- > 0) {
int opt = In.nextInt();
if (opt == 1) {
int a = In.nextInt();
int b = In.nextInt();
int c = In.nextInt();
fun1(a, b, c);
}
else if (opt == 2) {
int l = In.nextInt();
int r = In.nextInt();
fun2(l, r);
}
}
}
}