洛谷 P2296 寻找道路 —— bfs

题目:https://www.luogu.org/problemnew/show/P2296

第一次用 Emacs 对拍,写了半天;

注意那个 is 赋值的地方很容易错,千万别反复赋值;

一道水题写了一个下午,崩溃。

代码如下:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
int const maxn=1e4+5,maxm=2e5+5;
int n,m,s,t,hd[maxn],ct,to[maxm],nxt[maxm],dis[maxn],hdf[maxn],tof[maxm],nxtf[maxm],ctf;
bool vis[maxn],is[maxn];
queue<int>q;
void add(int x,int y){to[++ct]=y; nxt[ct]=hd[x]; hd[x]=ct;}
void add2(int x,int y){tof[++ctf]=y; nxtf[ctf]=hdf[x]; hdf[x]=ctf;}
int rd()
{
  int ret=0,f=1; char ch=getchar();
  while(ch<'0'||ch>'9'){if(ch=='-')f=-1; ch=getchar();}
  while(ch>='0'&&ch<='9')ret=(ret<<3)+(ret<<1)+ch-'0',ch=getchar();
  return ret*f;
}
void init()
{
  q.push(t); vis[t]=1; is[t]=1;
  while(q.size())
    {
      int x=q.front(); q.pop();
      for(int i=hdf[x],u;i;i=nxtf[i])
    {
      if(vis[u=tof[i]])continue;
      vis[u]=1; q.push(u); is[u]=1;//is=1
    }
    }
  for(int i=1;i<=n;i++)
    {
      //      if(vis[i]){is[i]=1; continue;}
      if(vis[i])continue;
      for(int j=hdf[i];j;j=nxtf[j])
    is[tof[j]]=0;
    }
}
void bfs()
{
  memset(vis,0,sizeof vis);
  while(q.size())q.pop();
  q.push(s); vis[s]=1;
  while(q.size())
    {
      int x=q.front(); q.pop();
      for(int i=hd[x],u;i;i=nxt[i])
    {
      if(vis[u=to[i]]||!is[u])continue;
      vis[u]=1; q.push(u); dis[u]=dis[x]+1;
    }
    }
}
int main()
{
  n=rd(); m=rd();
  for(int i=1,x,y;i<=m;i++)
    {
      x=rd(),y=rd();
      if(x==y)continue;
      add(x,y),add2(y,x);
    }
  s=rd(); t=rd();
  init(); 
  if(!is[s]){printf("-1\n"); return 0;}//
  bfs();
  if(!vis[t])printf("-1\n");
  else printf("%d\n",dis[t]);
  return 0;
}

 

posted @ 2018-09-13 16:53  Zinn  阅读(157)  评论(0编辑  收藏  举报