SCU4444-Travel

题目链接：http://acm.scu.edu.cn/soj/problem.action?id=4444

题意:有n个城市,编号为1~n,每个城市都相互连通,其中有m对城市通过公路连通,其他的城市通过铁路连通,经过公路的时间为a,经过铁路的时间为b,问从1到达n的时间最短为多少

解题思路:求最短路问题,但是这道题中,每个点都是相连通的,不过连通的边有2种,那么从1走到n就可以分成2种情况:
①1和n通过公路相连,直接到达n的时间为a,那么就要考虑是直接用时a到达n,还是经过若干个b到达n
②1和n通过铁路相连,直接到达n的时间为b,那么就要考虑是直接用时b到达n,还是经过若干个a到达n

对于情况②我们很容易想到用dij做,不过情况①的边有n(n-1)/2-m条,显然用dij会超时，可以先开一个set集合s里面存放的是还要访问的通过若干个b到达的点。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm>
#include <cmath>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <vector>
#include <bitset>

using namespace std;

#define LL long long
const LL INF=0x3f3f3f3f3f3f3f3f;

const int maxn=100100;
int n,m,visit[maxn];
LL a,b,dis[maxn];
int s[maxn],e[maxn*10],nt[maxn*10],cnt;

struct node
{
    int id;
    int sum;
    friend bool operator <(node a,node b)
    {
        return a.sum>b.sum;
    }
};

void Dijkstra()
{
    priority_queue<node>q;
    for(int i=0;i<=n;i++) dis[i]=INF;
    memset(visit,0,sizeof visit);
    dis[1]=0;
    node pre,nt1;
    pre.id=1,pre.sum=0;
    q.push(pre);
    while(!q.empty())
    {
        pre=q.top();
        q.pop();
        visit[pre.id]=1;
        for(int i=s[pre.id];~i;i=nt[i])
        {
            int v=e[i];
            if(dis[v]>dis[pre.id]+1)
            {
                dis[v]=dis[pre.id]+1;
                nt1.id=v;
                nt1.sum=pre.sum+1;
                q.push(nt1);
            }
        }
    }
    printf("%lld\n",min(dis[n]*a,b));
}

void bfs()
{
    for(int i=0;i<=n;i++) dis[i]=INF;
    set<int>ss,sss;
    set<int>::iterator it;
    for(int i=2;i<=n;i++) ss.insert(i);
    queue<int>q;
    q.push(1);
    dis[1]=0;
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        for(int i=s[u];~i;i=nt[i])
        {
            int v=e[i];
            if(!ss.count(v)) continue;
            ss.erase(v),sss.insert(v);
        }
        for(it=ss.begin();it!=ss.end();it++)
        {
            q.push(*it);
            dis[*it]=dis[u]+1;
        }
        ss.swap(sss);
        sss.clear();
    }
    printf("%lld\n",min(dis[n]*b,a));
}

int main()
{
    while(~scanf("%d%d%lld%lld",&n,&m,&a,&b))
    {
        memset(s,-1,sizeof s);
        cnt=0;
        int flag=0;
        for(int i=0;i<m;i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            nt[cnt]=s[u],s[u]=cnt,e[cnt++]=v;
            nt[cnt]=s[v],s[v]=cnt,e[cnt++]=u;
            if((u==1&&v==n)||(u==n&&v==1)) flag=1;
        }
        if(flag) bfs();
        else Dijkstra();
    }
    return 0;
}