[HNOI2019]校园旅行(建图优化+bfs)

30分的O(m^2)做法应该比较容易想到:令f[i][j]表示i->j是否有解,然后把每个路径点数不超过2的有解状态(u,v)加入队列,然后弹出队列时,两点分别向两边搜索边,发现颜色一样时,再修改答案,加入队列即可。

100分是挺难想的,是个思维题,可以把边分成连接同色和异色两种。发现走过的路径一定是若干同色连通块拼接而成,除了中间的连通块外,其余长度均相等。对于长度,如果短,可以反复走把长度走到相等,重点是奇偶性要相同。所以,我们能够联想和二分图有关的东西。异色连通块,很显然是二分图,于是我们可以直接建立生成树。而同色的,如果是二分图则无需改变奇偶性,也连成一棵树。反之,则在树上连个自环即可,这样可以改变奇偶性。复杂度O(n^2)。据说HNOI今年卡栈,写dfs的都爆零?算了AH已经不和HN联考也不管了,反正OJ上过了……

#include<bits/stdc++.h>
using namespace std;
typedef pair<int,int>pii;
const int N=5005,M=1e6+7;
int n,m,Q,cnt,fa[N],hd[N],v[M],nxt[M],col[N];
char str[N];
bool f[N][N];
vector<int>G[N];
queue<pii>q;
int find(int x){return x==fa[x]?x:fa[x]=find(fa[x]);}
void add(int x,int y){v[++cnt]=y,nxt[cnt]=hd[x],hd[x]=cnt;}
bool dfs(int u,int c)
{
    col[u]=c;
    bool ret=0;
    for(int i=0;i<G[u].size();i++)
    if(col[G[u][i]]==-1)
    {
        add(u,G[u][i]),add(G[u][i],u);
        ret|=dfs(G[u][i],c^1);
        f[u][G[u][i]]=f[G[u][i]][u]=1;
        q.push(pii(min(u,G[u][i]),max(u,G[u][i])));
    }
    else if(col[G[u][i]]==c)ret=1;
    return ret;
}
int main()
{
    scanf("%d%d%d",&n,&m,&Q);
    scanf("%s",str+1);
    for(int i=1;i<=n;i++)fa[i]=i;
    for(int i=1,x,y;i<=m;i++)
    {
        scanf("%d%d",&x,&y);
        if(str[x]==str[y])G[x].push_back(y),G[y].push_back(x);
        else{
            int u=find(x),v=find(y);
            if(u!=v)fa[u]=v,add(x,y),add(y,x);
        }
    }
    memset(col,-1,sizeof col);
    for(int i=1;i<=n;i++)if(col[i]==-1&&dfs(i,0))add(i,i);
    for(int i=1;i<=n;i++)f[i][i]=1,q.push(pii(i,i));
    while(!q.empty())
    {
        int x=q.front().first,y=q.front().second;
        q.pop();
        for(int i=hd[x];i;i=nxt[i])
        for(int j=hd[y];j;j=nxt[j])
        if(str[v[i]]==str[v[j]]&&!f[v[i]][v[j]])
        f[v[i]][v[j]]=f[v[j]][v[i]]=1,q.push(pii(min(v[i],v[j]),max(v[i],v[j])));
    }
    while(Q--)
    {
        int x,y;scanf("%d%d",&x,&y);
        if(f[x][y])puts("YES");else puts("NO");
    }
}
View Code

猜你喜欢

转载自www.cnblogs.com/hfctf0210/p/10800719.html