main.go 2.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169
  1. package main
  2. import (
  3. "fmt"
  4. "os"
  5. )
  6. var input *os.File
  7. var output *os.File
  8. var R int
  9. var C int
  10. var F int
  11. var N int
  12. var B int
  13. var T int
  14. var Rides []*Ride
  15. var Cars []*Car
  16. var Sched Scheduler
  17. type Ride struct {
  18. ID int
  19. a, b, x, y, s, f int
  20. used bool
  21. }
  22. type Scheduler interface {
  23. Add(*Car)
  24. Pop() *Car
  25. }
  26. type Car struct {
  27. ID int
  28. Rides []int
  29. Arrival int
  30. X int
  31. Y int
  32. }
  33. func (c *Car) Update(r *Ride) {
  34. c.moveTo(r.a, r.b)
  35. c.moveTo(r.x, r.y)
  36. c.Rides = append(c.Rides, r.ID)
  37. }
  38. func (c *Car) moveTo(x, y int) {
  39. xdist := c.X - x
  40. if xdist < 0 {
  41. xdist = -xdist
  42. }
  43. ydist := c.Y - y
  44. if ydist < 0 {
  45. ydist = -ydist
  46. }
  47. c.Arrival += xdist + ydist
  48. c.X = x
  49. c.Y = y
  50. }
  51. func Choose(c *Car) *Ride { return nil }
  52. func assign() bool {
  53. c := Sched.Pop()
  54. if c == nil {
  55. return false
  56. }
  57. r := Choose(c)
  58. if r == nil {
  59. return true
  60. }
  61. r.used = true
  62. c.Update(r)
  63. Sched.Add(c)
  64. return true
  65. }
  66. func solve() {
  67. Sched = &prioq{}
  68. // create cars
  69. for i := 0; i < F; i++ {
  70. c := &Car{
  71. ID: i,
  72. Arrival: 0,
  73. X: 0,
  74. Y: 0,
  75. }
  76. Cars = append(Cars, c)
  77. Sched.Add(c)
  78. }
  79. for assign() {
  80. }
  81. for _, c := range Cars {
  82. fmt.Fprintf(output, "%d", len(c.Rides))
  83. for _, ri := range c.Rides {
  84. fmt.Fprintf(output, " %d", ri)
  85. }
  86. fmt.Fprintf(output, "\n")
  87. }
  88. }
  89. func main() {
  90. sample := os.Args[1]
  91. fileIn := sample + ".in"
  92. fileOut := sample + ".out"
  93. var err error
  94. input, err = os.Open(fileIn)
  95. if err != nil {
  96. panic(fmt.Sprintf("open %s: %v", fileIn, err))
  97. }
  98. output, err = os.Create(fileOut)
  99. if err != nil {
  100. panic(fmt.Sprintf("creating %s: %v", fileOut, err))
  101. }
  102. defer input.Close()
  103. defer output.Close()
  104. // Global
  105. R = readInt()
  106. C = readInt()
  107. F = readInt()
  108. N = readInt()
  109. B = readInt()
  110. T = readInt()
  111. for i := 0; i < N; i++ {
  112. Rides = append(Rides, &Ride{
  113. ID: i,
  114. a: readInt(),
  115. b: readInt(),
  116. x: readInt(),
  117. y: readInt(),
  118. s: readInt(),
  119. f: readInt(),
  120. })
  121. }
  122. solve()
  123. }
  124. func readInt() int {
  125. var i int
  126. fmt.Fscanf(input, "%d", &i)
  127. return i
  128. }
  129. func readString() string {
  130. var str string
  131. fmt.Fscanf(input, "%s", &str)
  132. return str
  133. }
  134. func readFloat() float64 {
  135. var x float64
  136. fmt.Fscanf(input, "%f", &x)
  137. return x
  138. }