Submission #1356984

#TimeUsernameProblemLanguageResultExecution timeMemory
1356984alexddSecurity Guard (JOI23_guard)C++20
37 / 100
97 ms32948 KiB
#include <bits/stdc++.h>
using namespace std;
#define int long long
int n,m,q;
vector<int> con[200005], arb[200005];
int s[200005];

namespace DSU
{
    int father[200005], siz[200005], strongest[200005];
    void init()
    {
        for(int i=1;i<=n;i++)
        {
            father[i] = i;
            siz[i] = 1;
            strongest[i] = i;
        }
    }
    int get(int x)
    {
        if(father[x] != x)
            father[x] = get(father[x]);
        return father[x];
    }
    void unite(int x, int y)
    {
        x = get(x);
        y = get(y);
        if(x == y)
            return;
        if(siz[x] < siz[y])
            swap(x,y);
        siz[x] += siz[y];
        father[y] = x;
        if(s[strongest[y]] > s[strongest[x]])
            strongest[x] = strongest[y];
    }
}

signed main()
{
    ios_base::sync_with_stdio(0);cin.tie(0);
    cin>>n>>m>>q;
    for(int i=1;i<=n;i++)
        cin>>s[i];

    int rez = 0;
    int u,v;
    for(int i=1;i<=m;i++)
    {
        cin>>u>>v;
        con[u].push_back(v);
        con[v].push_back(u);
        rez += min(s[u], s[v]);
    }

    vector<pair<int,int>> ord;
    for(int i=1;i<=n;i++)
        ord.push_back({s[i], i});
    sort(ord.begin(), ord.end());

    vector<bool> done(n+2,0);
    DSU::init();
    for(auto [_,i]:ord)
    {
        for(int x:con[i])
        {
            if(!done[x])
                continue;
            if(DSU::get(x) != DSU::get(i))
            {
                int u = DSU::strongest[DSU::get(x)];
                rez += s[i] - s[u];
                arb[i].push_back(u);
                DSU::unite(x, i);
            }
        }
        done[i] = 1;
    }

    cout<<rez;

    return 0;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...