16-InSange

[InSange]

🔗 문제 링크

Miaximum Total Importance of Roads

✔️ 소요된 시간

1시간

✨ 수도 코드

첫 번째 풀이

각 노드에 연결지점 roads를 통해서 각 노드 n개에 해당하는 2차원 배열을 만들어준다.
노드의 개수가 많으면 높은 점수를 차례로 부여한다. n, n-1, n-2 ... 0 이런식으로..

그리고 해당 간선의 합을 answer에 차근차근 저장해주면 되는데

계산한 노드를 unordered_map에다가 넣어서 두 노드가 이미 계산된 값이라면 계산을 수행하지 않는 식으로 진행하려다보니 필요 메모리와 코드가 너무 지저분해졌다.

사실 연결된 노드를 각각 따로 계산할 필요 없이 연결된 degree만큼 부여된 점수를 연산해주면 되는 것이었다.

두 번째 풀이

위에서 말한대로 차수와 해당 인덱스를 우선순위 큐를 통해 정렬 시켜주고 높은 차수(degree)를 지닌 노드를 하나씩 꺼내 점수를 매기고 degree만큼 곱해서 바로 해결해주는 식으로 바꾸었다.

class Solution {
public:
    long long maximumImportance(int n, vector<vector<int>>& roads) {
        vector<vector<int>> conn;   // road
        priority_queue<pair<long long, int>> pq;  // connect degree first: degree, second: index
        long long answer = 0;
        
        conn.assign(n, vector<int>());
        
        for(auto road : roads)
        {
            int v1 = road[0];
            int v2 = road[1];
            
            conn[v1].push_back(v2);
            conn[v2].push_back(v1);
        }
        
        for(int i = 0; i < n; i++)
        {
            pq.push({conn[i].size(), i});
        }
        
        while(!pq.empty())
        {
            pair<long long, int> cur;
            cur = pq.top();
            pq.pop();

            answer += (n-- * cur.first);
        }
        
        return answer;
    }
};

결과

후기

class Solution {
public:
    long long maximumImportance(int n, vector<vector<int>>& roads) {
        vector<int> degree(n, 0); // Array to store the degree of each city
        
        // Calculate the degree of each city
        for (const auto& road : roads) {
            degree[road[0]]++;
            degree[road[1]]++;
        }
        
        // Create a list of cities and sort by degree
        vector<int> cities(n);
        for (int i = 0; i < n; i++) {
            cities[i] = i;
        }
        sort(cities.begin(), cities.end(), [&](int a, int b) {
            return degree[a] > degree[b];
        });
        
        // Assign values to cities starting from the highest degree
        long long totalImportance = 0;
        for (int i = 0; i < n; i++) {
            totalImportance += (long long)(n - i) * degree[cities[i]];
        }
        
        return totalImportance;
    }
};

위와 같이 굳이 우선순위를 사용할 필요가 없다. 얼마나 연결되어있는지 차수만 알면 되기 때문!

📚 새롭게 알게된 내용

[yuyu0830]

첫 번째 풀이의 방법, 두 번째 풀이의 코드와 후기의 코드까지 점점 최적화 되어가는 부분을 보면 역시 문제를 읽으면서 해결 방법을 찾기 보다 문제를 완전 파악한 뒤에 솔루션을 찾아야 좋은 코드가 나오는 것 같아요!

[seongwon030]

차수를 이용해 바로 연산이 가능하도록 한게 좋았어요. 백준은 문제를 풀어야 다른 사람의 코드를 참고할 수 있는데 리트코드는 상관없는 것 같아서 다음에 한 번 써 봐야 겠습니다 ㅎㅎ