cogs 826. Feb11] GF打dota

★★☆   输入文件：dota.in   输出文件：dota.out   简单对比

时间限制：1 s  
内存限制：128 MB

众所周知，GF同学喜欢打dota，而且打得非常好。今天GF和Spartan同学进行了一场大战。

现在GF拿到一张地图，地图上一共有n个地点，GF的英雄处于1号点，Spartan的基地位于n号点，

GF要尽快地选择较短的路线让他的英雄去虐掉Spartan的基地。但是Spartan早就料到了这一点，

他有可能会开挂(BS~)使用一种特别的魔法，一旦GF所走的路线的总长度等于最短路的总长度时，

GF的英雄就要和这种魔法纠缠不休。这时GF就不得不选择非最短的路线。现在请你替GF进行规划。

对于描述的解释与提醒：
1.无向路径，花费时间当然为非负值。

2.对于本题中非最短路线的定义：不管采取任何迂回、改道方式，

只要GF所走的路线总长度不等于1到n最短路的总长度时，就算做一条非最短的路线。

3.保证1~n有路可走。

输入：

第一行为n，m(表示一共有m条路径)

接下来m行，每行3个整数a，b，c，表示编号为a,b的点之间连着一条花费时间为c的无向路径。

接下来一行有一个整数p，p=0表示Spartan没有开挂使用这种魔法，p=1则表示使用了。

输出：

所花费的最短时间t，数据保证一定可以到达n。

样例输入1：

5 5

1 2 1

1 3 2

3 5 2

2 4 3

4 5 1

0

样例输入2：

5 5

1 2 1

1 3 2

3 5 2

2 4 3

4 5 1

1

样例输出1：
4
样例输出2：
5

对于50%的数据，1<=n,m<=5000

对于70%的数据，1<=n<=10000, 1<=m<=50000,p=0,

对于100%的数据，1<=n<=10000, 1<=m<=50000,p=1

无向图，花费时间c>=0

各个测试点1s

来源：lydliyudong    Tyvj February二月月赛第二场  第2道

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<queue>

using namespace std;
const int N=100010;
const int INF=99999999; 

int now1=1,now2=1;
int dis[N],head1[N];
int sta,ed;
bool vis[N];
int ans[N];
int n,m,k,js;
struct fan{
    int u,v,w,nxt;
}F[N];
struct zheng{
    int u,v,w,nxt;
}Z[N];
struct node{
    int point,noww,will;
}now,topp,nxt;

inline int read()
{
    int x=0;int f=1;char c=getchar();
    while(c<‘0‘||c>‘9‘){if(c==‘-‘)f=-1;c=getchar();}
    while(c>=‘0‘&&c<=‘9‘)x=x*10+c-‘0‘,c=getchar();
    return x*f;
}

inline void addzheng(int u,int v,int w)
{Z[now1].v=v;Z[now1].w=w;Z[now1].nxt=head1[u];head1[u]=now1++;}

bool operator < (node a,node b)
{return a.will>b.will;}

inline void spfa(int start)
{
    queue<int>q;
    for(int i=1;i<=n;i++)dis[i]=INF,vis[i]=0;
    dis[start]=0;vis[start]=1;
    q.push(start);
    while(!q.empty())
    {
        int ttop=q.front();q.pop();
        vis[ttop]=0;
        for(int i=head1[ttop];~i;i=Z[i].nxt)
            if(dis[Z[i].v]>dis[ttop]+Z[i].w)
                {dis[Z[i].v]=dis[ttop]+Z[i].w;if(!vis[Z[i].v])vis[Z[i].v]=1,q.push(Z[i].v);}
    }
}

inline void Astar(int start,int endd)
{
    if(dis[start]==INF)return ;
    now.point=start,now.noww=0;now.will=dis[start];
    priority_queue<node>q;
    q.push(now);
    while(!q.empty())
    {
        topp=q.top();
        q.pop();
        if(topp.point==endd){ans[++js]=topp.noww;if(js==2)return ;}
        for(int i=head1[topp.point];~i;i=Z[i].nxt)
            {nxt.point=Z[i].v;nxt.noww=topp.noww+Z[i].w;nxt.will=nxt.noww+dis[Z[i].v];q.push(nxt);}
    }
}

int main()
{
    freopen("dota.in","r",stdin);
    freopen("dota.out","w",stdout);
    n=read();m=read();
    for(int i=1;i<=n;i++)head1[i]=-1;
    for(int i=1;i<=m;i++)
        {int u=read(),v=read(),w=read();addzheng(u,v,w);addzheng(v,u,w);}
    sta=1;ed=n;k=read();
    spfa(ed);
    Astar(sta,ed);
    if(!k)printf("%d\n",ans[1]);
    else printf("%d\n",ans[2]);
    return 0;
}