Self-Driving-Rides/main.py at main · HashCode-ICL-2022/Self-Driving-Rides · GitHub

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
from ast import arg
from utils.utils import score
from utils.parse import import_data

from tqdm import tqdm

import argparse

def main_random(fname):

    R, C, F, N, B, T, rides, cars = import_data(fname)

    t = 0
    total_score = 0
    count = 0

    for t in tqdm(range(T)):
        for car in cars:
            count += 1
            finished, ride = car.check_ride_finished(t)
            if car.is_free:
                if finished and ride.finished_on_time:
                    total_score += score([ride], B)
                if len(rides) > 0:
                    car.add_ride(rides.pop(), t)

    print(f"{total_score:,}")


def main(fname):

    R, C, F, N, B, T, rides, cars = import_data(fname)

    t = 0
    total_score = 0
    count = 0


    for t in tqdm(range(T)):
        for car in cars:
            count += 1
            finished, ride = car.check_ride_finished(t)
            if car.is_free:
                if finished and ride.finished_on_time:
                    total_score += score([ride], B)
                if len(rides) > 0:
                    greedy_ride, rides = select_greedy_ride(car, rides, t)
                    car.add_ride(greedy_ride, t)

    print(f"{total_score:,}")


def select_greedy_ride(car, rides, t):

    rides.sort(key=lambda x: x.time_to_start(car.distance_to_start(x) + t))
    greedy_ride = rides.pop(0)
    return greedy_ride, rides


if __name__ == "__main__":

    parser = argparse.ArgumentParser()
    parser.add_argument('--fname', '-f', default='b')
    args = parser.parse_args()
    fname_dict = {
        "a" : "a_example.in",
        "b" : "b_should_be_easy.in",
        "c" : "c_no_hurry.in",
        "d" : "d_metropolis.in",
        "e" : "e_high_bonus.in",
    }
    main(fname_dict[args.fname])