poj3417 Network——LCA+树上差分

题目：http://poj.org/problem?id=3417

根据一条边被几个环覆盖来判断能不能删、有几种情况等；

用树上差分，终点 s++，LCA s-=2，统计时计算子树s值的和即可；

用ST表做LCA，不知为何WA了：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int const MAXN=1e5+;
int n,m,head[MAXN],ct,dep[MAXN],pre[MAXN][],s[MAXN];
long long ans;
struct N{
    int to,next;
    N(int t=,int n=):to(t),next(n) {}
}edge[MAXN<<];
void dfs(int x,int f)
{
    dep[x]=dep[f]+;
    pre[x][]=f;
    for(int i=head[x];i;i=edge[i].next)
        if(edge[i].to!=f)dfs(edge[i].to,x);
}
void init()
{
    dfs(,);
    for(int k=;k<=;k++)
        for(int i=;i<=n;i++)
            pre[i][k]=pre[pre[i][k-]][k-];
}
int lca(int x,int y)
{
    if(dep[x]>dep[y])swap(x,y);
    int d=dep[y]-dep[x];
    for(int i=;i<=;i++)
        if(/*d&(1<<i)*/(d>>i)&)y=pre[y][i];
    for(int i=;i>=;i--)
        if(pre[x][i]!=pre[y][i])
        {
            x=pre[x][i];
            y=pre[y][i];
        }
    return pre[x][];
}
long long dfs2(int x,int f)
{
    long long sum=s[x];
    for(int i=head[x],u;i;i=edge[i].next)
    {
        if((u=edge[i].to)==f)continue;
        long long k=dfs2(u,x);
        if(k==)ans+=m;
        if(k==)ans++;
        sum+=k;
    }
    return sum;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=,x,y;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        edge[++ct]=N(y,head[x]);head[x]=ct;
        edge[++ct]=N(x,head[y]);head[y]=ct;
    }
    init();
    for(int i=,x,y;i<=m;i++)
    {
        scanf("%d%d",&x,&y);
        if(x==y)continue;
        s[x]++;s[y]++;
        s[lca(x,y)]-=;
    }
    dfs2(,);
    printf("%lld",ans);
    return ;
}

ST表为什么WA

于是改成了tarjan，过程中求答案；

注意非树边加边时判掉x=y的情况。

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int const MAXN=1e5+;
int n,m,head[MAXN],ct,s[MAXN],fa[MAXN],ct2,head2[MAXN];
long long ans;
bool vis[MAXN];
struct N{
    int to,next;
    N(int t=,int n=):to(t),next(n) {}
}edge[MAXN<<],ed[MAXN<<];
int find(int x)
{
    if(fa[x]==x)return x;
    return fa[x]=find(fa[x]);
}
void tarjan(int x)
{
    fa[x]=x;vis[x]=;
    for(int i=head2[x],u;i;i=ed[i].next)
        if(vis[u=ed[i].to])s[find(u)]-=;
    for(int i=head[x],u;i;i=edge[i].next)
    {
        if(vis[u=edge[i].to])continue;//fa
        tarjan(u);fa[u]=x;s[x]+=s[u];
        if(s[u]==)ans+=m;
        if(s[u]==)ans++;
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=,x,y;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        edge[++ct]=N(y,head[x]);head[x]=ct;
        edge[++ct]=N(x,head[y]);head[y]=ct;
    }
    for(int i=,x,y;i<=m;i++)
    {
        scanf("%d%d",&x,&y);
        if(x==y)continue;//!
        s[x]++;s[y]++;
        ed[++ct2]=N(y,head2[x]);head2[x]=ct2;
        ed[++ct2]=N(x,head2[y]);head2[y]=ct2;
    }
    tarjan();
    printf("%d",ans);
    return ;
}