-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathCereal2.java
More file actions
71 lines (70 loc) · 1.7 KB
/
Cereal2.java
File metadata and controls
71 lines (70 loc) · 1.7 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
import java.io.*;
import java.util.Arrays;
public class Cereal2 {
static int[][] preferences;
static int[] cowsusing;
public static void main(String[] args) throws IOException
{
//BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
BufferedReader br = new BufferedReader(new FileReader("cereal.in"));
String[] temp = br.readLine().split(" ");
int N = Integer.parseInt(temp[0]);
int M = Integer.parseInt(temp[1]);
int[] sol = new int[N];
cowsusing = new int[M];
preferences = new int[N][2];
int first, second;
for (int i = 0; i < N; i++)
{
temp = br.readLine().split(" ");
first = Integer.parseInt(temp[0])-1;
second = Integer.parseInt(temp[1])-1;
preferences[i] = new int[] {first, second};
}
Arrays.fill(cowsusing, -1);
int cereal, current;
int count = 0;
for (int i = N-1; i > -1; i--)
{
cereal = preferences[i][0];
current = i;
count++;
if (cowsusing[cereal] == -1)
cowsusing[cereal] = i;
else
{
int evicted = cowsusing[cereal];
cowsusing[cereal] = i;
if (!resettle(evicted, cereal))
count--;
}
sol[i] = count;
}
PrintWriter pw = new PrintWriter(new BufferedWriter(new FileWriter("cereal.out")));
for (int el: sol)
pw.println(el);
pw.close();
br.close();
}
public static boolean resettle(int evicted, int cereal)
{
if (cereal == preferences[evicted][0])
{
int second = preferences[evicted][1];
if (cowsusing[second] == -1)
{
cowsusing[second] = evicted;
return true;
}
if (cowsusing[second] > evicted)
{
int new_evict = cowsusing[second];
cowsusing[second] = evicted;
return resettle(new_evict, second);
}
else
return false;
}
return false;
}
}