BZOJ 4238 电压 DFS+拓扑

题意:链接

方法: DFS+拓扑

解析:

想的差一点的题TAT

首先我们可以DFS出任意一棵树。

然后讨论树边与非树边。

如果非树边的两端点在树上差距为偶数，则该非树边为断边，若为奇数，则为连边。

考虑取非树边的情况:

当且仅当仅有一条非树边为断边时，可取该边，对答案贡献为1，无其他情况。

考虑取树边的情况:

如果一条树边被所有的断边覆盖且不被连边覆盖时，该树边为可取树边。

非树边情况直接扫边求解即可。

树边情况需要用些巧解。

由于我们是DFS出一棵树。

所以所有的非树边均为返祖边，即对于两点的lca来说，一定为两点deep较小的那一个。

现在我们扫所有的边。

如果扫到非树边并且该边是个连边，则给该非树边的两个端点打一个-标记，给两个端点的lca打两个+标记。

如果扫到非树边并且该边是个断边，则给该非树边的两个端点打一个+标记，给两个端点的lca打两个-标记。

最后拓扑推一下所有的标记。

再扫描所有点，如果该点的值恰为断边数量，则对答案贡献为1。

统计即可。

代码:

#include <queue>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define N 100100 
#define M 200100
using namespace std; 
int fa[N];
int deep[N];
int head[N];
int v[N];
int tag[M<<1];
int on[M<<1]; 
int col[N];
int du[N];
int canchose[N];
int cnt,n,m;
queue<int>q;
struct node
{
    int from,to,next;
}edge[M<<1];
void init()
{
    memset(head,-1,sizeof(head));
    cnt=0; 
}
void edgeadd(int from,int to)
{
    edge[cnt].from=from,edge[cnt].to=to,edge[cnt].next=head[from];
    head[from]=cnt++;
}
void dfs(int x)
{
    int i;
    deep[x]=deep[fa[x]]+1,v[x]=1,du[x]=1;
    for(i=head[x];i!=-1;i=edge[i].next)
    {
        canchose[x]=1;
        if(edge[i].to==fa[x]||v[edge[i].to])continue;
        on[i|1]=1;
        fa[edge[i].to]=x;
        dfs(edge[i].to);
        du[x]++;    
    }
    if(du[x]==1)q.push(x);
}
int main()
{   
    freopen("voltage.in","r",stdin);
    freopen("voltage.out","w",stdout);
    scanf("%d%d",&n,&m);
    init(); 
    for(int i=1;i<=m;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        if(x>y)swap(x,y); 
        edgeadd(x,y),edgeadd(y,x); 
    }
    for(int i=1;i<=n;i++)
    {
        if(!v[i])
        {
            dfs(i);
            canchose[i]=0;
        }
    }
    int cannot=0,ans=0;
    for(int i=1;i<cnt;i+=2)
    {
        if(!on[i])
        {
            int lenth=deep[edge[i].to]-deep[edge[i].from];
            if(lenth&1)
                tag[i]=1;
            else tag[i]=-1,cannot++; 
        }else
        {
            tag[i]=1;
        }
    }
    for(int i=1;i<cnt;i+=2)
    {
        if(!on[i]&&tag[i]==-1)
        {
            int t;
            if(deep[edge[i].from]<deep[edge[i].to])
                t=edge[i].from;
            else t=edge[i].to;
            col[edge[i].from]++,col[edge[i].to]++;
            col[t]-=2;
        }else if(!on[i]&&tag[i]==1)
        {
            int t;
            if(deep[edge[i].from]<deep[edge[i].to])
                t=edge[i].from;
            else t=edge[i].to;
            col[edge[i].from]--,col[edge[i].to]--;
            col[t]+=2;
        }
    }
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        col[fa[u]]+=col[u];
        du[fa[u]]--;
        if(du[fa[u]]==1)q.push(fa[u]);
    }
    for(int i=2;i<=n;i++)
        if(col[i]==cannot&&canchose[i])ans++;
    if(cannot==1)ans++;
    printf("%d\n",ans);
}