1. 程式人生 > >2018 Multi-University Training Contest 7 1001 Age of Moyu【SPFA】

2018 Multi-University Training Contest 7 1001 Age of Moyu【SPFA】

題意:給你一張圖,每條邊有一個代號。你可以花費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; }