diff --git a/du6/a_station.c b/du6/a_station.c index 9fe09ae..c1c6f64 100644 --- a/du6/a_station.c +++ b/du6/a_station.c @@ -63,6 +63,15 @@ int count_targets(struct station* station){ } int count_capacity(struct station* station){ - return 0; + int capacity = 0; + + for (int i = 0; i < station -> track_count; i++;){ + struct car* vagon = station ->tracks[i]; // kontrolujem kazdu kolaj, kazdy vagon linked hash tablu + while (vagon) { + capacity = capacity + vagon -> capacity; + vagon = vagon -> next; // zapisem kapacitu kazdeho vagonu na kolaji + } + } + return capacity; }