省选专练POI2007ATR

状压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;
}