-
Notifications
You must be signed in to change notification settings - Fork 1
/
TheCowDivTwo.java
136 lines (125 loc) · 3.25 KB
/
TheCowDivTwo.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
import java.util.*;
import java.util.regex.*;
import java.text.*;
import java.math.*;
public class TheCowDivTwo {
public int find(int N, int K) {
final int mod = 1000000007;
long[][] dp = new long[49][1000];
dp[0][0] = 1;
for (int cowCount = 0; cowCount < N; cowCount++) {
for (int i = 47; i >= 0; i--) {
for (int j = 0;j < N; j++) {
dp[i + 1][(j + cowCount) % N] += dp[i][j];
dp[i + 1][(j + cowCount) % N] %= mod;
}
}
}
return (int) dp[K][0];
}
// BEGIN KAWIGIEDIT TESTING
// Generated by KawigiEdit-pf 2.3.0
private static boolean KawigiEdit_RunTest(int testNum, int p0, int p1,
boolean hasAnswer, int p2) {
System.out.print("Test " + testNum + ": [" + p0 + "," + p1);
System.out.println("]");
TheCowDivTwo obj;
int answer;
obj = new TheCowDivTwo();
long startTime = System.currentTimeMillis();
answer = obj.find(p0, p1);
long endTime = System.currentTimeMillis();
boolean res;
res = true;
System.out.println("Time: " + (endTime - startTime) / 1000.0
+ " seconds");
if (hasAnswer) {
System.out.println("Desired answer:");
System.out.println("\t" + p2);
}
System.out.println("Your answer:");
System.out.println("\t" + answer);
if (hasAnswer) {
res = answer == p2;
}
if (!res) {
System.out.println("DOESN'T MATCH!!!!");
} else if ((endTime - startTime) / 1000.0 >= 2) {
System.out.println("FAIL the timeout");
res = false;
} else if (hasAnswer) {
System.out.println("Match :-)");
} else {
System.out.println("OK, but is it right?");
}
System.out.println("");
return res;
}
public static void main(String[] args) {
boolean all_right;
boolean disabled;
boolean tests_disabled;
all_right = true;
tests_disabled = false;
int p0;
int p1;
int p2;
// ----- test 0 -----
disabled = false;
p0 = 7;
p1 = 4;
p2 = 5;
all_right = (disabled || KawigiEdit_RunTest(0, p0, p1, true, p2))
&& all_right;
tests_disabled = tests_disabled || disabled;
// ------------------
// ----- test 1 -----
disabled = false;
p0 = 1;
p1 = 1;
p2 = 1;
all_right = (disabled || KawigiEdit_RunTest(1, p0, p1, true, p2))
&& all_right;
tests_disabled = tests_disabled || disabled;
// ------------------
// ----- test 2 -----
disabled = false;
p0 = 58;
p1 = 4;
p2 = 7322;
all_right = (disabled || KawigiEdit_RunTest(2, p0, p1, true, p2))
&& all_right;
tests_disabled = tests_disabled || disabled;
// ------------------
// ----- test 3 -----
disabled = false;
p0 = 502;
p1 = 7;
p2 = 704466492;
all_right = (disabled || KawigiEdit_RunTest(3, p0, p1, true, p2))
&& all_right;
tests_disabled = tests_disabled || disabled;
// ------------------
// ----- test 4 -----
disabled = false;
p0 = 1000;
p1 = 47;
p2 = 219736903;
all_right = (disabled || KawigiEdit_RunTest(4, p0, p1, true, p2))
&& all_right;
tests_disabled = tests_disabled || disabled;
// ------------------
if (all_right) {
if (tests_disabled) {
System.out
.println("You're a stud (but some test cases were disabled)!");
} else {
System.out.println("You're a stud (at least on given cases)!");
}
} else {
System.out.println("Some of the test cases had errors.");
}
}
// END KAWIGIEDIT TESTING
}
// Powered by KawigiEdit-pf 2.3.0!