Submission #555957

#	Time	Username	Problem	Language	Result	Execution time	Memory
555957		aryan12	Road Closures (APIO21_roads)	C++17	5 / 100	87 ms	14244 KiB

roads

#include "roads.h"
#include <bits/stdc++.h>
using namespace std;

const long long MAXN = 1e5 + 5;
vector<array<long long, 2> > g[MAXN];

vector<long long> minimum_closure_costs(int N, vector<int> U, vector<int> V, vector<int> W) 
{
    long long sum = 0;
    vector<long long> ok;
    for(long long i = 0; i < U.size(); i++)
    {
        sum += W[i];
        ok.push_back(W[i]);
        // cout << "U[i] = " << U[i] << ", V[i] = " << V[i] << ", W[i] = " << W[i] << "\n";
        g[U[i]].push_back({V[i], W[i]});
        g[V[i]].push_back({U[i], W[i]});
    }
    sort(ok.begin(), ok.end());
    vector<long long> ans;
    ans.push_back(sum);
    for(long long i = 1; i < N; i++)
    {
        sum -= ok[ok.size() - i];
        ans.push_back(sum);
    }
    return ans;
}

Compilation message (stderr)

roads.cpp: In function 'std::vector<long long int> minimum_closure_costs(int, std::vector<int>, std::vector<int>, std::vector<int>)':
roads.cpp:12:28: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   12 |     for(long long i = 0; i < U.size(); i++)
      |                          ~~^~~~~~~~~~

• Subtask 1: 5 / 5
• Subtask 2: 0 / 7
• Subtask 3: 0 / 14
• Subtask 4: 0 / 10
• Subtask 5: 0 / 17
• Subtask 6: 0 / 25
• Subtask 7: 0 / 22