您的位置:首页 > 其它

[洛谷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;
}

 

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: