hashcode_2018/Car.py at master · bonheml/hashcode_2018 · 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
from algo_utils import get_manhattan


class Car:
    def __init__(self, max_time, bonus):
        self.position = (0, 0)
        self.passed_rides = []
        self.current_ride = None
        self.max_time = max_time
        self.ctime = 0
        self.goal = (0, 0)
        self.ride_time = 0
        self.bonus = bonus

    def choose_best_ride(self, rides):
        scores = []
        ride = None
        for r in rides:
            dist = get_manhattan(self.position, r.start)
            if r.score + dist <= (self.max_time - self.ctime):
                ride_start = abs(self.ctime + dist - r.earliest_start)
                score = r.score - ride_start
                if ride_start == 0:
                    score += self.bonus
                scores.append((score, r))
        if scores:
            ride = max(scores, key=lambda x: x[0])[1]
            rides.remove(ride)
        return ride

    def get_new_ride(self, rides):
        self.current_ride = self.choose_best_ride(rides)
        if self.current_ride is None:
            return
        if self.position != self.current_ride.start:
            self.goal = self.current_ride.start
            self.ride_time = get_manhattan(self.position,
                                           self.current_ride.start)
        else:
            self.goal = self.current_ride.goal
            self.ride_time = get_manhattan(self.position,
                                           self.current_ride.goal)

    def update_position(self, rides):
        self.position = self.goal
        if self.current_ride.goal == self.goal:
            self.passed_rides.append(self.current_ride)
            self.get_new_ride(rides)
        elif self.ctime >= self.current_ride.earliest_start:
            self.goal = self.current_ride.goal
            self.ride_time = self.current_ride.score

    def drive(self, rides):
        if self.current_ride is None:
            self.get_new_ride(rides)
        elif self.ride_time <= 0:
            self.update_position(rides)
        self.ride_time -= 1
        self.ctime += 1

    def __str__(self):
        res = "{}".format(len(self.passed_rides))
        for p in self.passed_rides:
            res += " {}".format(p.id)
        return res