제출 #729702

#제출 시각아이디문제언어결과실행 시간메모리
729702Rafi22Cities (BOI16_cities)C++14
74 / 100
6099 ms172900 KiB
#include <bits/stdc++.h>

using namespace std;

#define endl '\n'
#define st first
#define nd second
#define pb push_back
#define sz(x) (int)(x).size()
#define all(x) (x).begin(), (x).end()
#define ll long long
ll mod=998244353;
int inf=1000000007;
ll infl=1000000000000000007;

const int N=100007,K=5;

ll DP[N][1<<K];
bool odw[N][1<<K];
vector<pair<int,ll>>G[N];

int main()
{
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    int n,k,m,a,b,c;
    cin>>n>>k>>m;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<(1<<k);j++) DP[i][j]=infl;
    }
    priority_queue<pair<ll,pair<int,int>>>Q;
    for(int i=1;i<=k;i++)
    {
        cin>>a;
        Q.push({0,{a,(1<<(i-1))}});
        DP[a][(1<<(i-1))]=0;
    }
    for(int i=0;i<m;i++)
    {
        cin>>a>>b>>c;
        G[a].pb({b,c});
        G[b].pb({a,c});
    }
    while(sz(Q)>0)
    {
        ll t=-Q.top().st;
        int v=Q.top().nd.st,x=Q.top().nd.nd;
        Q.pop();
        if(odw[v][x]) continue;
       // cout<<v<<" "<<x<<" "<<t<<endl;
        odw[v][x]=1;
        for(auto u:G[v])
        {
            int d=(1<<k)-1-x;
            for(int y=d;;y=(y-1)&d)
            {
                if(t+u.nd+DP[u.st][y]<DP[u.st][x^y])
                {
                    DP[u.st][x^y]=t+u.nd+DP[u.st][y];
                    Q.push({-DP[u.st][x^y],{u.st,x^y}});
                }
                if(y==0) break;
            }
        }
    }
    ll ans=infl;
    for(int i=1;i<=n;i++) ans=min(ans,DP[i][(1<<k)-1]);
    cout<<ans;
    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...