aboutsummaryrefslogtreecommitdiff
path: root/max_total_importance_road.c
blob: 145c3e2486dfae3b3b07861317498b0efd85eeec (plain)
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
// 2285. Maximum Total Importance of Roads
#include <stdio.h>
#include <stdlib.h>

int cmp(const void *a, const void *b) { return (*(int *)a - *(int *)b); }

long long maximumImportance(int n, int **roads, int roadsSize,
                            int *roadsColSize) {
  long long ans = 0, *timeShown = calloc(n, sizeof(long long));
  for (int i = 0; i < roadsSize; i++)
    for (int j = 0; j < 2; j++)
      timeShown[roads[i][j]]++;
  qsort(timeShown, n, sizeof(long long), cmp);
  for (int i = n - 1; i >= 0; i--)
    ans += timeShown[i] * (i + 1);
  free(timeShown);
  return ans;
}

int main() {
  int roads1[][2] = {{0, 1}, {1, 2}, {2, 3}, {0, 2}, {1, 3}, {2, 4}},
      roadsColSize1[] = {6};
  int roads2[][2] = {{0, 3}, {2, 4}, {1, 3}}, roadsColSize2[] = {3};
  printf("%llu\n", maximumImportance(5, roads1, 6, roadsColSize1)); // expect:
                                                                    // 43
  printf("%llu\n", maximumImportance(5, roads1, 3, roadsColSize1)); // expect:
                                                                    // 20
  return 0;
}