【BZOJ5492】[HNOI2019]校园旅行(bfs)

时间:2020-12-26 14:05:06

[HNOI2019]校园旅行(bfs)

题面

洛谷

题解

首先考虑暴力做法怎么做。

把所有可行的二元组全部丢进队列里,每次两个点分别向两侧拓展一个同色点,然后更新可行的情况。

这样子的复杂度是\(O(m^2)\)的。

考虑如何优化边数,先说结论:

首先对于一个同色联通块,如果它是一个二分图,那么只需要保留一棵生成树就行了,否则随便找个点连一条自环。

对于连接不同色两个点的边,一定构成一个二分图,只需要保留一棵生成树就行了。

证明是这样子的:

首先我们把路径划分成若干个同色连续段,那么我们要做的就是对应的两段长度要相等。

长度短了是无所谓的,我们可以反复走一条边,达到把序列边长的目的。

对于一个二分图而言,如果反复走,其长度的奇偶性不会改变,否则奇偶性可以任意改变,那么需要连一个自环来改变奇偶性。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
#define ll long long
#define MAX 5050
inline int read()
{
int x=0;char ch=getchar();bool fl=false;
while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
if(ch=='-')fl=true,ch=getchar();
while(ch>='0'&&ch<='9')x=x*10+ch-48,ch=getchar();
return fl?-x:x;
}
struct Line{int v,next;}e[500500<<1];
int h[MAX],cnt=1;
inline void Add(int u,int v){e[cnt]=(Line){v,h[u]};h[u]=cnt++;}
int n,m,Qr;char a[MAX];
struct Node{int x,y;};queue<Node> Q;
bool vis[MAX][MAX];
vector<int> E[MAX];
int col[MAX];bool chk;
int f[MAX];
int getf(int x){return x==f[x]?x:f[x]=getf(f[x]);}
void dfs(int u,int c)
{
col[u]=c;
for(int i=0,l=E[u].size();i<l;++i)
{
int v=E[u][i];
if(a[u]!=a[v])continue;
if(col[v]==col[u])chk=false;
if(col[v])continue;
Add(u,v),Add(v,u);dfs(v,c^1);
vis[u][v]=vis[v][u]=true;
Q.push((Node){u,v});
}
}
int main()
{
freopen("tour.in","r",stdin);
freopen("tour.out","w",stdout);
n=read();m=read();Qr=read();scanf("%s",a+1);
for(int i=1;i<=n;++i)f[i]=i;
for(int i=1;i<=m;++i)
{
int u=read(),v=read();
E[u].push_back(v);
E[v].push_back(u);
if(a[u]!=a[v])
{
if(getf(u)==getf(v))continue;
Add(u,v);Add(v,u);
f[getf(u)]=getf(v);
}
}
for(int i=1;i<=n;++i)
if(!col[i])
{
chk=true;dfs(i,2);
if(!chk)Add(i,i);
}
for(int i=1;i<=n;++i)vis[i][i]=true,Q.push((Node){i,i});
while(!Q.empty())
{
Node u=Q.front();Q.pop();
int x=u.x,y=u.y;
for(int i=h[x];i;i=e[i].next)
{
int xx=e[i].v;
for(int j=h[y];j;j=e[j].next)
{
int yy=e[j].v;
if(vis[xx][yy])continue;
if(a[xx]!=a[yy])continue;
vis[xx][yy]=vis[yy][xx]=true;
Q.push((Node){xx,yy});
}
}
}
while(Qr--)
{
int x=read(),y=read();
if(vis[x][y])puts("YES");
else puts("NO");
}
return 0;
}