【东方模拟题】伊吹萃香 分层图最短路

---

分层图最短路。

对于每个点，保存当它是黑洞和白洞时的最短路。转移时要看一看当前点是否是初始状态，注意好细节。

另外计算质量差的时候，要计算转移之前的状态。

其他的看代码就行了…

代码：

#include
#include
#include
#include
#include
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 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<
            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
*/