power oj 1687: 将军回家(最短路)

传送门

题意：n个城市，可以分成两类，被一条河相隔，求城市1到城市2的最短路，要求只能经过河
      一次；
分析：把同类城市的边单独跑dijskra，再枚举连接两类城市的边。       

#include <set>
#include <map>
#include <list>
#include <cmath>
#include <stack>
#include <queue>
#include <string>
#include <bitset>
#include <vector>
#include<cstring>
#include <stdio.h>
#include <iostream>
#include <algorithm>
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
inline int read(){int s=0,w=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();return s*w;}
const int maxn = 1e4+5;
int n,m,a[610];
int tot,head[610],to,h[610];
struct Edge{
    int to,next,val;
}edge[maxn<<2],E[maxn<<2],f[maxn<<2];
void add_edge(int u,int v,int w){
    edge[tot].to=v;
    edge[tot].next=head[u];
    edge[tot].val=w;
    head[u]=tot++;
}
void Add_edge(int u,int v,int w){
    E[to].to=v;
    E[to].next=h[u];
    E[to].val=w;
    h[u]=to++;
}
int dis[610],d[610];
bool vis[610];
struct node{
    int dis,pos;
    bool operator<(const node &x)const {
        return x.dis<dis;
    }
};
int e[610][610];
void Dijskra(int s)
{
    memset(vis,false,sizeof(vis));
    memset(dis,INF,sizeof(dis));
    priority_queue<node>q;
    q.push({0,s});
    dis[s]=0;
    while (!q.empty())
    {
        int x=q.top().pos; q.pop();
        if(vis[x]) continue; vis[x]=true;
        for(int i=head[x];i!=-1;i=edge[i].next){
            int y=edge[i].to;
 
            if(dis[y]>dis[x]+edge[i].val){
                dis[y]=dis[x]+edge[i].val;
                if(!vis[y]) q.push({dis[y],y});
            }
        }
    }
}
void dijskra(int s)
{
    memset(vis,false,sizeof(vis));
    memset(d,INF,sizeof(d));
    priority_queue<node>q;
    q.push({0,s});
    d[s]=0;
    while (!q.empty())
    {
        int x=q.top().pos; q.pop();
        if(vis[x]) continue; vis[x]=true;
        for(int i=h[x];i!=-1;i=E[i].next){
 
            int y=E[i].to;
            if(d[y]>d[x]+E[i].val){
                d[y]=d[x]+E[i].val;
                if(!vis[y]) q.push({d[y],y});
            }
        }
    }
}
int main()
{
    while (~scanf("%d",&n)&&n)
    {
        tot=0,to=0;
        memset(head,-1,sizeof(head));
        memset(h,-1,sizeof(h));
        memset(e,INF,sizeof(e));
        scanf("%d",&m);
        for(int i=1;i<=m;i++){
 
            int u=read(),v=read(),w=read();
 
            e[u][v]=min(e[u][v],w);
            e[v][u]=e[u][v];
        }
 
        int cnt=0;
        for(int i=1;i<=n;i++) a[i]=read();
 
        for(int i=1;i<=n;i++)
            for(int j=i+1;j<=n;j++)
            {
                if(e[i][j]==INF) continue;
 
                if(a[i]==a[j]&&a[i]==1)
                    add_edge(i,j,e[i][j]),add_edge(j,i,e[i][j]);
 
                else if(a[i]==a[j]&&a[i]==2)
                    Add_edge(i,j,e[i][j]),Add_edge(j,i,e[i][j]);
 
                else f[++cnt]={i,j,e[i][j]},f[++cnt]={j,i,e[i][j]};
 
        }
 
        Dijskra(1);
        dijskra(2);
 
        int Min=INF;
 
        for(int i=1;i<=cnt;i++)
        {
 
            if(a[f[i].to]==1&&a[f[i].next]==2){
                if(dis[f[i].to]==INF||d[f[i].next]==INF) continue;
                Min=min(Min,dis[f[i].to]+d[f[i].next]+e[f[i].to][f[i].next]);
            }
 
            else if(a[f[i].next]==1&&a[f[i].to]==2){
                if(dis[f[i].next]==INF||d[f[i].to]==INF) continue;
                Min=min(Min,dis[f[i].next]+d[f[i].to]+e[f[i].to][f[i].next]);
            }
 
        }
 
        if(Min==INF) puts("-1");
        else printf("%d\n",Min);
    }
    return 0;
}