Advertisement
Guest User

Untitled

a guest
May 24th, 2019
106
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C 1.86 KB | None | 0 0
  1. #include <stdio.h>
  2.  
  3. //#define DEBUG
  4.  
  5. #ifdef DEBUG
  6. #define INPUT_FILE "C:\\Users\\Maksim Ignatyev\\CLionProjects\\AaDS\\cats\\taskR\\input.txt"
  7. #define OUTPUT_FILE "C:\\Users\\Maksim Ignatyev\\CLionProjects\\AaDS\\cats\\taskR\\output.txt"
  8. #else
  9. #define INPUT_FILE "input.txt"
  10. #define OUTPUT_FILE "output.txt"
  11. #endif
  12.  
  13. #define INF 11
  14.  
  15. int uv_to_i(int u, int v);
  16.  
  17. int graph[499500];
  18. signed char weights[499500];
  19.  
  20. int min_edge[1000];
  21. int out_edge[1000];
  22. char visited[1000];
  23.  
  24. int main() {
  25.  
  26.     FILE *input = fopen(INPUT_FILE, "r");
  27.     FILE *output = fopen(OUTPUT_FILE, "w+");
  28.  
  29.     int n, m;
  30.  
  31.     fscanf(input, "%d %d", &n, &m);
  32.  
  33.     for (int i = 0; i < 499500; i++) {
  34.         graph[i] = -1;
  35.     }
  36.  
  37.     for (int i = 0; i < 1000; i++) {
  38.         min_edge[i] = INF;
  39.         out_edge[i] = -1;
  40.         visited[i] = 0;
  41.     }
  42.  
  43.     for (int i = 0; i < m; i++) {
  44.         int u, v;
  45.         int w;
  46.         fscanf(input, "%d %d %d", &u, &v, &w);
  47.         graph[uv_to_i(u, v)] = i;
  48.         weights[i] = (signed char) w;
  49.     }
  50.  
  51.     fclose(input);
  52.  
  53.     min_edge[0] = 0;
  54.  
  55.     for (int i = 0; i < n; i++) {
  56.  
  57.         int v = -1;
  58.  
  59.         for (int j = 0; j < n; j++) {
  60.             if (visited[j] == 0 && (v == -1 || min_edge[j] < min_edge[v])) {
  61.                 v = j;
  62.             }
  63.         }
  64.  
  65.         visited[v] = 1;
  66.  
  67.         if(out_edge[v] != -1){
  68.             fprintf(output, "%d\n", graph[uv_to_i(v, out_edge[v])]);
  69.         }
  70.  
  71.         for (int j = 0; j < n; j++) {
  72.             if(graph[uv_to_i(i, j)] != -1 && weights[graph[uv_to_i(v, j)]] < min_edge[j]){
  73.                 min_edge[j] = weights[graph[uv_to_i(v, j)]];
  74.                 out_edge[j] = v;
  75.             }
  76.         }
  77.  
  78.     }
  79.  
  80.     fclose(output);
  81.  
  82. }
  83.  
  84. int uv_to_i(int u, int v) {
  85.     if (v < u) {
  86.         return (((u - 1) * u) / 2) + v;
  87.     }
  88.     return (((v - 1) * v) / 2) + u;
  89. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement