【BZOJ 1097】 [POI2007]旅游景点atr

由于给出的第二个条件所以一定是要知道k+1个点之间的相互最短距离，但是Floyd肯定是没法搞的，观察k很小，直接暴力跑k次dijkstra就好了。然后设f[i][pos]表示到达i节点，经过节点状态为pos的最短距离，枚举状态然后枚举点转移就好了。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>
#define LL long long
#define inf 0x3f3f3f3f
#define maxn 20021
#define maxm 200021
#define MK make_pair
using namespace std;
typedef pair<int,int> pii;
int n,m,k,d[maxn],head[maxn],tot=1,f[22][1048576],vis[maxn],dis[23][23],pre[23];
void read(int& x){
	x=0;char c=getchar();int f=1;
	for(;c>'9'||c<'0';c=getchar());if(c=='-')f=-1;
	for(;c>='0'&&c<='9';c=getchar())x=x*10+c-'0';x*=f;
}
struct edge{int v,w,next;}e[maxm*2];
void adde(int a,int b,int c){e[tot].v=b,e[tot].w=c,e[tot].next=head[a];head[a]=tot++;}

void dijkstra(int x){
	priority_queue<pii,vector<pii>,greater<pii> >q;
	for(int i=0;i<=n;i++)d[i]=inf,vis[i]=0;
	d[x]=0;
	q.push(MK(0,x));
	while(!q.empty()){
		int u=q.top().second;q.pop();
		if(vis[u])continue;vis[u]=1;
		for(int v,i=head[u];i;i=e[i].next){
			if(d[u]+e[i].w<d[v=e[i].v]){
				d[v]=d[u]+e[i].w;
				q.push(MK(d[v],v));
			}
		}
	}
	for(int i=1;i<=k+1;i++)dis[x][i]=d[i];
	dis[x][0]=d[n];
}

void solve(){
	f[1][0]=0;int t=(1<<k)-1;
	for(int i=0;i<=t;i++){
		for(int a=1;a<=k+1;a++)if(f[a][i]!=-1){
			for(int b=1;b<=k+1;b++)if((i&pre[b])==pre[b]){
				if(f[b][i|(1<<b-2)]==-1||f[a][i]+dis[a][b]<f[b][i|(1<<b-2)])
					f[b][i|(1<<b-2)]=f[a][i]+dis[a][b];
			}
		}
	}
	int ans=inf;
	for(int i=1;i<=k+1;i++)
		if(f[i][t]!=-1)ans=min(ans,f[i][t]+dis[i][0]);
	printf("%d",ans);
}

int main(){
	memset(f,-1,sizeof(f));
	read(n),read(m),read(k);
	for(int a,b,c,i=1;i<=m;i++){
		read(a),read(b),read(c);
		adde(a,b,c),adde(b,a,c);
	}
	for(int i=1;i<=k+1;i++)dijkstra(i);
	int q;read(q);
	for(int a,b,i=1;i<=q;i++){
		read(a),read(b);
		pre[b]|=1<<a-2;
	}
	solve();
	return 0;
}