zoukankan      html  css  js  c++  java
  • D. GCD Counting(树上dp)

    题目链接:http://codeforces.com/contest/1101/problem/D

    题目大意:给你n个点,每个点都有权值,然后给你边的关系,问你树上的最大距离。(这里的最大距离指的是这条路径上的所有数最大gcd>1)。

     具体思路:首先,我们可以找出对于一个父亲节点,他和子节点能够在不是互素的前提下,对于他们每一个因子的个数。

    然后我们先dfs到底部,然后看他的每一个子节点之间产生的最大公因数是不是1,如果不是的话,我们就求这条路径上的最大值,在求最大值的时候,我们还是按照树的结构来,如图所示,从4号节点开始,4号节点更新之后会回到上一层,也就是3号节点,然后再求3号节点的所有因子和4号节点的因子是不是互素的,如果不是就从不是互素的这些状态里面找出一个最大值,然后在更新3号节点,这样更新下去就可以了。

    注意:应该是先求最大状态再去更新父亲节点,如果先更新再去找最大值的话,会有一部分路径是重复的。举个例子,对于2,3,6这三个点,正确的解法是先求2和当前3号节点的最大距离,然后更新2号节点,然后再轮到6号节点,这个时候2号节点里面存储的是和3号节点之间的关系,我们再去求和6号节点之间的关系,这样就不会有路径重复了。

     

    AC代码:

     1 #include<bits/stdc++.h>
     2 using namespace std;
     3 # define ll long long
     4 const int maxn = 4e5+100;
     5 int a[maxn],num,head[maxn],maxx;
     6 map<int,int>dis[maxn];
     7 struct node
     8 {
     9     int nex;
    10     int to;
    11 } edge[maxn];
    12 void init()
    13 {
    14     maxx=0;
    15     num=0;
    16     memset(head,-1,sizeof(head));
    17 }
    18 int gcd(int n,int m)
    19 {
    20     if(n<m)
    21         swap(n,m);
    22     return n%m==0?m:gcd(m,n%m);
    23 }
    24 void addedge(int fr,int to)
    25 {
    26     edge[num].nex=head[fr];
    27     edge[num].to=to;
    28     head[fr]=num++;
    29 }
    30 void dfs(int fr,int rt)
    31 {
    32     if(a[fr]>1)
    33     {
    34         dis[fr][a[fr]]=1;
    35     }
    36     for(int i=head[fr]; i!=-1; i=edge[i].nex)
    37     {
    38         int to=edge[i].to;
    39         if(to==rt)continue;
    40             dfs(to,fr);
    41         for(auto t1:dis[fr])
    42         {
    43             for(auto t2:dis[to])
    44             {
    45                 if(gcd(t1.first,t2.first)!=1)
    46                 {
    47                     maxx=max(maxx,t1.second+t2.second);
    48                 }
    49             }
    50         }
    51         for(auto t1:dis[to])
    52         {
    53             int tmp=gcd(a[fr],t1.first);
    54             if(tmp==1)
    55                 continue;
    56             dis[fr][tmp]=max(dis[fr][tmp],t1.second+1);
    57         }
    58     }
    59 }
    60 int main()
    61 {
    62     init();
    63     int n;
    64     scanf("%d",&n);
    65     int t1,t2;
    66     for(int i=1; i<=n; i++)
    67     {
    68         scanf("%d",&a[i]);
    69         if(a[i]!=1)
    70             maxx=1;
    71     }
    72     for(int i=1; i<=n-1; i++)
    73     {
    74         scanf("%d %d",&t1,&t2);
    75         addedge(t1,t2);
    76         addedge(t2,t1);
    77     }
    78     dfs(1,1);
    79     printf("%d
    ",maxx);
    80     return 0;
    81 }
  • 相关阅读:
    java算法----------常用的加密算法
    java多线程---------java.util.concurrent并发包----------等待多线程完成
    java多线程---------java.util.concurrent并发包----------ThreadPoolExecutor
    java多线程---------java.util.concurrent并发包
    一头扎进Spring之---------Spring核心容器----------
    深入理解java集合框架之---------Linked集合 -----构造函数
    深入理解java集合框架之---------Arraylist集合 -----添加方法
    深入理解java集合框架之---------Arraylist集合 -----构造函数
    深入理解java集合框架之---------HashMap集合
    linux输出与查看的几种方式
  • 原文地址:https://www.cnblogs.com/letlifestop/p/10262736.html
Copyright © 2011-2022 走看看