La ruta más corta después de k puntos de tránsito + DFS (enumeración de matriz completa)

Descripción:
Dada una gráfica dirigida con n puntos ym aristas, hay k puntos de transferencia. Se requiere pasar todos los puntos marcados desde el punto de inicio especificado en cualquier orden para alcanzar el punto final especificado, y preguntar cuál es la distancia más corta.

Tome 3 puntos de transferencia como ejemplo, a saber, k1, k2, k3.
Hay 6 posibilidades:
① s-> k1-> k2-> k3-> e
② s-> k1-> k3-> k2-> e
…………

⑥ s -> k3 -> k2 -> k1 -> e

Podemos encontrar, por el camino de la ruta, que necesitamos ejecutar el algoritmo de ruta más corta a partir de los puntos de transferencia sy puntos de transferencia k respectivamente. Tenga en cuenta que debe guardar los resultados intermedios y utilizarlos más tarde.

Almacenar resultados intermedios ena[][]

	for (int i=1;i<=k;i++)
	 a[0][i]=dis[sp[i]];  //求起点到每个中转点的距离
	for (int i=1;i<=k;i++)  //以每一个标记点开始跑SPFA
	{
		spfa(sp[i]);
		for (int j=1;j<=k;j++)
	 	 a[i][j]=dis[sp[j]];
	 	a[i][k+1]=dis[t];  //一定要记得连终点!
	}

DFS enumera todos los posibles:

void dfs(int x,ll sum,int dep)  //DFS枚举顺序
{
	if (dep==k)   //分配好所有k个点的顺序
	{
		if (a[x][k+1]==Inf) return;
		if (ans>sum+(ll)a[x][k+1]) ans=sum+(ll)a[x][k+1];  //更新答案(第k+1个点是终点)
		return;
	}
	for (int i=1;i<=k;i++)  //枚举下一个到达的点
	 if (x!=i&&!p[i])
	 {
	 	if (a[x][i]==Inf) continue;
	 	p[i]=1;
	 	dfs(i,sum+(ll)a[x][i],dep+1);
	 	p[i]=0;
	 }
}

Código completo:
complejidad de tiempo general O ((m + n) log (n))

#include <cstdio>
#include <queue>
#include <cstring>
#include <iostream>
#define N 101000
#define Inf 1e9
#define ll_Inf 1e17
#define ll long long
using namespace std;

int n,m,k,s,t,tot,x,y,z,head[N],dis[N],sp[21],a[21][21];
bool vis[N],p[21];
ll ans=ll_Inf;

struct node
{
	int next,to,dis;
}e[N];

void add(int from,int to,int dis)  //存图(原图)
{
	tot++;
	e[tot].dis=dis;
	e[tot].to=to;
	e[tot].next=head[from];
	head[from]=tot;
}

void spfa(int start)  //以start为原点的SPFA
{
	for (int i=0;i<=n;i++)
	{
		dis[i]=Inf;
		vis[i]=0;
	}
	queue<int> q;
	q.push(start);
	vis[start]=1;
	dis[start]=0;
	while (q.size())
	{
		int u=q.front();
		q.pop();
		vis[u]=0;
		for (int i=head[u];~i;i=e[i].next)
		{
			int v=e[i].to;
			if (dis[v]>dis[u]+e[i].dis)
			{
				dis[v]=dis[u]+e[i].dis;
				if (!vis[v])
				{
					vis[v]=1;
					q.push(v);
				}
			}
		}
	}
}

void dfs(int x,ll sum,int dep)  //DFS枚举顺序
{
	if (dep==k)   //分配好所有k个点的顺序
	{
		if (a[x][k+1]==Inf) return;
		if (ans>sum+(ll)a[x][k+1]) ans=sum+(ll)a[x][k+1];  //更新答案(第k+1个点是终点)
		return;
	}
	for (int i=1;i<=k;i++)  //枚举下一个到达的点
	 if (x!=i&&!p[i])
	 {
	 	if (a[x][i]==Inf) continue;
	 	p[i]=1;
	 	dfs(i,sum+(ll)a[x][i],dep+1);
	 	p[i]=0;
	 }
}

int main()
{
	memset(head,-1,sizeof(head));
	scanf("%d%d%d%d%d",&n,&m,&k,&s,&t);
	for (int i=1;i<=m;i++)
	{
		scanf("%d%d%d",&x,&y,&z);
		add(x,y,z);
	}
	for (int i=1;i<=k;i++)
	 scanf("%d",&sp[i]);
	spfa(s);  //以s点为起点跑SPFA
	if (!k)  //k为0的特殊情况
	{
		if (dis[t]!=Inf) printf("%d\n",dis[t]);
		 else printf("-1");
		return 0;
	}
	for (int i=1;i<=k;i++)
	 a[0][i]=dis[sp[i]];  //求每个点的图
	for (int i=1;i<=k;i++)  //以每一个标记点开始跑SPFA
	{
		spfa(sp[i]);
		for (int j=1;j<=k;j++)
	 	 a[i][j]=dis[sp[j]];
	 	a[i][k+1]=dis[t];  //一定要记得连终点!!!
	}
	dfs(0,0,0);
	if (ans==ll_Inf) 
	{
		printf("-1");
		return 0;
	}
	cout<<ans;
	return 0;
}


Publicado 78 artículos originales · elogiado 0 · visitas 3416

Supongo que te gusta

Origin blog.csdn.net/qq_44846324/article/details/104781395
Recomendado
Clasificación