【ybtoj 冲刺 NOIP2021 模拟赛 B 组 Day9】 【二分】 【bfs】 道路与航线

【ybtoj 冲刺 NOIP2021 模拟赛 B 组 Day9】 【二分】 【bfs】 道路与航线

题目

题面

---

解题思路

二分时刻，跑bfs判断是否能连通
快读提速

---

代码

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
struct lzf{
	int to,nxt;
}f[3000010];
char w;
int a,b,n,m,l,r,tot,ans;
int u[3000010],v[3000010],c[3000010];
int head[3000010],p[3000010],q[3000010],dis[3000010];
int read() 
{
    int x=0;
    char c=getchar();
    while (c<'0'||c>'9')
	      c=getchar();
    while (c>='0'&&c<='9') 
    {
	      x=x*10+c-'0';
		  c=getchar();
	}
    return x;
}
void add(int x,int y)
{
	 f[++tot].to=y;
	 f[tot].nxt=head[x];
	 head[x]=tot;
}
bool bfs(int a,int b)
{
	memset(p,0,sizeof(p));
	memset(dis,0x7f,sizeof(dis));
	p[a]=1;	
	q[1]=a,dis[a]=0;
	int t=1,h=0;
	while (h<t)
	{ 
		  h++; 
		  for (int i=head[q[h]];i;i=f[i].nxt) 
		  if (dis[q[h]]+1<dis[f[i].to])
		  {
			 dis[f[i].to]=dis[q[h]]+1;
			 if (f[i].to==b) return true;
			 if (!p[f[i].to])
			 {
			    q[++t]=f[i].to;
				p[f[i].to]=1;
			 }
		   }
		  p[q[h]]=0;
	}
	return false;
}
int main()
{
	freopen("road.in","r",stdin);
	freopen("road.out","w",stdout);
	n=read(),m=read(),a=read(),b=read();
	for (int i=1;i<=m;i++)
		u[i]=read(),v[i]=read(),c[i]=read();
	l=1,r=m;
	while (l<=r)
	{
		  int mid=(l+r)/2;
		  memset(head,0,sizeof(head));
		  tot=0;
		  for (int i=1;i<=mid;i++)
		  {
		  	  add(u[i],v[i]);
		  	  if (!c[i]) add(v[i],u[i]);
		  }
		  if (bfs(a,b)) r=mid-1,ans=mid;
		     else l=mid+1; 
	}
	printf("%d ",ans);
	l=1,r=m;
	while (l<=r)
	{
		  int mid=(l+r)/2; 
		  memset(head,0,sizeof(head));
		  tot=0;
		  for (int i=1;i<=mid;i++)
		  {
		  	  add(u[i],v[i]);
		  	  if (!c[i]) add(v[i],u[i]);
		  }
		  if (bfs(b,a)) r=mid-1,ans=mid;
		     else l=mid+1; 
	}
	printf("%d",ans);
	return 0;
}