题意
给出一张有向图,求其中 K K K 个点之间的最短路
显然求 K K K 次最短路会超时,可以换个角度思考。把 K K K个点分成两个集合,同一集合内的点互相距离为0,最短路也就是这两个点集之间的最短路(最短路的两个端点不在同一个集合内)。
现在只需要考虑这个集合需要分多少次。从二进制的角度看,把集合 K K K 中第 i i i 位为 0 和为 1 的分为两个集合,这样对于无论最短路端点是哪两个,从第1位拆到第 l o g ( n ) + 1 log(n)+1 log(n)+1位,一定有一次会把他们拆到不同的集合。
每次对于两个集合求最短路就行了
#include<iostream>
#include<sstream>
#include<string>
#include<queue>
#include<map>
#include<unordered_map>
#include<set>
#include<vector>
#include<stack>
#include <utility>
#include<list>
#include<bitset>
#include<algorithm>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<iomanip>
#include<time.h>
#include<random>
using namespace std;
#define PI acos(-1.0)
#define eps 1e-9
#define lowbit(a) ((a)&-(a))
#define mid ((l+r)>>1)
#define mem(x,y) memset(x,y,sizeof x)
const int mod = 1e9+7;
int qpow(int a,int b){
int ans=1;
while(b){
if(b&1)ans=(ans*a)%mod;
a=(a*a)%mod;
b>>=1;
}
return ans;
}
const int INF = 0x3f3f3f3f;
const int N = 1e5+10;
struct node{
int w,to;
bool friend operator<(const node x,const node y){
return x.w>y.w;
}
};
vector<node>g[N];
int n,m,k;
int vis[N],dis[N],a[N],in[N];
priority_queue<node>q;
void init(){
for(int i=1;i<=n;i++)dis[i]=INF;
mem(vis,0),mem(in,0); while(!q.empty())q.pop();
}
int dij(){
int ans=INF;
while(!q.empty()){
node f=q.top(); q.pop();
int u=f.to;
if(vis[u])continue; vis[u]=1;
for(auto t:g[u]){
int v=t.to;
if(!vis[v]&&dis[v]>dis[u]+t.w){
dis[v]=dis[u]+t.w;
q.push({dis[v],v});
}
}
}
for(int i=1;i<=k;i++)if(!in[i])ans=min(ans,dis[a[i]]);
return ans;
}
#define endl '\n'
signed main(){
std::ios::sync_with_stdio(false);
cin.tie(0),cout.tie(0);
int t,cs=0; scanf("%d",&t);
while(t--){
int ans=INF;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)g[i].clear();
for(int i=1;i<=m;i++){
int u,v,w; scanf("%d%d%d",&u,&v,&w);
g[u].push_back({w,v});
}
scanf("%d",&k);
for(int i=1;i<=k;i++)scanf("%d",&a[i]);
for(int c=0;c<17;c++){
init();
for(int i=1;i<=k;i++)
if(a[i]&(1<<c)) q.push({0,a[i]}),dis[a[i]]=0,in[i]=1;
ans=min(ans,dij());
init();
for(int i=1;i<=k;i++)
if(!(a[i]&(1<<c))) q.push({0,a[i]}),dis[a[i]]=0,in[i]=1;
ans=min(ans,dij());
}
printf("Case #%d: %d\n",++cs,ans);
}
}