-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathRabbitNumbering.java
138 lines (111 loc) · 3.41 KB
/
RabbitNumbering.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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
import java.util.Arrays;
public class RabbitNumbering {
public int theCount(int[] maxNumber) {
Arrays.sort(maxNumber);
long ans = 1;
for(int i=0;i<maxNumber.length;i++)
ans = (ans * (maxNumber[i]-i))%1000000007;
return (int)ans;
}
// BEGIN CUT HERE
public static void main(String[] args) {
if (args.length == 0) {
RabbitNumberingHarness.run_test(-1);
} else {
for (int i=0; i<args.length; ++i)
RabbitNumberingHarness.run_test(Integer.valueOf(args[i]));
}
}
// END CUT HERE
}
// BEGIN CUT HERE
class RabbitNumberingHarness {
public static void run_test(int casenum) {
if (casenum != -1) {
if (runTestCase(casenum) == -1)
System.err.println("Illegal input! Test case " + casenum + " does not exist.");
return;
}
int correct = 0, total = 0;
for (int i=0;; ++i) {
int x = runTestCase(i);
if (x == -1) {
if (i >= 100) break;
continue;
}
correct += x;
++total;
}
if (total == 0) {
System.err.println("No test cases run.");
} else if (correct < total) {
System.err.println("Some cases FAILED (passed " + correct + " of " + total + ").");
} else {
System.err.println("All " + total + " tests passed!");
}
}
static boolean compareOutput(int expected, int result) { return expected == result; }
static String formatResult(int res) {
return String.format("%d", res);
}
static int verifyCase(int casenum, int expected, int received) {
System.err.print("Example " + casenum + "... ");
if (compareOutput(expected, received)) {
System.err.println("PASSED");
return 1;
} else {
System.err.println("FAILED");
System.err.println(" Expected: " + formatResult(expected));
System.err.println(" Received: " + formatResult(received));
return 0;
}
}
static int runTestCase(int casenum__) {
switch(casenum__) {
case 0: {
int[] maxNumber = {5};
int expected__ = 5;
return verifyCase(casenum__, expected__, new RabbitNumbering().theCount(maxNumber));
}
case 1: {
int[] maxNumber = {4, 4, 4, 4};
int expected__ = 24;
return verifyCase(casenum__, expected__, new RabbitNumbering().theCount(maxNumber));
}
case 2: {
int[] maxNumber = {5, 8};
int expected__ = 35;
return verifyCase(casenum__, expected__, new RabbitNumbering().theCount(maxNumber));
}
case 3: {
int[] maxNumber = {2, 1, 2};
int expected__ = 0;
return verifyCase(casenum__, expected__, new RabbitNumbering().theCount(maxNumber));
}
case 4: {
int[] maxNumber = {25, 489, 76, 98, 704, 98, 768, 39, 697, 8, 56, 74, 36, 95, 87, 2, 968, 4, 920, 54, 873, 90};
int expected__ = 676780400;
return verifyCase(casenum__, expected__, new RabbitNumbering().theCount(maxNumber));
}
// custom cases
/* case 5: {
int[] maxNumber = ;
int expected__ = ;
return verifyCase(casenum__, expected__, new RabbitNumbering().theCount(maxNumber));
}*/
/* case 6: {
int[] maxNumber = ;
int expected__ = ;
return verifyCase(casenum__, expected__, new RabbitNumbering().theCount(maxNumber));
}*/
/* case 7: {
int[] maxNumber = ;
int expected__ = ;
return verifyCase(casenum__, expected__, new RabbitNumbering().theCount(maxNumber));
}*/
default:
return -1;
}
}
}
// END CUT HERE