ping ping ping HDU - 6203

http://acm.hdu.edu.cn/showproblem.php?pid=6203

给出q条链 由u和v确定 每条链上至少有一个节点是有故障的 问整个树图中至少有多少故障节点

对于每条链 求出lca 再按lca的深度降序排序

然后对于一条链 如果uv节点都没有被其他链覆盖过 那就将lca对应的整棵子树标记覆盖 答案加加 否则就说明当前这条链上的故障点已经可以和别的链合并了 可以忽略

至于为什么要按深度降序排序 我认为这样每次只需要判断一条链是不是已经通过其他链确定 如果升序排序 每一次要看lca到u和v这条链上有多少其它链上的lca被影响 很难写   (借鉴)同时 由于优先处理 LCA 深度大的点 不会出现点 U V 同时在同一个被禁止通行点 P 的子树内

#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
using namespace std;

struct node0
{
    int u;
    int v;
    int lca;
};

struct node1
{
    int v;
    int next;
};

node0 pre[50010];
node1 edge[60010];
int dp[30010][15];
int val[120010];
int first[30010],deep[30010],mp[30010],sum[30010];
int n,q,num;

bool cmp(node0 n1,node0 n2)
{
    return deep[n1.lca]>deep[n2.lca];
}

void addedge(int u,int v)
{
    edge[num].v=v;
    edge[num].next=first[u];
    first[u]=num++;
}

void dfs(int cur,int fa)
{
    int i,v;
    mp[cur]=++num,sum[cur]=1;
    for(i=first[cur];i!=-1;i=edge[i].next)
    {
        v=edge[i].v;
        if(v!=fa)
        {
            dp[v][0]=cur;
            deep[v]=deep[cur]+1;
            dfs(v,cur);
            sum[cur]+=sum[v];
        }
    }
    return;
}

void solve()
{
    int i,j;
    dp[1][0]=0;
    deep[1]=1;
    num=0;
    dfs(1,0);
    for(j=1;(1<<j)<=n;j++)
    {
        for(i=1;i<=n;i++)
        {
            dp[i][j]=dp[dp[i][j-1]][j-1];
        }
    }
    return;
}

int getlca(int u,int v)
{
    int i;
    if(deep[u]<deep[v]) swap(u,v);
    for(i=log2(n);i>=0;i--)
    {
        if(deep[dp[u][i]]>=deep[v])
        {
            u=dp[u][i];
        }
    }
    if(u==v) return u;
    for(i=log2(n);i>=0;i--)
    {
        if(dp[u][i]!=dp[v][i])
        {
            u=dp[u][i];
            v=dp[v][i];
        }
    }
    return dp[u][0];
}

void query(int tar,int &res,int l,int r,int cur)
{
    int m;
    res|=val[cur];
    if(l==r) return;
    m=(l+r)/2;
    if(tar<=m) query(tar,res,l,m,2*cur);
    else query(tar,res,m+1,r,2*cur+1);
}

void update(int pl,int pr,int l,int r,int cur)
{
    int m;
    if(pl<=l&&r<=pr)
    {
        val[cur]=1;
        return;
    }
    m=(l+r)/2;
    if(pl<=m) update(pl,pr,l,m,2*cur);
    if(pr>m) update(pl,pr,m+1,r,2*cur+1);
}

int main()
{
    int i,u,v,resu,resv,ans;
    while(scanf("%d",&n)!=EOF)
    {
        n++;
        memset(first,-1,sizeof(first));
        num=0;
        for(i=1;i<=n-1;i++)
        {
            scanf("%d%d",&u,&v);
            u++,v++;
            addedge(u,v);
            addedge(v,u);
        }
        solve();
        scanf("%d",&q);
        for(i=1;i<=q;i++)
        {
            scanf("%d%d",&pre[i].u,&pre[i].v);
            pre[i].u++,pre[i].v++;
            pre[i].lca=getlca(pre[i].u,pre[i].v);
        }
        sort(pre+1,pre+q+1,cmp);
        for(i=1;i<=4*n;i++) val[i]=0;
        ans=0;
        for(i=1;i<=q;i++)
        {
            resu=0,resv=0;
            query(mp[pre[i].u],resu,1,n,1);
            query(mp[pre[i].v],resv,1,n,1);
            if(!resu&&!resv)
            {
                update(mp[pre[i].lca],mp[pre[i].lca]+sum[pre[i].lca]-1,1,n,1);
                ans++;
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/sunyutian1998/article/details/82155271