状压DP
我们先跑k次dijkstra
记录下全源的最短两两路径。
那么,我们状态压缩,考虑是否可达。
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<queue>
using namespace std;
const int N=2e5+10;
const int INF=1e9+10;
int f[1<<21][22]={0};
int dist[22][22]={0};
int a[22]={0};
int len[25]={0};
int mi[25]={0};
struct Front_star{
int u,v,w,nxt;
}e[N*4];
int cnt=0;
int first[N]={0};
void add(int u,int v,int w){
cnt++;
e[cnt].u=u;
e[cnt].v=v;
e[cnt].w=w;
e[cnt].nxt=first[u];
first[u]=cnt;
}
//
int inqueue[N]={0};
int dis[N]={0};
int n,m,k;
struct Node{
int u,w;
};
bool operator <(Node A,Node B){
return A.w>B.w;
}
priority_queue<Node> q;
void dijkstra(int start){
q.push((Node){start,0});
memset(dis,0x3f,sizeof(dis));
dis[start]=0;
while(!q.empty()){
Node tmp=q.top();
int x=tmp.u;
q.pop();
for(int i=first[x];i;i=e[i].nxt){
int v=e[i].v;
if(dis[x]+e[i].w<dis[v]){
dis[v]=dis[x]+e[i].w;
q.push((Node){v,tmp.w+e[i].w});
}
}
}
len[start]=dis[n];
for(int i=1;i<=k+1;i++){
for(int j=1;j<=k+1;j++){
if(i==j)
continue;
dist[start][i]=dis[i];
}
}
}
int main(){
memset(dist,0x3f,sizeof(dist));
scanf("%d%d%d",&n,&m,&k);
for(int i=1;i<=m;i++){
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
add(u,v,w);
add(v,u,w);
}
for(int i=1;i<=k+1;i++){
dijkstra(i);
}
mi[0]=1;
for(int i=1;i<=30;i++){
mi[i]=mi[i-1]<<1;
}
int Q;
scanf("%d",&Q);
while(Q--){
int u,v;
scanf("%d%d",&u,&v);
a[v]+=mi[u-2];
}
memset(f,-1,sizeof(f));
// for(int i=1;i<=k+1;i++){
// for(int j=1;j<=k+1;j++){
// cout<<dist[i][j]<<" ";
// }
// cout<<endl;
// }
int towards=mi[k]-1;
f[0][1]=0;
for(int i=0;i<=towards;i++){
for(int j=1;j<=k+1;j++){
if(f[i][j]!=-1){
for(int ii=2;ii<=k+1;ii++){
if(ii==j)
continue;
if((i&a[ii])==a[ii]){
if(f[i|(mi[ii-2])][ii]==-1||f[i|(mi[ii-2])][ii]>f[i][j]+dist[j][ii]){
f[i|(mi[ii-2])][ii]=f[i][j]+dist[j][ii];
}
}
}
}
}
}
int ans=INF;
for(int i=1;i<=k+1;i++){
if(f[towards][i]!=-1)
ans=min(ans,len[i]+f[towards][i]);
}
// cout<<towards<<endl;
// cout<<f[towards][1]<<endl;
cout<<ans;
}