题意: 给你一个图(无重边且双向),每条边都有长度和颜色,要求不能连续走两条颜色相同的边。给你起点和终点,求最短路径。
解法:用SPFA解决,不过队列里的元素不是一个节点,而是代表从起点到某一节点以某一种颜色结束的二位元素。
#include <iostream>
#include <stdio.h>
#include <string>
#include <algorithm>
#include <math.h>
#include <string.h>
#include <stdlib.h>
#include <map>
#include <queue>
#include <stack>
#include <vector>
using namespace std;
const int N=60;
const int M=4100;
int head[N];
const int Max=10000007;
struct node
{
int v,len,c,next;
} edge[M];
int n,m,k,p;
int cnt;
int dis[N][N],vis[N][N];
struct Node
{
int v,last;
Node() {}
Node(int a,int b):v(a),last(b) {}
};
void init()
{
cnt=0;
memset(head,-1,sizeof(head));
}
void add(int u,int v,int len,int c)
{
edge[cnt].v=v;
edge[cnt].len=len;
edge[cnt].c=c;
edge[cnt].next=head[u];
head[u]=cnt++;
}
int SPFA(int s,int f)
{
memset(vis,0,sizeof(vis));
for(int i=0; i<=N; i++)
for(int j=0; j<=N; j++)
dis[i][j]=Max;
queue<Node > q;
for(int i=1; i<=k; i++)
dis[s][i]=0,vis[s][i]=1;
q.push(Node(s,0));
Node tmp;
while(!q.empty())
{
tmp=q.front();
q.pop();
int u=tmp.v;
vis[u][tmp.last]=0;
for(int i=head[u]; i!=-1; i=edge[i].next)
{
int v=edge[i].v;
for(int j=1; j<=k; j++)
{
if(u==s||j!=edge[i].c)// 不加u==s会WA
{
if(dis[u][j]+edge[i].len<dis[v][edge[i].c])
{
dis[v][edge[i].c]=dis[u][j]+edge[i].len;
if(!vis[v][edge[i].c])
{
vis[v][edge[i].c]=1;
q.push(Node(v,edge[i].c));
}
}
}
}
}
}
int Min=Max;
for(int i=1; i<=k; i++)
{
if(dis[f][i]<Min)
{
Min=dis[f][i];
}
}
return Min;
}
void solve()
{
scanf("%d%d%d%d",&n,&m,&k,&p);
init();
int x,y,len,c;
while(m--)
{
scanf("%d%d%d%d",&x,&y,&len,&c);
add(x,y,len,c);
add(y,x,len,c);
}
while(p--)
{
scanf("%d%d",&x,&y);
int ans=SPFA(x,y);
if(ans>=Max)
puts("infinity");
else
printf("%d\n",ans);
}
}
int main()
{
int cas;
scanf("%d",&cas);
while(cas--)
{
solve();
}
return 0;
}