Cod sursa(job #2166108)

Utilizator zhm248Mustatea Radu zhm248 Data 13 martie 2018 15:35:26
Problema Ubuntzei Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.22 kb
#include<cstdio>
#include<queue>
#include<vector>
#include<cstring>
using namespace std;
int put2(int x)
{
    int nr=1;
    for(int i=1;i<=x;++i)
    {
        nr*=2;
    }
    return nr;
}

bool det(int x,int poz)
{
    for(int i=0;i<poz;++i)
    {
        x-=(x%2);
        x/=2;
    }
    if(x%2)
        return 1;
    return 0;
}

typedef pair<int,int>ii;
typedef pair<int,ii>iii;
int city[16],pcity[16];
vector<ii>g[2001];
int dist[2001][32768];
priority_queue<iii,vector<iii>,greater<iii> >pq;
int main()
{
    freopen("ubuntzei.in","r",stdin);
    freopen("ubuntzei.out","w",stdout);
    int n,m,k,i,x,y,z;
    scanf("%d%d%d",&n,&m,&k);
    for(i=1;i<=k;++i)
    {
        scanf("%d",&city[i]);
        pcity[city[i]]=i;
    }
    for(i=1;i<=m;++i)
    {
        scanf("%d%d%d",&x,&y,&z);
        g[x].push_back(ii(z,y));
        g[y].push_back(ii(z,x));
    }
    memset(dist,0x3f3f3f,sizeof(dist));
    dist[1][0]=0;
    pq.push(iii(0,ii(1,0)));
    while(!pq.empty())
    {
        iii top=pq.top();
        pq.pop();
        int d=top.first,val1=top.second.first,val2=top.second.second;
        if(d==dist[val1][val2])
        {
            for(i=0;i<g[val1].size();++i)
            {
                int v=g[val1][i].second;
                int dv=g[val1][i].first;
                if(pcity[v])
                {
                    if(!det(val2,pcity[v]-1))
                    {
                        int val3=val2+put2(pcity[v]-1);
                        if(dist[v][val3]>dist[val1][val2]+dv)
                        {
                            dist[v][val3]=dist[val1][val2]+dv;
                            pq.push(iii(dist[v][val3],ii(v,val3)));
                        }
                    }
                }
                else
                {
                    if(dist[v][val2]>dist[val1][val2]+dv)
                    {
                        dist[v][val2]=dist[val1][val2]+dv;
                        pq.push(iii(dist[v][val2],ii(v,val2)));
                    }
                }
            }
        }
    }
    int kk=1,sum=1;
    for(i=1;i<k;++i)
    {
        kk*=2;
        sum+=kk;
    }
    printf("%d\n",dist[n][sum]);
    return 0;
}