/
zhk_info.go
87 lines (66 loc) · 1.39 KB
/
zhk_info.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
79
80
81
82
83
84
85
86
87
package internal
const (
minFloor = 2
maxFloor = 23
floorCount = maxFloor - minFloor + 1
buildingCount = 4
minFlat = 1
liftCount = 3
)
var (
flatPerFloor []int
flatPerBuilding []int
maxFlat int
)
func init() {
flatPerFloor = []int{10, 9, 8, 9}
for _, k := range flatPerFloor {
flatPerBuilding = append(flatPerBuilding, k*floorCount)
}
for _, k := range flatPerBuilding {
maxFlat += k
}
}
func getBuilding(flat int) int {
for i := 0; i < buildingCount; i++ {
if flat <= flatPerBuilding[i] {
return i + 1
}
flat -= flatPerBuilding[i]
}
return 0
}
func getFloor(flat int) int {
var i int
for ; i < buildingCount; i++ {
if flat <= flatPerBuilding[i] {
break
}
flat -= flatPerBuilding[i]
}
return (flat-1)/flatPerFloor[i] + minFloor
}
func getMinBuildingFlat(building int) int {
flat := 1
for i := 0; i < building-1; i++ {
flat += flatPerBuilding[i]
}
return flat
}
func getMaxBuildingFlat(building int) int {
flat := 0
for i := 0; i < building; i++ {
flat += flatPerBuilding[i]
}
return flat
}
func getMinFloorFlat(building, floor int) int {
flat := getMinBuildingFlat(building)
flat += (floor - minFloor) * flatPerFloor[building-1]
return flat
}
func getMaxFloorFlat(building, floor int) int {
flat := getMinBuildingFlat(building) - 1
flat += (floor - minFloor + 1) * flatPerFloor[building-1]
return flat
}