-
Notifications
You must be signed in to change notification settings - Fork 27
/
Copy path307.go
78 lines (71 loc) · 1.15 KB
/
307.go
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
// UVa 307 - Sticks
package main
import (
"fmt"
"os"
"sort"
)
var (
n, l int
sticks []int
visited []bool
)
func dfs(level, curr, length int) bool {
if level == n {
return true
}
for i := curr; i < n; i++ {
if !visited[i] {
switch {
case length+sticks[i] < l:
visited[i] = true
if dfs(level+1, i+1, length+sticks[i]) {
return true
}
visited[i] = false
if length == 0 {
return false
}
case length+sticks[i] == l:
visited[i] = true
if dfs(level+1, 0, 0) {
return true
}
visited[i] = false
return false
}
}
}
return false
}
func solve(sum int) int {
sort.Sort(sort.Reverse(sort.IntSlice(sticks)))
for i := n; i > 0; i-- {
if sum%i == 0 {
l = sum / i
visited = make([]bool, n)
if dfs(0, 0, 0) {
return l
}
}
}
return -1
}
func main() {
in, _ := os.Open("307.in")
defer in.Close()
out, _ := os.Create("307.out")
defer out.Close()
for {
if fmt.Fscanf(in, "%d", &n); n == 0 {
break
}
var sum int
sticks = make([]int, n)
for i := range sticks {
fmt.Fscanf(in, "%d", &sticks[i])
sum += sticks[i]
}
fmt.Fprintln(out, solve(sum))
}
}