제출 #745982

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
745982		vjudge1	Cities (BOI16_cities)	C++17	22 / 100	202 ms	3480 KiB

cities

#include <bits/stdc++.h>

using namespace std;

struct edge
{
    int dist,from,to;
    bool operator<(const edge x) const
    {
        return dist<x.dist;
    }
};

int n,k,m;
vector<int>parent;
vector<edge>edges;
vector<int>imp;

int holvan(int a) {
    return (parent[a]==a ? a : holvan(parent[a]));
}
void union_set(int a, int b) {

    parent[a]=b;
}
int main()
{
    cin>>n>>k>>m;
    edges.resize(m);
    imp.assign(k,0);
    for(int i=0;i<k;i++)
    {
        cin>>imp[i];
        imp[i]--;
    }
    for(int i=0;i<m;i++)
    {
        cin>>edges[i].from>>edges[i].to>>edges[i].dist;
        edges[i].from--;
        edges[i].to--;
    }
    long long mini=1e12;
    sort(edges.begin(),edges.end());
    for(int i=0;i<(1<<n);i++)
    {
        bool ok=true;
        for(int x:imp)
        {
            if(((1<<x)&i)==0)
            {
                ok=false;
            }
        }
        //cout << "proba " <<i << " " << ok <<endl;
        if(ok)
        {
            int sum_city=__builtin_popcount(i);
            int roads=0;
            long long sum=0;
            parent.assign(n,0);
            for(int i=0;i<n;i++)
            {
                parent[i]=i;
            }
            for(edge x:edges)
            {

                if(((1<<x.from)&i)!=0 && ((1<<x.to)&i)!=0)
                {
                    int a=x.from;
                    int b=x.to;
                    a=holvan(a), b=holvan(b);
                    //cout << a << " " << holvan(a) << " " << b << " " << holvan(b) << endl;
                    if(a!=b)
                    {
                        union_set(a, b);
                        //cout << "unio " << x.dist << "\n";
                        sum+=x.dist;
                        roads++;
                    }
                }
            }
            //cout << "mask: " << i << " " << sum << endl;
            if(sum_city==roads+1)
            {
                mini=min(mini,sum);
            }
        }
    }
    cout<<mini<<"\n";
    return 0;
}

/*

4 3 6
1 3 4
1 2 4
1 3 9
1 4 6
2 3 2
2 4 5
3 4 8

*/

• Subtask 1: 22 / 22
• Subtask 2: 0 / 14
• Subtask 3: 0 / 15
• Subtask 4: 0 / 23
• Subtask 5: 0 / 26