[洛谷1346]电车
2017-07-26 18:48
330 查看
思路:
将问题抽象成一张图。需要切开关则边权是1。求单源最短路即可。
(由于数据水,所以暴力也能过)
#include<cstdio> #include<cctype> #include<vector> #include<algorithm> #include<functional> #include<ext/pb_ds/priority_queue.hpp> inline int getint() { char ch; while(!isdigit(ch=getchar())); int x=ch^'0'; while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0'); return x; } const int N=101,inf=0x7fffffff; struct Edge { int to,w; bool operator > (const Edge &x) const { return w>x.w; } }; std::vector<int> e ; int w ; inline void add_edge(const int u,const int v,const int ww) { e[u].push_back(v); w[u][v]=ww; } int d = {0}; bool v = {0}; int main() { int n=getint(),a=getint(),b=getint(); for(int i=1; i<=n; i++) { int k=getint(); for(int j=0; j<k; j++) { add_edge(i,getint(),j?1:0); } } __gnu_pbds::priority_queue<Edge,std::greater<Edge> > q; __gnu_pbds::priority_queue<Edge,std::greater<Edge> >::point_iterator p[n+1]; for(int i=1; i<=n; i++) p[i]=q.push((Edge){i,d[i]=(i==a)?0:inf}); for(int i=0; i<n; i++) { Edge u=q.top(); if(u.w==inf) break; for(unsigned int j=0; j<e[u.to].size(); j++) { if(v[e[u.to][j]]) continue; int &v=e[u.to][j]; q.modify(p[v],(Edge) { v,d[v]=std::min(p[v]->w,u.w+w[u.to][v]) }); } q.modify(p[u.to],(Edge){u.to,inf}); v[u.to]=true; } printf("%d\n",(d[b]==inf)?-1:d[b]); return 0; }