LeetCode——2285. Maximum Total Importance of Roads(C++)

最新推荐文章于 2024-06-12 10:00:23 发布

begforgood

最新推荐文章于 2024-06-12 10:00:23 发布

阅读量130

点赞数

分类专栏： Leetcode 笔记文章标签： leetcode c++

本文链接：https://blog.csdn.net/begforgood/article/details/125320827

版权

Leetcode 同时被 2 个专栏收录

11 篇文章 0 订阅

订阅专栏

笔记

10 篇文章 0 订阅

订阅专栏

LeetCode——2285. Maximum Total Importance of Roads(C++)
Description:
You are given an integer n denoting the number of cities in a country. The cities are numbered from 0 to n - 1.

You are also given a 2D integer array roads where roads[i] = [ai, bi] denotes that there exists a bidirectional road connecting cities ai and bi.

You need to assign each city with an integer value from 1 to n, where each value can only be used once. The importance of a road is then defined as the sum of the values of the two cities it connects.

Return the maximum total importance of all roads possible after assigning the values optimally.
Example 1:
在这里插入图片描述
Input: n = 5, roads = [[0,1],[1,2],[2,3],[0,2],[1,3],[2,4]]
Output: 43
Explanation: The figure above shows the country and the assigned values of [2,4,5,3,1].

The road (0,1) has an importance of 2 + 4 = 6.
The road (1,2) has an importance of 4 + 5 = 9.
The road (2,3) has an importance of 5 + 3 = 8.
The road (0,2) has an importance of 2 + 5 = 7.
The road (1,3) has an importance of 4 + 3 = 7.
The road (2,4) has an importance of 5 + 1 = 6.
The total importance of all roads is 6 + 9 + 8 + 7 + 7 + 6 = 43.
It can be shown that we cannot obtain a greater total importance than 43.
Example 2:

Input: n = 5, roads = [[0,3],[2,4],[1,3]]
Output: 20
Explanation: The figure above shows the country and the assigned values of [4,3,2,5,1].
The road (0,3) has an importance of 4 + 5 = 9.
The road (2,4) has an importance of 2 + 1 = 3.
The road (1,3) has an importance of 3 + 5 = 8.
The total importance of all roads is 9 + 3 + 8 = 20.
It can be shown that we cannot obtain a greater total importance than 20.

Constraints:
2 <= n <= 5 * 104
1 <= roads.length <= 5 * 104
roads[i].length == 2
0 <= ai, bi <= n - 1
ai != bi
There are no duplicate roads.

题目解析：
看似图题，但更像数组题，计算一下各个点的度，排序后乘以对应的系数，就可以了。

代码：

class Solution {
public:
    long long maximumImportance(int n, vector<vector<int>>& roads) {
        if(roads.size()==0)return 0;
        vector<long long>cnt;
        cnt.resize(n);
        for(int i=0;i<roads.size();i++)
        {
            int j,k;
            j=roads[i][0];
            k=roads[i][1];
            cnt[j]++;               //记录每个点的度；
            cnt[k]++;
        }
        sort(cnt.begin(),cnt.end());      
        long long sum=0;
        for(int i=n-1,nums=n;i>=0&&cnt[i]!=0;i--,nums--)   //到度为零的就不需要再继续遍历了
            sum+=cnt[i]*nums;
        return sum;
    }
};