BZOJ 1726 严格次短路

不说什么了，模板题

题目数据有点弱，不严格的次短路也能ac。。。。

 

View Code 
  1 #include <cstring>
  2 #include <cstdlib>
  3 #include <algorithm>
  4 #include <iostream>
  5 #include <cstdio>
  6 
  7 #define N 100000
  8 #define M 2000000
  9 
 10 using namespace std;
 11 
 12 struct HP
 13 {
 14     int x,d;
 15 }hp[M];
 16 
 17 int head[N],next[M],dis[N],to[M],len[M],cnt,n,qk,m,q[M],cs[N],size;
 18 bool vis[N];
 19 
 20 inline void add(int u,int v,int w)
 21 {
 22     to[cnt]=v; len[cnt]=w; next[cnt]=head[u]; head[u]=cnt++;
 23 }
 24 
 25 inline bool cmp(const HP &a,const HP &b)
 26 {
 27     return a.d+dis[a.x]>b.d+dis[b.x];
 28 }
 29 
 30 void read()
 31 {
 32     memset(head,-1,sizeof head); cnt=0;
 33     for(int i=1,a,b,c;i<=m;i++)
 34     {
 35         scanf("%d%d%d",&a,&b,&c);
 36         add(a,b,c); add(b,a,c);
 37     }
 38     qk=2;
 39 }
 40 
 41 void spfa()
 42 {
 43     memset(dis,0x3f,sizeof dis);
 44     int h=1,t=2,sta;
 45     q[1]=n; vis[n]=true; dis[n]=0;
 46     while(h<t)
 47     {
 48         sta=q[h++];
 49         vis[sta]=false;
 50         for(int i=head[sta];~i;i=next[i])
 51             if(dis[to[i]]>dis[sta]+len[i])
 52             {
 53                 dis[to[i]]=dis[sta]+len[i];
 54                 if(!vis[to[i]]) vis[to[i]]=true,q[t++]=to[i];
 55             }
 56     }
 57 }
 58 
 59 int bfs()
 60 {
 61     memset(cs,0,sizeof cs);
 62     bool fg=false;int mindis=-1;
 63     hp[1].x=1; hp[1].d=0; size=1;
 64     while(size)
 65     {
 66         HP sta=hp[1];
 67         pop_heap(hp+1,hp+1+size,cmp); size--;
 68         
 69         if(sta.x!=n) cs[sta.x]++;
 70         else if(sta.d!=mindis) cs[sta.x]++;
 71         
 72         if(sta.x==n&&!fg)
 73         {
 74             mindis=sta.d;
 75             fg=true;
 76         }
 77         if(cs[n]==qk) return sta.d;
 78         if(cs[sta.x]>qk) continue;
 79         for(int i=head[sta.x];~i;i=next[i])
 80         {
 81             size++;
 82             hp[size].x=to[i]; hp[size].d=sta.d+len[i];
 83             push_heap(hp+1,hp+1+size,cmp);
 84         }
 85     }
 86 }
 87 
 88 void go()
 89 {
 90     spfa();
 91     if(1==n) qk++;
 92     printf("%d\n",bfs());
 93 }
 94 
 95 int main()
 96 {
 97     while(scanf("%d%d",&n,&m)!=EOF)
 98     {
 99         read();
100         go();
101     }
102     return 0;
103 }

 

 

posted @ 2012-10-29 22:55  proverbs  阅读(313)  评论(0)  编辑  收藏  举报