main.go 1.6 KB

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