HDU 6166 Senior Pan（Dijkstra）

Description

给出一个$n$个点$m$条边的有向图，边有边权，给出一个点集有$k$个点，问这个点集内的点两两之间最短路的最小值

Input

第一行一整数$T$表示用例组数，每组用例首先输入两个整数$n,m$表示点数和边数，之后$m$行每行输入三个整数$u_i,v_i,c_i$表示$u_i$到$v_i$有一条边权为$c_i$的有向边，然后输入一整数$k$表示所选点集的点数，最后输入$k$个整数$a_1,...,a_k$表示这些点的编号$(1\le T\le 5,1\le n,m\le 10^5,1\le c_i\le 10^5)$

Output

输出这$k$个点两两最短距离的最小值

Sample Input

1
5 6
1 2 1
2 3 3
3 1 3
2 5 1
2 4 2
4 3 1
3
1 3 5

Sample Output

Case #1: 2

Solution

对$k$个点中每个点开始求单源最短路不现实，最终最短距离最小的两点，其二进制表示中必然至少有一位不同，考虑按位把点分成两块，第$i$位为$0$的一块，第$i$位为$1$的一块，对于每个$i$，我们可以分别以两块中的一块作为起点去求另一块到这一块的最短路，找出最短路中的最小值，对于每个$i$去更新这个最小值即得到答案，这样只需要求$2log_2n$遍最短路

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define INF 1e12
#define maxn 100005
typedef pair<int,ll>P;
struct edge
{
    int to,cost;
};
vector<edge>g[maxn];
ll dis[maxn];
void add(int u,int v,int c)
{
    g[u].push_back((edge){v,c});
}
priority_queue< P,vector<P>,greater<P> >que;
void Dijkstra()//求所有顶点到起点s的最短距离
{
    while(!que.empty())
    {
        P p=que.top();
        que.pop();
        int v=p.second;
        if(dis[v]<p.first) continue;//剪枝
        for(int i=0;i<g[v].size();i++)
        {
            edge e=g[v][i];
            if(dis[e.to]>dis[v]+e.cost)
            {
                dis[e.to]=dis[v]+e.cost;
                que.push(P(dis[e.to],e.to));
            }
        }
    }
}
int T,n,m,k,a[maxn],Case=1;
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)g[i].clear();
        while(m--)
        {
            int u,v,c;
            scanf("%d%d%d",&u,&v,&c);
            add(u,v,c);
        } 
        scanf("%d",&k);
        for(int i=1;i<=k;i++)scanf("%d",&a[i]);
        ll ans=INF;
        for(int i=0;i<20;i++)
        {
            for(int i=1;i<=n;i++)dis[i]=INF;
            for(int j=1;j<=k;j++)
                if(a[j]&(1<<i))dis[a[j]]=0,que.push(P(0,a[j]));
            Dijkstra();
            for(int j=1;j<=k;j++)
                if((a[j]&(1<<i))==0)ans=min(ans,dis[a[j]]);
            for(int i=1;i<=n;i++)dis[i]=INF;
            for(int j=1;j<=k;j++)
                if((a[j]&(1<<i))==0)dis[a[j]]=0,que.push(P(0,a[j]));
            Dijkstra();
            for(int j=1;j<=k;j++)
                if(a[j]&(1<<i))ans=min(ans,dis[a[j]]);      
        }
        printf("Case #%d: %I64d\n",Case++,ans);
    }
}