-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathCrowdedCows.java
More file actions
80 lines (78 loc) · 2.05 KB
/
CrowdedCows.java
File metadata and controls
80 lines (78 loc) · 2.05 KB
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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
import java.io.*;
import java.util.*;
public class CrowdedCows {
static Comparator<int[]> c = new Comparator<int[]>()
{
public int compare(int[] o1, int[]o2)
{
return o1[0] - o2[0];
}
};
public static void main(String[] args) throws IOException
{
//BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
BufferedReader br = new BufferedReader(new FileReader("crowded.in"));
String[] temp = br.readLine().split(" ");
int N = Integer.parseInt(temp[0]);
int D = Integer.parseInt(temp[1]);
int[][] cows = new int[N][2];
int x;
int h;
for (int i = 0; i < N; i++)
{
temp = br.readLine().split(" ");
x = Integer.parseInt(temp[0]);
h = Integer.parseInt(temp[1]);
cows[i] = new int[] { x, h };
}
Arrays.sort(cows, c);
TreeMap<Integer, Integer> before = new TreeMap<Integer, Integer>();
TreeMap<Integer, Integer> after = new TreeMap<Integer, Integer>();
int i = 0;
int j = 0;
int k = 1;
int count;
int ans = 0;
while (i < N-1)
{
count = 0;
while (cows[i][0] - cows[j][0] >= D)
{
before.put(cows[j][1], before.get(cows[j][1])-1);
if (before.get(cows[j][1])==0)
before.remove(cows[j][1]);
j++;
}
while (k < N && cows[k][0] - cows[i][0] <= D)
{
if (after.containsKey(cows[k][1]))
after.put(cows[k][1], after.get(cows[k][1])+1);
else
after.put(cows[k][1], 1);
k++;
}
//System.out.println(i);
//System.out.println(before);
//System.out.println(after);
if (!before.isEmpty() && before.lastKey() >= 2*cows[i][1])
count++;
if (!after.isEmpty() && after.lastKey() >= 2*cows[i][1])
count++;
if (count == 2)
ans++;
if (before.containsKey(cows[i][1]))
before.put(cows[i][1], before.get(cows[i][1])+1);
else
before.put(cows[i][1], 1);
i++;
after.put(cows[i][1], after.get(cows[i][1])-1);
if (after.get(cows[i][1])==0)
after.remove(cows[i][1]);
}
//System.out.println(ans);
PrintWriter pw = new PrintWriter(new BufferedWriter(new FileWriter("crowded.out")));
pw.println(ans);
pw.close();
br.close();
}
}