diff --git a/cv6/program.c b/cv6/program.c index 945c9b4..71d63c1 100644 --- a/cv6/program.c +++ b/cv6/program.c @@ -1 +1,141 @@ -. \ No newline at end of file +#include +#include +#include +#include "a_station.h" + +// Функція для створення порожньої станції +struct station* create_station() { + struct station* new_station = (struct station*)malloc(sizeof(struct station)); + if (new_station == NULL) { + return NULL; + } + new_station->tracks = (struct car**)malloc(sizeof(struct car*) * STATION_SIZE); + if (new_station->tracks == NULL) { + free(new_station); + return NULL; + } + for (int i = 0; i < STATION_SIZE; i++) { + new_station->tracks[i] = NULL; + } + new_station->track_count = STATION_SIZE; + return new_station; +} + +// Функція для звільнення пам'яті, зайнятої станцією +void destroy_station(struct station* station) { + if (station != NULL) { + for (int i = 0; i < station->track_count; i++) { + struct car* current = station->tracks[i]; + while (current != NULL) { + struct car* next = current->next; + free(current); + current = next; + } + } + free(station->tracks); + free(station); + } +} + +// Хеш-функція для вибору треку +int select_track(struct station* station, const char* target) { + unsigned int hash = 0; + for (int i = 0; target[i] != '\0'; i++) { + hash = (hash * 31 + target[i]) % station->track_count; + } + return hash; +} + +// Додавання або збільшення запису про кількість пасажирів +void add_target_capacity(struct station* station, const char* target, int capacity) { + int track_index = select_track(station, target); + struct car* current = station->tracks[track_index]; + + while (current != NULL) { + if (strcmp(current->value, target) == 0) { + current->capacity += capacity; + return; + } + current = current->next; + } + + // Створення нового запису, якщо цільова станція не знайдена + struct car* new_car = (struct car*)malloc(sizeof(struct car)); + if (new_car == NULL) { + return; + } + strncpy(new_car->value, target, TARGET_SIZE - 1); + new_car->value[TARGET_SIZE - 1] = '\0'; + new_car->capacity = capacity; + new_car->next = station->tracks[track_index]; + station->tracks[track_index] = new_car; +} + +// Отримання запису про кількість пасажирів для цільової станції +int get_target_capacity(struct station* station, const char* target) { + int track_index = select_track(station, target); + struct car* current = station->tracks[track_index]; + + while (current != NULL) { + if (strcmp(current->value, target) == 0) { + return current->capacity; + } + current = current->next; + } + return 0; +} + +// Підрахунок кількості цільових станцій +int count_targets(struct station* station) { + int target_count = 0; + for (int i = 0; i < station->track_count; i++) { + struct car* current = station->tracks[i]; + while (current != NULL) { + target_count++; + current = current->next; + } + } + return target_count; +} + +// Підрахунок загальної кількості пасажирів на всі станції +int count_capacity(struct station* station) { + int total_capacity = 0; + for (int i = 0; i < station->track_count; i++) { + struct car* current = station->tracks[i]; + while (current != NULL) { + total_capacity += current->capacity; + current = current->next; + } + } + return total_capacity; +} + + +int main() { + // Створення станції + struct station* my_station = create_station(); + if (my_station == NULL) { + printf("Помилка створення станції.\n"); + return 1; + } + + // Додавання нових записів + add_target_capacity(my_station, "Kyiv", 50); + add_target_capacity(my_station, "Lviv", 30); + add_target_capacity(my_station, "Odessa", 20); + + // Отримання інформації про місто + printf("Кількість пасажирів до Kyiv: %d\n", get_target_capacity(my_station, "Kyiv")); + printf("Кількість пасажирів до Lviv: %d\n", get_target_capacity(my_station, "Lviv")); + printf("Кількість пасажирів до Odessa: %d\n", get_target_capacity(my_station, "Odessa")); + + // Підрахунок загальної кількості станцій та пасажирів + printf("Загальна кількість станцій: %d\n", count_targets(my_station)); + printf("Загальна кількість пасажирів: %d\n", count_capacity(my_station)); + + // Звільнення пам'яті + destroy_station(my_station); + + return 0; +} \ No newline at end of file