2018 Multi-University Training Contest 7 1001 Age of Moyu【SPFA】
阿新 • • 發佈:2018-12-24
題意:給你一張圖,每條邊有一個代號。你可以花費1的代價來收買當前代號來通過這條邊,如果相鄰的邊屬於同一個代號走第二條邊不需要花費額外代價(如果經過了其他代號的邊再經過已收買的邊需要重新花費1代價)。
分析:那麼其實相比於常規的最短路只多了一個pre的判斷,一個點可能有很多了pre過來並且總花費相同,那麼這些pre都需要考(需要儲存)。
#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <set>
#include <queue>
#define LL long long
using namespace std;
const int N = 100000+33;
const int M = 200000+33;
const int inf = 0x3f3f3f3f;
int head[N],nxt[M<<1],to[M<<1],W[M<<1],cnt=0;
void add(int u,int v,int w)
{
nxt[cnt]=head[u];
to[cnt]=v;
W[cnt]=w;
head[u]=cnt++;
}
int n,m;
set<int> p[N];
int vis[N];
int dis[N];
int spfa()
{
queue<int> q;
q.push(1);
for(int i=1;i<=n;i++){
dis[i]=inf;
vis[i]=0;
}
dis[1]=0;
p[1].clear();
while(!q.empty()){
int u = q.front();
q.pop();
vis[u]=0;
for(int i=head[u];~i;i=nxt[i]){
int v=to[i];
int flag=p[u].find(W[i])!=p[u].end();
if(flag){
if(dis[v]>dis[u]){
dis[v]=dis[u];
p[v].clear();
p[v].insert(W[i]);
if(!vis[v])
q.push(v),vis[v]=1;
}else if(dis[v]==dis[u]){
if(p[v].find(W[i])==p[v].end()){
p[v].insert(W[i]);
if(!vis[v])
q.push(v),vis[v]=1;
}
}
}else{
if(dis[v]>dis[u]+1){
dis[v]=dis[u]+1;
p[v].clear();
p[v].insert(W[i]);
if(!vis[v])
q.push(v),vis[v]=1;
}else if(dis[v]==dis[u]+1){
p[v].insert(W[i]);
if(!vis[v])
q.push(v),vis[v]=1;
}
}
}
}
return dis[n];
}
int main()
{
while(~scanf("%d%d",&n,&m))
{
for(int i=1;i<=n;i++) head[i]=-1;
cnt=0;
for(int i=1;i<=m; i++)
{
int l,r,w;
scanf("%d%d%d",&l,&r,&w);
add(l,r,w);
add(r,l,w);
}
int ans=spfa();
if(ans==inf)
printf("-1\n");
else
printf("%d\n",ans);
}
return 0;
}