题意:
连续相同类型的边花费为1,走不同的边花费加一,求最少花费
BFS:维护当前最少花费,每次往外扩展一步即可
最短路:set记录到达改点的边的类型都有哪些(距离相同),如果出现更小的则清空,否则若相同,加入集合。 在进行距离更新时判断一下set里有没有当前类型的边即可
BFS
#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5 + 10;;
const int INF = 0x3f3f3f3f;
int n,m;
int head[maxn],tot;
queue<int> que;
int dis[maxn];
bool vis[maxn];
struct Edge
{
int to,id,next;
bool vis;
} E[maxn*4];
inline void add(int from,int to,int id)
{
E[++tot].next = head[from];
E[tot].to = to;
E[tot].id = id;
E[tot].vis = 0;
head[from] = tot;
E[++tot].next = head[to];
E[tot].id = id;
E[tot].to = from;
E[tot].vis = 0;
head[to] = tot;
}
inline void init()
{
tot = 0;
memset(head,0,sizeof head);
}
void dfs(int x,int id,int num)
{
if(!vis[x])
{
dis[x] = num;
vis[x] = true;
if(x == n)return;
que.push(x);
}
for(int i=head[x] ; i ; i=E[i].next)
{
if(E[i].vis)continue;
if(E[i].id == id)
{
E[i].vis = true;
dfs(E[i].to,id,num);
}
}
}
void bfs()
{
while(!que.empty())que.pop();
memset(vis,false,sizeof vis);
dis[1] = 0;
dis[n] = -1;
vis[1] = true;
que.push(1);
while(!que.empty())
{
int t = que.front();
que.pop();
for(int i=head[t] ; i ; i=E[i].next)
{
if(E[i].vis)continue;
E[i].vis = true;
dfs(E[i].to,E[i].id,dis[t]+1);
if(dis[n] > 0)break;
}
if(dis[n] > 0)break;
}
}
int main()
{
while(scanf("%d %d",&n,&m) == 2)
{
init();
if(m == 0)
{
printf("-1\n");
continue;
}
int a,b,c;
for(int i=1 ; i<=m ; ++i)
{
scanf("%d %d %d",&a,&b,&c);
add(a,b,c);
}
bfs();
printf("%d\n",dis[n]);
}
return 0;
}
最短路:
#include <cstdio>
#include <cstring>
#include <queue>
#include <set>
using namespace std;
#define INF 0x3f3f3f3f
struct edge{
int to;
int id;
int next;
}a[400020];
int head[200020];
int n,m,q;
int vis[200020];
struct nodope{
int val;
set<int> d;
}d[200020];
inline int spfa()
{
memset(vis,0,sizeof vis); vis[1]=1;
queue<int> q;
q.push(1);
for(int i=0;i<=n;i++) d[i].val=INF,d[i].d.clear();
d[1].val=0;
while(!q.empty())
{
int u=q.front();q.pop();vis[u]=0;
for(int i=head[u];i!=-1;i=a[i].next)
{
int to=a[i].to;
if(d[to].val>(d[u].val+(d[u].d.find(a[i].id)==d[u].d.end())))
{
d[to].val=d[u].val+(d[u].d.find(a[i].id)==d[u].d.end());
d[to].d.clear();d[to].d.insert(a[i].id);
if(!vis[to])
{
q.push(to);vis[to]=1;
}
}
else if(d[to].val==d[u].val&&d[to].d.find(a[i].id)==d[to].d.end())
{
d[to].d.insert(a[i].id);
if(!vis[to])
{
q.push(to);vis[to]=1;
}
}
}
}
if(d[n].val!=INF)
printf("%d\n",d[n].val);
else printf("-1\n");
return 0;
}
int nn;
inline void add(int x,int y,int z)
{
a[nn].id=z; a[nn].to=y;a[nn].next=head[x];head[x]=nn++;
a[nn].id=z; a[nn].to=x;a[nn].next=head[y];head[y]=nn++;
}
int main()
{
while(~scanf("%d%d",&n,&m))
{
for(int i=0;i<=n;i++)
head[i]=-1;
nn=0;
for(int i=0;i<m;i++)
{
int x,y,z;
scanf("%d%d%d",&x,&y,&z);add(x,y,z);
}
spfa();
}
return 0;
}