【题解】codevs 1391 伊吹萃香（分层图 最短路）

对于这道题我们可以利用分层图来解决。

我们首先设点1~n为白洞，点n+1~2n为黑洞。首先对于每一个点，由于可以留在原地，而且过了一秒之后路另一端会变色，所以我们将自己所代表的白点与自己所代表的黑点连边，边权为零，意思就是在某个点为白洞时选择留在原地，不消耗体力。同样的，我们将自己所代表的黑点也就是n+i与自己所代表的白点连边，边权为读入的是停留的点为黑洞时所要消耗的体力。然后我们将读入的点连边。如果读入的点为同色，那么妖怪从一个点走到另一个点后那边就变成了相反的颜色，所以我们从当前点到与当前点不同颜色的下一个点连边，因为走的时候路的两端颜色相同，所以边权为读入的z。如果读入的点颜色不同，那么走过去之后那边也变成相反的颜色，所以我们将起止点在同一颜色的那层里连边。注意将都是白色的情况连边权为0与本来的消耗的体力减掉节省的体力的最大值，这样可以避免负数的情况出现。都是黑色的情况边权就是本来消耗的体力加上要再消耗的体力。最后跑最短路就行，答案就是dis[n]和dis[2n]较小的那一个。注意team数组可以开大点。。。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdlib>
using namespace std;
const int maxn=10010;
const int maxm=1500100;
int n,m,tot;
int s,t,team[maxm*4];
int head[maxn],nnext[maxm*2],to[maxm*2],length[maxm*2];
int dis[maxn];
bool b[maxn];
int col[maxn],w[maxn],sx[maxn];
void add(int x,int y,int l)
{
	tot++;
	nnext[tot]=head[x];
	head[x]=tot;
	to[tot]=y;
	length[tot]=l;
}
void spfa()
{
	int w;
	if(col[1]==0) w=1;
	if(col[1]==1) w=n+1;
	for(int i=1;i<=n*2;i++)
	{
		dis[i]=1e9;
	}dis[w]=0;
	
	team[t]=w;
	t++;
	b[w]=true;
	
	while(s!=t)
	{
		int now=team[s];
		s++;
		b[now]=false;
		for(int i=head[now];i;i=nnext[i])
		{
			int y=to[i];
			if(dis[y]>dis[now]+length[i])
			{
				dis[y]=dis[now]+length[i];
				if(!b[y])
				{
					team[t]=y;
					t++;
					b[y]=true;
				}
			}
		}
	}
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) scanf("%d",&col[i]);
	for(int i=1;i<=n;i++) scanf("%d",&w[i]);
	for(int i=1;i<=n;i++) scanf("%d",&sx[i]);
	for(int i=1;i<=m;i++)
	{
		int x,y,z;
		scanf("%d%d%d",&x,&y,&z);
		if(col[x]==col[y]) 
		{
			add(x,n+y,z);
			add(n+x,y,z);
		}
		else
		{
			add(x,y,max(0,z-abs(w[x]-w[y])));
		 	add(n+x,n+y,z+abs(w[x]-w[y]));	
		} 
	}	
	for(int i=1;i<=n;i++)
	{
		add(i,n+i,0);
		add(n+i,i,sx[i]);	
	}	
	spfa();
	printf("%d",min(dis[n],dis[2*n]));
	return 0;
}