bzoj5506:[gzoi2019]旅行者

传送门

正反两边dijkstra染色,然后枚举一下边,求出最小值就好啦
代码:

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
void read(int &x) {
    char ch; bool ok;
    for(ok=0,ch=getchar(); !isdigit(ch); ch=getchar()) if(ch=='-') ok=1;
    for(x=0; isdigit(ch); x=x*10+ch-'0',ch=getchar()); if(ok) x=-x;
}
#define rg register
#define ll long long
const int maxn=1e5+10;bool vis[maxn];
struct oo{int x;ll dis;};
priority_queue<oo>q;ll ans,dis[2][maxn];
bool operator<(oo a,oo b){return a.dis>b.dis;}
int n,T,m,k,col[2][maxn],pre[maxn*5],nxt[maxn*5],v[maxn*5],h[maxn],cnt;
int x[maxn*5],y[maxn*5],z[maxn*5],t[maxn];
void add(int x,int y,int z){pre[++cnt]=y,nxt[cnt]=h[x],h[x]=cnt,v[cnt]=z;}
void dijkstra(ll *dis,int *col){
    memset(vis,0,sizeof vis);
    for(rg int i=1;i<=k;i++)dis[t[i]]=0,col[t[i]]=t[i],q.push((oo){t[i],dis[t[i]]});
    while(!q.empty()){
        oo x=q.top();q.pop();if(vis[x.x])continue;vis[x.x]=1;
        for(rg int i=h[x.x];i;i=nxt[i])
            if(!vis[pre[i]]&&dis[pre[i]]>dis[x.x]+v[i])
                dis[pre[i]]=dis[x.x]+v[i],col[pre[i]]=col[x.x],
                q.push((oo){pre[i],dis[pre[i]]});
    }
}
int main()
{
    read(T);
    while(T--){
        read(n),read(m),read(k);cnt=0;memset(h,0,sizeof h);
        memset(col,0,sizeof col);memset(dis,63,sizeof dis);
        for(rg int i=1;i<=m;i++)read(x[i]),read(y[i]),read(z[i]),add(x[i],y[i],z[i]);
        for(rg int i=1;i<=k;i++)read(t[i]);
        dijkstra(dis[0],col[0]);memset(h,0,sizeof h),cnt=0;
        for(rg int i=1;i<=m;i++)add(y[i],x[i],z[i]);
        dijkstra(dis[1],col[1]);ans=1e18;
        for(rg int i=1;i<=m;i++)
            if(col[0][x[i]]&&col[1][y[i]]&&col[0][x[i]]!=col[1][y[i]])ans=min(ans,dis[0][x[i]]+dis[1][y[i]]+z[i]);
        printf("%lld\n",ans);
    }
}
posted @ 2019-05-04 18:49 蒟蒻--lichenxi 阅读(...) 评论(...) 编辑 收藏