Fenwick Tree Java
Ich habe versucht, den Fenwick-Baum in Java zu implementieren, erhalte jedoch nicht das gewünschte Ergebnis. Hier ist mein Code:
import java.io.*;
import java.util.*;
import java.math.*;
class fenwick1 {
public static int N;
public static long[] a;
public static void main(String[] args)throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
N = Integer.parseInt(br.readLine());
a = new long[N];
String[] str = br.readLine().split(" ");
for (int i = 0; i < N; i++) {
a[i] = Long.parseLong(str[i]);
}
increment(2, 10);
System.out.println(a[2]);
System.out.println(query(4));
}
public static void increment(int at, int by) {
while (at < a.length) {
a[at] += by;
at |= (at + 1);
}
}
public static int query(int at) {
int res = 0;
while (at >= 0) {
res += a[at];
at = (at & (at + 1)) - 1;
}
return res;
}
}
Wenn ich etwas eingebe:
10
1 2 3 4 5 6 7 8 9 10
Ich bekomme
13
19
So funktioniert die Inkrementfunktion einwandfrei. Aber Abfrage (4) sollte die kumulative Summe bis zu Index 4 ergeben, d. H.
(1 + 2 + 13 + 4 + 5) = 25