제출 #1302399

#제출 시각아이디문제언어결과실행 시간메모리
1302399goldencheemsCities (BOI16_cities)C++20
14 / 100
221 ms18732 KiB
/*DEP TRAI CO J SAI*/
#include <bits/stdc++.h>
using namespace std;
#define fi first
#define se second
#define pu emplace_back
#define ll long long
typedef pair <ll,ll> ii;
const int N=2e5;
long long mod=1e17;
ll n,m,k,y;
ll a[N+1],b[N+1],d[10][N+10],dp[(1<<6)];
void add(int &a,int b){
    a+=b;
    if(a>=mod) a-=mod;
    }
vector <ii> adj[N+10];
void distra(int s){
        int z=a[s];
        for(int i=1;i<=n;i++) d[s][i]=mod;
        d[s][z]=0;
        priority_queue <ii,vector<ii>,greater<ii> > pq;
        pq.push({0,z});
        while(pq.size()){
            ll du=pq.top().fi,u=pq.top().se;
            pq.pop();
            if(du!=d[s][u]) continue;
            for(auto x:adj[u]){
                ll dv=x.fi+du,v=x.se;
                if(dv<d[s][v]){
                    d[s][v]=dv;
                    pq.push({dv,v});
                }
            }
        }
        }
namespace sub1{
    void solve(){
        ll ans=mod;
        for(int i=1;i<=n;i++){
            ll res=0;
            for(int j=1;j<=k;j++) {
                //if(j==i) continue;
                res+=d[j][i];
            }
            ans=min(ans,res);
        }
        cout<<ans;
        }
    }

void tinh(){
    cin>>n>>k>>m;
    for(int i=1;i<=k;i++) cin>>a[i];
    for(int i=1;i<=m;i++){
        int u,v,w;
        cin>>u>>v>>w;
        adj[u].pu(w,v);
        adj[v].pu(w,u);
    }
    for(int i=1;i<=k;i++){
        distra(i);
    }
    if(k<=3){
        sub1::solve();
        return;
    }
    ll ans=mod;
    for(int i=1;i<=k;i++){
        for(int j=1;j<(1<<k);j++) dp[j]=mod;
        dp[0]=0;
        for(int j=1;j<=n;j++){
            for(int x=1;x<(1<<k);x++){
                if(x&(1<<(i-1))) continue;
                ll res=d[i][j];
                for(int z=1;z<=k;z++){
                    if(x&(1<<(z-1))) {
                        res+=d[z][j];
                    }
                }
                dp[x]=min(dp[x],res);
            }
        }
        for(int x=1;x<(1<<k);x++){
            if(x&(1<<(i-1))) continue;
            for(int j=x;j>0;j=((j-1)&x)){
                int z=(x^j);
                dp[x]=min(dp[x],dp[z]+dp[j]);
            }
        }
        int z=(1<<k)-1;
        z^=(1<<(i-1));
        ans=min(ans,dp[z]);
    }
    cout<<ans;
    }
int main(){

    ios_base::sync_with_stdio(false);
    //freopen("cities.inp","r",stdin);
    //freopen("cities.out","w",stdout);
    cin.tie(0);
    cout.tie(0);
    tinh();
    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...