• 【POJ3417】Network-LCA算法+树上差分


    题目大意:一棵有N个点的树,再往里面加入M条新边,现在要破坏其中的两条边,要求一条是原来树中的边,一条是新边,求方案的数量。

    做法:点击打开链接

    以下是本人代码:

    #include <cstdio>
    #include <cstdlib>
    #include <cstring>
    #include <iostream>
    #include <algorithm>
    using namespace std;
    int n,m,tot,first[100010],firstq[100010],f[100010],fa[100010],dp[100010],lca[100010];
    long long ans=0;
    bool vis[100010];
    struct {int v,next;} e[200010],q[200010];
    
    void insert(int a,int b)
    {
      e[++tot].v=b;
      e[tot].next=first[a];
      first[a]=tot;
    }
    
    void insertq(int a,int b)
    {
      q[++tot].v=b;
      q[tot].next=firstq[a];
      firstq[a]=tot;
    }
    
    int find(int x)
    {
      int r=x,i=x,j;
      while(f[r]!=r) r=f[r];
      while(i!=r) {j=f[i];f[i]=r;i=j;}
      return r;
    }
    
    void merge(int a,int b)
    {
      f[find(a)]=find(b);
    }
    
    void tarjan(int v)
    {
      f[v]=v;
      for(int i=first[v];i;i=e[i].next)
        if (e[i].v!=fa[v])
    	{
    	  fa[e[i].v]=v;
    	  tarjan(e[i].v);
    	}
      vis[v]=1;
      for(int i=firstq[v];i;i=q[i].next)
        if (vis[q[i].v]) lca[(i+1)/2]=find(q[i].v);
      merge(v,fa[v]);
    }
    
    void DP(int v)
    {
      for(int i=first[v];i;i=e[i].next)
        if (e[i].v!=fa[v])
    	{
    	  DP(e[i].v);
    	  dp[v]+=dp[e[i].v];
    	}
      if (v!=1)
      {
        if (dp[v]==0) ans+=m;
        if (dp[v]==1) ans++;
      }
    }
    
    int main()
    {
      scanf("%d%d",&n,&m);
      memset(first,0,sizeof(first));
      memset(firstq,0,sizeof(firstq));
      memset(vis,0,sizeof(vis));
      memset(dp,0,sizeof(dp));
      fa[1]=tot=0;
      for(int i=1;i<n;i++)
      {
        int a,b;
    	scanf("%d%d",&a,&b);
    	insert(a,b);insert(b,a);
      }
      tot=0;
      for(int i=1;i<=m;i++)
      {
        int a,b;
    	scanf("%d%d",&a,&b);
    	insertq(a,b);insertq(b,a);
      }
      tarjan(1);
      for(int i=1;i<=m;i++)
      {
        int a=q[i*2].v,b=q[i*2-1].v;
        dp[a]++;dp[b]++;dp[lca[i]]-=2;
      }
      DP(1);
      printf("%lld
    ",ans);
      
      return 0;
    }
    


  • 相关阅读:
    sh_09_字典的定义
    sh_08_格式化字符串
    sh_07_元组遍历
    sh_06_元组基本使用
    sh_05_列表遍历
    sh_04_列表排序
    sh_03_列表的数据统计
    图片懒加载
    UA池和ip代理池
    爬虫篇 --- 分布式爬虫
  • 原文地址:https://www.cnblogs.com/Maxwei-wzj/p/9793951.html
Copyright © 2020-2023  润新知