分层图最短路。
对于每个点,保存当它是黑洞和白洞时的最短路。转移时要看一看当前点是否是初始状态,注意好细节。
另外计算质量差的时候,要计算转移之前的状态。
其他的看代码就行了…
代码:
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
const int size=100010;
int head[size],nxt[size],dist[size][2],tot=0;
struct edge{
int t,d;
}l[size];
void build(int f,int t,int d)
{
l[++tot].t=t;
l[tot].d=d;
nxt[tot]=head[f];
head[f]=tot;
}
struct Heap{
int u,d,sta;
Heap(int u,int d,int sta):u(u),d(d),sta(sta){}
};
priority_queue<Heap> q;
bool operator <(Heap a,Heap b)
{
return a.d>b.d;
}
int n,m;
bool vis[size][2];
int start[size],w[size],s[size];
void dij(int s)
{
memset(dist,63,sizeof(dist));
dist[s][start[s]]=0;
q.push(Heap(s,0,start[s]));
while(q.size())
{
Heap f=q.top(); q.pop();
if(vis[f.u][f.sta]) continue;
vis[f.u][f.sta]=1;
for(int i=head[f.u];i;i=nxt[i])
{
int v=l[i].t;
int d=abs(w[f.u]-w[v]);
int et= f.sta==start[f.u]?start[v]:(start[v]^1);
int dd;
if(et==f.sta) dd=l[i].d;
else
{
if(f.sta==1) //1->0
dd=l[i].d+d;
else //0->1
dd=max(0,l[i].d-d);
}
// cout<<f.u<<" "<<f.sta<<" "<<v<<" "<<et<<" "<<dd<<endl;
if(dist[v][et^1]>dist[f.u][f.sta]+dd)
{
dist[v][et^1]=dist[f.u][f.sta]+dd;
q.push(Heap(v,dist[v][et^1],et^1));
}
}
int dd= f.sta==0?0: (::s[f.u]);
if(dist[f.u][f.sta^1]>dist[f.u][f.sta]+dd)
{
dist[f.u][f.sta^1]=dist[f.u][f.sta]+dd;
q.push(Heap(f.u,dist[f.u][f.sta^1],f.sta^1));
}
}
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) scanf("%d",&start[i]);
for(int i=1;i<=n;i++) scanf("%d",&w[i]);
for(int i=1;i<=n;i++) scanf("%d",&s[i]);
for(int i=1;i<=m;i++)
{
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
build(a,b,c);
}
dij(1);
printf("%d",min(dist[n][0],dist[n][1]));
return 0;
}
/*
4 5
1 0 1 0
10 10 100 10
5 20 15 10
1 2 30
2 3 40
1 3 20
1 4 200
3 4 200
*/