forked from maiquynhtruong/algorithms-and-problems
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathjapan.java
66 lines (61 loc) · 2 KB
/
japan.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
// http://www.spoj.com/problems/MSE06H/
/* package whatever; // don't place package name! */
import java.util.*;
import java.lang.*;
import java.io.*;
/* Name of the class has to be "Main" only if the class is public. */
class Ideone
{
public static void main (String[] args) throws java.lang.Exception
{
List<int[]> bridges = new ArrayList<int[]>();
Scanner sc = new Scanner(System.in);
int n = sc.nextInt(), m = sc.nextInt(), k = sc.nextInt();
for (int i = 0; i < k; i++) {
int east = sc.nextInt(), west = sc.nextInt();
bridges.add(new int[] {east, west});
}
for (int i = 0; i < k; i++)
// System.out.println("i= " + i + ", east= " + bridges.get(i)[0] + ", west= " + bridges.get(i)[1]);
// System.out.println();
Collections.sort(bridges, new Comparator<int[]>() {
public int compare(int[] bridge1, int[] bridge2) {
if (bridge1[0] == bridge2[0]) return -(bridge2[1] - bridge1[1]);
else return -(bridge2[0] - bridge1[0]);
}
});
for (int i = 0; i < k; i++)
System.out.println("i= " + i + ", east= " + bridges.get(i)[0] + ", west= " + bridges.get(i)[1]);
int[] BIT = new int[20];
Ideone o = new Ideone();
System.out.println(o.solve(m, k, bridges, BIT));
}
public int solve(int m, int k, List<int[]> bridges, int[] BIT) {
int sum = 0;
for (int i = 0; i < k; i++) {
System.out.println("i= " + i + ", west= " + bridges.get(i)[1]);
System.out.println("Pre-update: " + Arrays.toString(BIT));
update(bridges.get(i)[1], m, BIT);
System.out.println("Post-update: " + Arrays.toString(BIT));
int sumM = query(m, BIT);
int sumI = query(bridges.get(i)[1], BIT);
sum += sumM - sumI;
System.out.println("sum= " + sum + ", sumM= " + sumM + ", sumI= " + sumI);
}
return sum;
}
public void update(int index, int max, int[] BIT) {
while (index <= max) {
BIT[index] += 1;
index += index & (-index);
}
}
public int query(int index, int[] BIT) {
int ans = 0;
while (index > 0) {
ans += BIT[index];
index -= index & (-index);
}
return ans;
}
}