【Leetcode】2285. Maximum Total Importance of Roads

题目地址：

https://leetcode.com/problems/maximum-total-importance-of-roads/description/?envType=company&envId=hrt&favoriteSlug=hrt-all

给定$n$个点，标号$0\sim n-1$，每个点可以分配一个$1\sim n$的一个数字，每个数字只能分配一次。点之间有无向边，每条边的权值是其连接的两个点的分配数字之和。问怎么分配数字能使得所有边的总权值之和最大，返回最大的那个和。

先统计一下每个点连了多少条边，显然连最多边的点分配的数字应该最大。代码如下：

class Solution {
 public:
  using ll = long long;
  ll maximumImportance(int n, vector<vector<int>>& rs) {
    vector<int> cnt(n);
    for (auto& r : rs) cnt[r[0]]++, cnt[r[1]]++;
    sort(cnt.begin(), cnt.end());
    ll res = 0;
    for (int i = 0; i < n; i++) res += (ll)(i + 1) * cnt[i];
    return res;
  }
};

时间复杂度$O(n\log n)$，空间$O(n)$。