sgu 103(poj 1158) 最短路（含蓝紫灯的最短路）

题意：一个普通最短路模型（节点间最多只有一条路，路径双向）。多加的条件是：每个节点上都一个蓝紫灯，初始为某种颜色，隔一定时间变换到下一种颜色，每种颜色的持续时长不同。只有当一条路径两个端点具有相同颜色的时候才能够通过（只是开始行走的那一刻相同就行，在路中央行走过程中无碍）。问从给定的起点到终点最小时间（最短路）。poj 1158 完全相同，只是不需要输出方案。

节点输入解释：

B 2 16 99

B：表示初始时灯为蓝（Blue）色

2：表示一开始的颜色持续2单位长

16：表示蓝色灯持续时长

99：表示紫色灯持续时长

思路：一个普通的最短路模型，因为边权恒为正且节点不多，用dijkstra就行。其中需要添加的地方是更新dis数组的时候为dis[j] = min(dis[j] ,dis[now]+weight+wait);其中的wait是等待now节点和j节点灯光颜色相同的时间，而wait值的求法比较麻烦。（这种需要灵活处理的地方还是不行，参考了http://blog.csdn.net/orpinex/article/details/6976510）

他将起始颜色全部转化为B,节点中的r代表到下一种颜色的时间。这样遇到灯颜色变换的时候就判断一次，最多只需判断四次，非常巧妙！！

#include <stdio.h>
#include <string.h>
#define MIN(a,b) ((a)<(b)?(a):(b))
#define INF 0x3fffffff
#define N 305
struct node{
	int r,b,loop;
}p[N];
struct edge{
	int y,w,next;
}e[30008];
int top,n,m,s,t;
int first[N],dis[N],pre[N],visited[N];
void init(){
	int i;
	top = 0;
	memset(first,-1,sizeof(first));
	memset(visited,0,sizeof(visited));
	for(i = 0;i<N;i++)
		dis[i] = INF;
}
void add(int x,int y,int w){
	e[top].y = y;
	e[top].w = w;
	e[top].next = first[x];
	first[x] = top++;
}
int update(int x,int y,int now,int w){
	int tx,ty,temp,res=0,k;
	tx = (now+p[x].r)%p[x].loop;
	ty = (now+p[y].r)%p[y].loop;
	for(k = 4;k;k--){
		if(tx<p[x].b&&ty<p[y].b || tx>=p[x].b&&ty>=p[y].b)
			break;
		if(tx < p[x].b)
			temp = p[x].b - tx;
		else
			temp = p[x].loop - tx;
		if(ty < p[y].b)
			temp = MIN(temp,p[y].b-ty);
		else
			temp = MIN(temp,p[y].loop-ty);
		res += temp;
		tx = (tx+temp)%p[x].loop;
		ty = (ty+temp)%p[y].loop;
	}
	if(k && dis[y]>dis[x]+w+res){
		dis[y] = dis[x]+w+res;
		return 1;
	}
	return 0;
}
int dijkstra(){
	int i,now,min;
	dis[s] = 0;
	while(1){
		min = INF;
		for(i = 1;i<=n;i++)
			if(!visited[i] && dis[i] < min){
				min = dis[i];
				now = i;
			}
		if(min == INF)
			return 0;
		if(now == t)
			return 1;
		visited[now] = 1;
		for(i = first[now];i!=-1;i=e[i].next)
			if(!visited[e[i].y] && (1 == update(now,e[i].y,dis[now],e[i].w))){
				pre[e[i].y] = now;
			}
	}
}
void print(int num){
	if(num != s)
		print(pre[num]);
	if(num != t)
		printf("%d ",num);
	else
		printf("%d\n",num);
}
int main(){
	int i,a,b,w,blu,pur,r;
	char ch;
	freopen("a.txt","r",stdin);
	init();
	scanf("%d %d",&s,&t);
	scanf("%d %d\n",&n,&m);
	for(i = 1;i<=n;i++){
		scanf("%c %d %d %d\n",&ch,&r,&blu,&pur);
		if(ch == 'B')
			p[i].r = blu-r;
		else
			p[i].r = blu+pur-r;
		p[i].loop = blu+pur;//l代表周期  
		p[i].b = blu;//b代表blue的时间
	}
	for(i = 0;i<m;i++){
		scanf("%d %d %d",&a,&b,&w);
		add(a,b,w);
		add(b,a,w);
	}
	if(dijkstra()){
		printf("%d\n",dis[t]);
		print(t);
	}else
		printf("0\n");
	return 0;//必不可少，否则GCC有可能RE
}