题意:
给你一个n(n<=200)个点,m(m<=n*n)的有向图,然后每条边都有一个颜色,然后要求求出1->n的一条最短路径,满足相邻的两条边不能是同一种颜色。输出最短路径长度。
tip:
bfs vis多加一维来的颜色。。
#include <cstdio>
#include <iostream>
#include <cstring>
#include <queue>
using namespace std;
const int maxn = 210;
int tot,head[maxn],vis[maxn][5];
struct node{
int v,next,c;
}edges[maxn*maxn];
void add(int u,int v,int c){
edges[tot].v = v;edges[tot].c = c;edges[tot].next = head[u];head[u] = tot++;
}
int u,v,c,n,m;
void init(){
tot = 0;
for(int i = 1; i <= n ; i++)
head[i] = -1;
for(int i = 1; i <= m; i++){
scanf("%d%d%d",&u,&v,&c);
add(u,v,c);
}
}
typedef pair<int,int>pii;
int step[maxn][5];
void bfs(){
queue<pii> q;
memset(vis,0,sizeof(vis));
q.push(make_pair(1,0));vis[1][0] = 1;step[1][0] = 0;
while(!q.empty()){
pii tmp = q.front();q.pop();
if(tmp.first == n){
printf("%d\n",step[tmp.first][tmp.second]);
return ;
}
for(int k = head[tmp.first] ; k != -1 ; k = edges[k].next){
int v = edges[k].v,c = edges[k].c;
if(!vis[v][c] && c != tmp.second){
vis[v][c] = 1;
q.push(make_pair(v,c));
step[v][c] = step[tmp.first][tmp.second]+1;
}
}
}
printf("-1\n");
}
int main(){
while(~scanf("%d%d",&n,&m)){
init();
bfs();
}
}