main.go 1.9 KB

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