forked from ckcz123/codejam
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathC.java
More file actions
49 lines (43 loc) · 1.56 KB
/
Copy pathC.java
File metadata and controls
49 lines (43 loc) · 1.56 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
import java.io.PrintStream;
import java.util.*;
/**
* Codejam 2009 Round 1C
* Problem C. Bribe the Prisoners
*/
public class Main {
public String solve(Scanner scanner) {
int n=scanner.nextInt(), m=scanner.nextInt();
int[] a=new int[m];
for (int i=0;i<m;i++) a[i]=scanner.nextInt();
return String.valueOf(dfs(1, n, a, new HashMap<>()));
}
private int dfs(int start, int end, int[] a, HashMap<Integer, Integer> map) {
if (end<start) return 0;
int state=20000*start+end, ans=Integer.MAX_VALUE;
if (map.containsKey(state)) return map.get(state);
for (int v: a) {
if (v>=start && v<=end)
ans=Math.min(ans, end-start+dfs(start, v-1, a, map)+dfs(v+1, end, a, map));
}
if (ans==Integer.MAX_VALUE) ans=0;
map.put(state, ans);
return ans;
}
public static void main(String[] args) throws Exception {
System.setOut(new PrintStream("output.txt"));
Scanner scanner=new Scanner(System.in);
int times=Integer.parseInt(scanner.nextLine());
long start=System.currentTimeMillis();
for (int t=1;t<=times;t++) {
try {
System.out.println(String.format("Case #%d: %s", t, new Main().solve(scanner)));
}
catch (Throwable e) {
System.err.println("ERROR in case #"+t);
e.printStackTrace();
}
}
long end=System.currentTimeMillis();
System.err.println(String.format("Time used: %.3fs", (end-start)/1000.0));
}
}