mipt_clang/algo/first/dijkstra.c

#include <stdlib.h>
#include <stdio.h>
#include <limits.h>

/* O(n^2) */

int* solve(int** massiv, int n, int x); // 0 вершина(можно заменить в коде)
void route(int** massiv, int* problem, int n, int x);// для 0 вершины

int main() {
	int n;
	int x;

	scanf("%d", &n);
	int **adj = (int**)calloc(n, sizeof(int*));
	for (int i = 0; i < n; i++) {
		adj[i] = (int*)calloc(n, sizeof(int));
	}
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < n; j++) {
			scanf("%d", &adj[i][j]);
		}
	}

	/* Номер вершины */
	scanf("%d", &x);

	int *res = solve(adj, n, x);

	route(adj, res, n, x);

	for (int i = 0; i < n; i++) {
		free(adj[i]);
	}
	free(adj);

	free(res);
}

int *solve(int **adj, int n, int x) {
	/* Пройденные вершины - 1, непройденные - 0 */
	int *vertics = (int*)calloc(n, sizeof(int));
	/* Вершины, из которых быстрее всего добраться */
	int *res = (int*)calloc(n, sizeof(int));
	/* Количество пройденных вершин */
	int size = 1;
	/* Массив с длинами ребер, из которых выбираем */
	int* arr = (int*)calloc(n, sizeof(int));
	/* Начальная вершина */
	int vertic = x;
	/* Делаем начальную вершину выбранной */
	vertics[vertic] = 1;
	for (int i = 0; i < n; i++) {
		arr[i] = adj[vertic][i];
		res[i] = vertic;
	}
	while (size != n) {
		int min = INT_MAX;
		int k;
		for (int i = 0; i < n; i++) {
			/* Пропускаем пройденные вершины */
			if (vertics[i] == 1) {
				continue;
			}
			/* Ищем вершину, до которой ближе всего (с номером k) */
			if (min > arr[i]) {
				min = arr[i];
				k = i;
			}
		}
		/* Отмечаем найденную вершину "пройденной" */
		vertics[k] = 1;
		/* Смотрим, изменится ли наш массив рёбер, если будем "шагать" из выбранной вершины */
		for (int i = 0; i < n; i++) {
			/* Игнорируем пройденные вершины */
			if (vertics[i] == 1) {
				continue;
			}
			/* Если из вершины k ближе до вершины i, то обновляем знaчение для i */
			if (arr[i] > min + adj[k][i]) {
				arr[i] = min + adj[k][i];
				res[i] = k;
			}
		}

		size++;
	}
	free(vertics);
	free(arr);
	return res;
}

void route(int **adj, int *res, int n, int x) {
	for (int i = 0; i < n; i++) {
		/* Игнорируем петли */
		if (i == x) {
			continue;
		}
		int w = i;
		/* Суммарная длина */
		int sum = 0;
		/* Если есть прямой короткий путь, то выводим его */
		if (res[i] == x) {
			sum = adj[x][i];
			printf("%d: %d-%d\n", i, i, x);
		}
		/* Иначе, последовательно идем по вершинам из res */
		else {
			printf("%d: %d-", i, i);
			while (res[w] != x) {
				sum += adj[res[w]][w];
				printf("%d-", res[w]);
				w = res[w];
			}
			sum += adj[x][w];
			printf("%d\n", x);
		}
		printf("Расстояние: %d\n###\n", sum);
	}
}